ASPLOS 2023 – Author Index

A B C D E F G H I J K L M N O P Q R S T V W X Y Z

	Abu-Ghazaleh, Nael	ASPLOS '23: "CommonGraph: Graph Analytics ..." CommonGraph: Graph Analytics on Evolving Data Mahbod Afarin, Chao Gao, Shafiur Rahman, Nael Abu-Ghazaleh, and Rajiv Gupta (University of California at Riverside, USA) We consider the problem of graph analytics on evolving graphs (i.e., graphs that change over time). In this scenario, a query typically needs to be applied to different snapshots of the graph over an extended time window, for example to track the evolution of a property over time. Solving a query independently on multiple snapshots is inefficient due to repeated execution of subcomputation common to multiple snapshots. At the same time, we show that using streaming, where we start from the earliest snapshot and stream the changes to the graph incrementally updating the query results one snapshot at a time is also inefficient. We propose CommonGraph, an approach for efficient processing of queries on evolving graphs. We first observe that deletion operations are significantly more expensive than addition operations for many graph queries (those that are monotonic). CommonGraph converts all deletions to additions by finding a common graph that exists across all snapshots. After computing the query on this graph, to reach any snapshot, we simply need to add the missing edges and incrementally update the query results. CommonGraph also allows sharing of common additions among snapshots that require them, and breaks the sequential dependency inherent in the traditional streaming approach where snapshots are processed in sequence, enabling additional opportunities for parallelism. We incorporate the CommonGraph approach by extending the KickStarter streaming framework. We implement optimizations that enable efficient handling of edge additions without resorting to expensive in place graph mutations, significantly reducing the streaming overhead, and enabling direct reuse of shared edges among different snapshots. CommonGraph achieves 1.38x-8.17x improvement in performance over Kickstarter across multiple benchmarks. @InProceedings{ASPLOS23p133, author = {Mahbod Afarin and Chao Gao and Shafiur Rahman and Nael Abu-Ghazaleh and Rajiv Gupta}, title = {CommonGraph: Graph Analytics on Evolving Data}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {133--145}, doi = {10.1145/3575693.3575713}, year = {2023}, } Publisher's Version
	Acun, Bilge	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Afarin, Mahbod	ASPLOS '23: "CommonGraph: Graph Analytics ..." CommonGraph: Graph Analytics on Evolving Data Mahbod Afarin, Chao Gao, Shafiur Rahman, Nael Abu-Ghazaleh, and Rajiv Gupta (University of California at Riverside, USA) We consider the problem of graph analytics on evolving graphs (i.e., graphs that change over time). In this scenario, a query typically needs to be applied to different snapshots of the graph over an extended time window, for example to track the evolution of a property over time. Solving a query independently on multiple snapshots is inefficient due to repeated execution of subcomputation common to multiple snapshots. At the same time, we show that using streaming, where we start from the earliest snapshot and stream the changes to the graph incrementally updating the query results one snapshot at a time is also inefficient. We propose CommonGraph, an approach for efficient processing of queries on evolving graphs. We first observe that deletion operations are significantly more expensive than addition operations for many graph queries (those that are monotonic). CommonGraph converts all deletions to additions by finding a common graph that exists across all snapshots. After computing the query on this graph, to reach any snapshot, we simply need to add the missing edges and incrementally update the query results. CommonGraph also allows sharing of common additions among snapshots that require them, and breaks the sequential dependency inherent in the traditional streaming approach where snapshots are processed in sequence, enabling additional opportunities for parallelism. We incorporate the CommonGraph approach by extending the KickStarter streaming framework. We implement optimizations that enable efficient handling of edge additions without resorting to expensive in place graph mutations, significantly reducing the streaming overhead, and enabling direct reuse of shared edges among different snapshots. CommonGraph achieves 1.38x-8.17x improvement in performance over Kickstarter across multiple benchmarks. @InProceedings{ASPLOS23p133, author = {Mahbod Afarin and Chao Gao and Shafiur Rahman and Nael Abu-Ghazaleh and Rajiv Gupta}, title = {CommonGraph: Graph Analytics on Evolving Data}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {133--145}, doi = {10.1145/3575693.3575713}, year = {2023}, } Publisher's Version
	Agarwal, Ishwar	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Agrawal, Gaurav	ASPLOS '23: "FLAT: An Optimized Dataflow ..." FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks Sheng-Chun Kao, Suvinay Subramanian, Gaurav Agrawal, Amir Yazdanbakhsh, and Tushar Krishna (Georgia Institute of Technology, USA; Google, USA; Microsoft, USA) Attention mechanisms, primarily designed to capture pairwise correlations between words, have become the backbone of machine learning, expanding beyond natural language processing into other domains. This growth in adaptation comes at the cost of prohibitively large memory requirements and computational complexity, especially at higher number of input elements. This limitation is due to inherently limited data reuse opportunities and quadratic growth in memory footprints, leading to severe memory-boundedness and limited scalability of input elements. This work addresses these challenges by devising a tailored dataflow optimization, called FLAT, for attention mechanisms without altering their functionality. This dataflow processes costly attention operations through a unique fusion mechanism, transforming the memory footprint quadratic growth to merely a linear one. To realize the full potential of this bespoke mechanism, we propose a tiling approach to enhance the data reuse across attention operations. Our method both mitigates the off-chip bandwidth bottleneck as well as reduces the on-chip memory requirement. FLAT delivers 1.94x (1.76x) speedup and 49% and (42%) of energy savings compared to the state-of-the-art Edge (Cloud) accelerators with no customized dataflow optimization. When on-chip resources are scarce (20 KB-200 KB), FLAT yields, on average, 1.5x end-to-end latency reduction across a diverse range of conventional attention-based models with input sequence lengths ranging from 512-token to 64K-token. Our evaluations demonstrate that state-of-the-art DNN dataflow applied to attention operations reach the efficiency limit for inputs above 512 elements. In contrast, FLAT unblocks transformer models for inputs with up to 64K elements. @InProceedings{ASPLOS23p295, author = {Sheng-Chun Kao and Suvinay Subramanian and Gaurav Agrawal and Amir Yazdanbakhsh and Tushar Krishna}, title = {FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {295--310}, doi = {10.1145/3575693.3575747}, year = {2023}, } Publisher's Version
	Aguilera, Marcos K.	ASPLOS '23: "uBFT: Microsecond-Scale BFT ..." uBFT: Microsecond-Scale BFT using Disaggregated Memory Marcos K. Aguilera, Naama Ben-David, Rachid Guerraoui, Antoine Murat, Athanasios Xygkis, and Igor Zablotchi (VMware Research, USA; EPFL, Switzerland; Massachusetts Institute of Technology, USA) We propose uBFT, the first State Machine Replication (SMR) system to achieve microsecond-scale latency in data centers, while using only 2f+1 replicas to tolerate f Byzantine failures. The Byzantine Fault Tolerance (BFT) provided by uBFT is essential as pure crashes appear to be a mere illusion with real-life systems reportedly failing in many unexpected ways. uBFT relies on a small non-tailored trusted computing base—disaggregated memory—and consumes a practically bounded amount of memory. uBFT is based on a novel abstraction called Consistent Tail Broadcast, which we use to prevent equivocation while bounding memory. We implement uBFT using RDMA-based disaggregated memory and obtain an end-to-end latency of as little as 10 us. This is at least 50× faster than MinBFT, a state-of-the-art 2f+1 BFT SMR based on Intel’s SGX. We use uBFT to replicate two KV-stores (Memcached and Redis), as well as a financial order matching engine (Liquibook). These applications have low latency (up to 20 us) and become Byzantine tolerant with as little as 10 us more. The price for uBFT is a small amount of reliable disaggregated memory (less than 1 MiB), which in our prototype consists of a small number of memory servers connected through RDMA and replicated for fault tolerance. @InProceedings{ASPLOS23p862, author = {Marcos K. Aguilera and Naama Ben-David and Rachid Guerraoui and Antoine Murat and Athanasios Xygkis and Igor Zablotchi}, title = {uBFT: Microsecond-Scale BFT using Disaggregated Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {862--877}, doi = {10.1145/3575693.3575732}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Ainsworth, Sam	ASPLOS '23: "Hacky Racers: Exploiting Instruction-Level ..." Hacky Racers: Exploiting Instruction-Level Parallelism to Generate Stealthy Fine-Grained Timers Haocheng Xiao and Sam Ainsworth (University of Edinburgh, UK) Side-channel attacks pose serious threats to many security models, especially sandbox-based browsers. While transient-execution side channels in out-of-order processors have previously been blamed for vulnerabilities such as Spectre and Meltdown, we show that in fact, the capability of out-of-order execution itself to cause mayhem is far more general. We develop Hacky Racers, a new type of timing gadget that uses instruction-level parallelism, another key feature of out-of-order execution, to measure arbitrary fine-grained timing differences, even in the presence of highly restricted JavaScript sandbox environments. While such environments try to mitigate timing side channels by reducing timer precision and removing language features such as SharedArrayBuffer that can be used to indirectly generate timers via thread-level parallelism, no such restrictions can be designed to limit Hacky Racers. We also design versions of Hacky Racers that require no misspeculation whatsoever, demonstrating that transient execution is not the only threat to security from modern microarchitectural performance optimization. We use Hacky Racers to construct novel backwards-in-time Spectre gadgets, which break many hardware countermeasures in the literature by leaking secrets before misspeculation is discovered. We also use them to generate the first known last-level cache eviction set generator in JavaScript that does not require SharedArrayBuffer support. @InProceedings{ASPLOS23p354, author = {Haocheng Xiao and Sam Ainsworth}, title = {Hacky Racers: Exploiting Instruction-Level Parallelism to Generate Stealthy Fine-Grained Timers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {354--369}, doi = {10.1145/3575693.3575700}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Akella, Aditya	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Alavisamani, Narges	ASPLOS '23: "FrozenQubits: Boosting Fidelity ..." FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes Ramin Ayanzadeh, Narges Alavisamani, Poulami Das, and Moinuddin Qureshi (Georgia Institute of Technology, USA) Quantum Approximate Optimization Algorithm (QAOA) is one of the leading candidates for demonstrating the quantum advantage using near-term quantum computers. Unfortunately, high device error rates limit us from reliably running QAOA circuits for problems with more than a few qubits. In QAOA, the problem graph is translated into a quantum circuit such that every edge corresponds to two 2-qubit CNOT operations in each layer of the circuit. As CNOTs are extremely error-prone, the fidelity of QAOA circuits is dictated by the number of edges in the problem graph. We observe that the majority of graphs corresponding to real-world applications follow a “power-law” distribution, where some hotspot nodes have significantly higher number of connections. We leverage this insight and propose “FrozenQubits” that freezes the hotspot nodes or qubits and intelligently partitions the state-space of the given problem into several smaller sub-spaces, which are then solved independently. The corresponding QAOA sub-circuits are significantly less vulnerable to gate and decoherence errors due to the reduced number of CNOT operations in each sub-circuit. Unlike prior circuit-cutting approaches, FrozenQubits does not require any exponentially complex postprocessing step. Our evaluations with 5,300 QAOA circuits on eight different quantum computers from IBM show that FrozenQubits can improve the quality of solutions by 8.73x on average (and by up to 57x), albeit while utilizing 2x more quantum resources. @InProceedings{ASPLOS23p311, author = {Ramin Ayanzadeh and Narges Alavisamani and Poulami Das and Moinuddin Qureshi}, title = {FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {311--324}, doi = {10.1145/3575693.3575741}, year = {2023}, } Publisher's Version Info
	Amarasinghe, Saman	ASPLOS '23: "WACO: Learning Workload-Aware ..." WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program Jaeyeon Won, Charith Mendis, Joel S. Emer, and Saman Amarasinghe (Massachusetts Institute of Technology, USA; University of Illinois at Urbana-Champaign, USA; NVIDIA, USA) In this paper, we present WACO, a novel method of co-optimizing the format and the schedule of a given sparsity pattern in a sparse tensor program. A core challenge in this paper is the design of a lightweight cost model that accurately predicts the runtime of a sparse tensor program by considering the sparsity pattern, the format, and the schedule. The key idea in addressing this is exploiting a sparse convolutional network to learn meaningful features of the sparsity pattern and embedding a coupled behavior between the format and the schedule using a specially designed schedule template. In addition, within the enormous search space of co-optimization, our novel search strategy, an approximate nearest neighbor search, efficiently and accurately retrieves the best format and schedule for a given sparsity pattern. We evaluated WACO for four different algorithms (SpMV, SpMM, SDDMM, and MTTKRP) on a CPU using 726 different sparsity patterns. Our experimental results showed that WACO outperformed four state-of-the-art baselines, Intel MKL, BestFormat, TACO with a default schedule, and ASpT. Compared to the best of four baselines, WACO achieved 1.43×, 1.18×, 1.14×, and 1.27× average speedups on SpMV, SpMM, SDDMM, and MTTKRP, respectively. @InProceedings{ASPLOS23p920, author = {Jaeyeon Won and Charith Mendis and Joel S. Emer and Saman Amarasinghe}, title = {WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {920--934}, doi = {10.1145/3575693.3575742}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Amit, Nadav	ASPLOS '23: "Copy-on-Pin: The Missing Piece ..." Copy-on-Pin: The Missing Piece for Correct Copy-on-Write David Hildenbrand, Martin Schulz, and Nadav Amit (TU Munich, Germany; Red Hat, Germany; VMware, USA) Operating systems utilize Copy-on-Write (COW) to conserve memory and improve performance. During the last two decades, a series of COW-related bugs - which compromised security, corrupted memory and degraded performance - was found. The majority of these bugs are related to page "pinning", which operating systems employ to access process memory efficiently and to perform direct I/O. Unfortunately, the true cause of these bugs is not well understood, resulting in incomplete bug fixes. We show this by: (1) surveying previously reported pinning-related COW bugs; (2) uncovering new such bugs in Linux, FreeBSD, and NetBSD; and (3) showing that they occur because the COW logic does not consider page pinnings correctly, resulting in incorrect behavior (e.g., I/O of stale data). We then address the underlying problem by deriving when/how shared pages must be copied and under which conditions pinned pages can be shared to maintain correctness. Based on this assessment, we introduce the "Copy-on-Pin (COP)" scheme, an extension of the COW mechanism that handles pinned pages correctly by ensuring pinned pages and shared pages are mutually exclusive. However, we find that a naive implementation of this scheme hampers performance and increases complexity if pages are copied only when strictly necessary. To compensate, we introduce a relaxed-COP design, which does not require precise tracking of page sharing, maintains correctness without increasing complexity, and (while potentially needlessly copying pages in some corner cases) marginally improves performance. Our relaxed-COP solution has been integrated into Linux 5.19. @InProceedings{ASPLOS23p176, author = {David Hildenbrand and Martin Schulz and Nadav Amit}, title = {Copy-on-Pin: The Missing Piece for Correct Copy-on-Write}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {176--191}, doi = {10.1145/3575693.3575716}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Amvrosiadis, George	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Andersen, David G.	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Arora, Nikhil	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Assaiante, Cristian	ASPLOS '23: "Where Did My Variable Go? ..." Where Did My Variable Go? Poking Holes in Incomplete Debug Information Cristian Assaiante, Daniele Cono D'Elia, Giuseppe Antonio Di Luna, and Leonardo Querzoni (Sapienza University of Rome, Italy) The availability of debug information for optimized executables can largely ease crucial tasks such as crash analysis. Source-level debuggers use this information to display program state in terms of source code, allowing users to reason on it even when optimizations alter program structure extensively. A few recent endeavors have proposed effective methodologies for identifying incorrect instances of debug information, which can mislead users by presenting them with an inconsistent program state. In this work, we identify and study a related important problem: the completeness of debug information. Unlike correctness issues for which an unoptimized executable can serve as reference, we find there is no analogous oracle to deem when the cause behind an unreported part of program state is an unavoidable effect of optimization or a compiler implementation defect. In this scenario, we argue that empirically derived conjectures on the expected availability of debug information can serve as an effective means to expose classes of these defects. We propose three conjectures involving variable values and study how often synthetic programs compiled with different configurations of the popular gcc and LLVM compilers deviate from them. We then discuss techniques to pinpoint the optimizations behind such violations and minimize bug reports accordingly. Our experiments revealed, among others, 24 bugs already confirmed by the developers of the gcc-gdb and clang-lldb ecosystems. @InProceedings{ASPLOS23p935, author = {Cristian Assaiante and Daniele Cono D'Elia and Giuseppe Antonio Di Luna and Leonardo Querzoni}, title = {Where Did My Variable Go? Poking Holes in Incomplete Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {935--947}, doi = {10.1145/3575693.3575720}, year = {2023}, } Publisher's Version
	Ayanzadeh, Ramin	ASPLOS '23: "FrozenQubits: Boosting Fidelity ..." FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes Ramin Ayanzadeh, Narges Alavisamani, Poulami Das, and Moinuddin Qureshi (Georgia Institute of Technology, USA) Quantum Approximate Optimization Algorithm (QAOA) is one of the leading candidates for demonstrating the quantum advantage using near-term quantum computers. Unfortunately, high device error rates limit us from reliably running QAOA circuits for problems with more than a few qubits. In QAOA, the problem graph is translated into a quantum circuit such that every edge corresponds to two 2-qubit CNOT operations in each layer of the circuit. As CNOTs are extremely error-prone, the fidelity of QAOA circuits is dictated by the number of edges in the problem graph. We observe that the majority of graphs corresponding to real-world applications follow a “power-law” distribution, where some hotspot nodes have significantly higher number of connections. We leverage this insight and propose “FrozenQubits” that freezes the hotspot nodes or qubits and intelligently partitions the state-space of the given problem into several smaller sub-spaces, which are then solved independently. The corresponding QAOA sub-circuits are significantly less vulnerable to gate and decoherence errors due to the reduced number of CNOT operations in each sub-circuit. Unlike prior circuit-cutting approaches, FrozenQubits does not require any exponentially complex postprocessing step. Our evaluations with 5,300 QAOA circuits on eight different quantum computers from IBM show that FrozenQubits can improve the quality of solutions by 8.73x on average (and by up to 57x), albeit while utilizing 2x more quantum resources. @InProceedings{ASPLOS23p311, author = {Ramin Ayanzadeh and Narges Alavisamani and Poulami Das and Moinuddin Qureshi}, title = {FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {311--324}, doi = {10.1145/3575693.3575741}, year = {2023}, } Publisher's Version Info
	Baker, Alan	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Baker, Jonathan M.	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version ASPLOS '23: "Qompress: Efficient Compilation ..." Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction Andrew Litteken, Lennart Maximilian Seifert, Jason Chadwick, Natalia Nottingham, Frederic T. Chong, and Jonathan M. Baker (University of Chicago, USA) Quantum computing is in an era of limited resources. Current hardware lacks high fidelity gates, long coherence times, and the number of computational units required to perform meaningful computation. Contemporary quantum devices typically use a binary system, where each qubit exists in a superposition of the 0 and 1 states. However, it is often possible to access the 2 or even 3 states in the same physical unit by manipulating the system in different ways. In this work, we consider automatically encoding two qubits into one four-state ququart via a compression scheme. We use quantum optimal control to design efficient proof-of-concept gates that fully replicate standard qubit computation on these encoded qubits. We extend qubit compilation schemes to efficiently route qubits on an arbitrary mixed-radix system consisting of both qubits and ququarts, reducing communication and minimizing excess circuit execution time introduced by longer-duration ququart gates. In conjunction with these compilation strategies, we introduce several methods to find beneficial compressions, reducing circuit error due to computation and communication by up to 50 @InProceedings{ASPLOS23p646, author = {Andrew Litteken and Lennart Maximilian Seifert and Jason Chadwick and Natalia Nottingham and Frederic T. Chong and Jonathan M. Baker}, title = {Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {646--659}, doi = {10.1145/3575693.3575726}, year = {2023}, } Publisher's Version ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Balkind, Jonathan	ASPLOS '23: "A Prediction System Service ..." A Prediction System Service Zhizhou Zhang, Alvin Oliver Glova, Timothy Sherwood, and Jonathan Balkind (University of California at Santa Barbara, USA) To better facilitate application performance programming we propose a software optimization strategy enabled by a novel low-latency Prediction System Service (PSS). Rather than relying on nuanced domain-specific knowledge or slapdash heuristics, a system service for prediction encourages programmers to spend their time uncovering new levers for optimization rather than worrying about the details of their control. The core idea is to write optimizations that improve performance in specific cases, or under specific tunings, and leave the decision of how and when exactly to apply those optimizations to the system to learn through feedback-directed learning. Such a prediction service can be implemented in any number of ways, including as a shared library that can be easily reused by software written in different programming languages, and opens the door to both new software optimization patterns and hardware design possibilities. As a demonstration of the utility of this approach, we show that three very different application-targeted optimization scenarios can each benefit from even a very straightforward perceptron-based implementation of the PSS as long as the service latency can be held low. First, we show that PSS can be used to more intelligently guide hardware lock elision with resulting speedups over a baseline implementation by 34% on average. Second, we show that a PSS can find good configuration parameters for PyPy’s Just-In-Time (JIT) compiler resulting in 15% speedup on average. Last, we show PSS can guide the page reclamation task within a kernel memory management subsystem to reduce the average memory latency by 33% on average. In all three cases, this new optimization pattern with service support is able to meet or beat the best-known hand-crafted methods with a fraction of the complexity. @InProceedings{ASPLOS23p48, author = {Zhizhou Zhang and Alvin Oliver Glova and Timothy Sherwood and Jonathan Balkind}, title = {A Prediction System Service}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {48--60}, doi = {10.1145/3575693.3575714}, year = {2023}, } Publisher's Version
	Bashir, Noman	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Basu, Arkaprava	ASPLOS '23: "Scoped Buffered Persistency ..." Scoped Buffered Persistency Model for GPUs Shweta Pandey, Aditya K Kamath, and Arkaprava Basu (IISc Bangalore, India; University of Washington, USA) While the implications of persistent memory (PM) on CPU hardware and software are well-explored, the same is not true for GPUs (Graphics Processing Units). A recent work, GPM, demonstrated how GPU programs can benefit from the fine-grain persistence of PM. However, in the absence of a persistency model, one cannot reason about the correctness of PM-aware GPU programs. Persistency models define the order in which writes to PM are persisted. We explore persistency models for GPUs. We explore persistency models for GPUs. We demonstrate that CPU persistency models fall short for GPUs. We qualitatively and quantitatively argue that GPU persistency models should support scopes and buffering of writes to PM to leverage parallelism while adapting to higher NVM latencies. We formally specify a GPU persistency model that supports both scopes and buffers. We detail how GPU architecture can efficiently realize such a model. Finally, we quantitatively demonstrate the usefulness of scopes and buffers for PM-aware GPU programs. @InProceedings{ASPLOS23p688, author = {Shweta Pandey and Aditya K Kamath and Arkaprava Basu}, title = {Scoped Buffered Persistency Model for GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {688--701}, doi = {10.1145/3575693.3575749}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Beck, Martin	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Behrens, Diogo	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Ben-David, Naama	ASPLOS '23: "uBFT: Microsecond-Scale BFT ..." uBFT: Microsecond-Scale BFT using Disaggregated Memory Marcos K. Aguilera, Naama Ben-David, Rachid Guerraoui, Antoine Murat, Athanasios Xygkis, and Igor Zablotchi (VMware Research, USA; EPFL, Switzerland; Massachusetts Institute of Technology, USA) We propose uBFT, the first State Machine Replication (SMR) system to achieve microsecond-scale latency in data centers, while using only 2f+1 replicas to tolerate f Byzantine failures. The Byzantine Fault Tolerance (BFT) provided by uBFT is essential as pure crashes appear to be a mere illusion with real-life systems reportedly failing in many unexpected ways. uBFT relies on a small non-tailored trusted computing base—disaggregated memory—and consumes a practically bounded amount of memory. uBFT is based on a novel abstraction called Consistent Tail Broadcast, which we use to prevent equivocation while bounding memory. We implement uBFT using RDMA-based disaggregated memory and obtain an end-to-end latency of as little as 10 us. This is at least 50× faster than MinBFT, a state-of-the-art 2f+1 BFT SMR based on Intel’s SGX. We use uBFT to replicate two KV-stores (Memcached and Redis), as well as a financial order matching engine (Liquibook). These applications have low latency (up to 20 us) and become Byzantine tolerant with as little as 10 us more. The price for uBFT is a small amount of reliable disaggregated memory (less than 1 MiB), which in our prototype consists of a small number of memory servers connected through RDMA and replicated for fault tolerance. @InProceedings{ASPLOS23p862, author = {Marcos K. Aguilera and Naama Ben-David and Rachid Guerraoui and Antoine Murat and Athanasios Xygkis and Igor Zablotchi}, title = {uBFT: Microsecond-Scale BFT using Disaggregated Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {862--877}, doi = {10.1145/3575693.3575732}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Bent, John	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version
	Berger, Daniel S.	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Berlstein, Griffin	ASPLOS '23: "Stepwise Debugging for Hardware ..." Stepwise Debugging for Hardware Accelerators Griffin Berlstein, Rachit Nigam, Christophe Gyurgyik, and Adrian Sampson (Cornell University, USA) High-level programming models for hardware design let domain experts quickly produce specialized accelerators. However, tools for debugging these accelerators remain tied to low-level hardware description languages (HDLs). High-level descriptions contain control-flow information that is lost in HDL code. We describe Cider, a stepwise debugger that exploits this information to provide software-like debugging abstractions for languages that compile to hardware. Cider uses Calyx, an intermediate language for accelerator generators that preserves control information. Cider provides breakpoints, watchpoints, state inspection, and source-level position mapping. Using case studies that examine one new and two preexisting accelerator generators, we demonstrate how Cider helps find and localize previously unreported bugs. By directly simulating a control-rich representation, Cider avoids wasting effort on inactive parts of the design and, despite being largely unoptimized, performs competitively with open-source HDL simulators. @InProceedings{ASPLOS23p778, author = {Griffin Berlstein and Rachit Nigam and Christophe Gyurgyik and Adrian Sampson}, title = {Stepwise Debugging for Hardware Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {778--790}, doi = {10.1145/3575693.3575717}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Beschastnikh, Ivan	ASPLOS '23: "Compiling Distributed System ..." Compiling Distributed System Models with PGo Finn Hackett, Shayan Hosseini, Renato Costa, Matthew Do, and Ivan Beschastnikh (University of British Columbia, Canada) Distributed systems are difficult to design and implement correctly. In response, both research and industry are exploring applications of formal methods to distributed systems. A key challenge in this domain is the missing link between the formal design of a system and its implementation. Today, practitioners bridge this link through manual effort. We present a language called Modular PlusCal that extends PlusCal by cleanly separating the model of a system from a model of its environment. We then present a compiler tool-chain called PGo that automatically translates MPCal models to TLA+ for model checking, and that also compiles MPCal models to runnable Go code. PGo provides system designers with a new ability to model and check their designs, and then re-use their modeling efforts to mechanically extract runnable implementations of their designs. Our evaluation shows that the PGo approach works for complex models: we model check, compile, and evaluate the performance of MPCal systems based on Raft, CRDTs, and primary-backup. Compared to previous work, PGo requires less time to develop a checked model and derive a fully working implementation. With PGo we created a formally checked Raft model and its corresponding implementation in under 1 person-month, which is 3× less time than Ivy. Our evaluation shows that a PGo-based Raft KV store with three nodes has 41% higher throughput than a Raft KV store based on Ivy, the highest performing verified Raft-based KV store from related work. A PGo-based CRDT set has a latency within 2 × of a CRDT set implementation from SoundCloud called Roshi. @InProceedings{ASPLOS23p159, author = {Finn Hackett and Shayan Hosseini and Renato Costa and Matthew Do and Ivan Beschastnikh}, title = {Compiling Distributed System Models with PGo}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {159--175}, doi = {10.1145/3575693.3575695}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Bhat, Koustubha	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Bianchini, Ricardo	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Bjørling, Matias	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Brooks, David	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Carlson, Trevor E.	ASPLOS '23: "AfterImage: Leaking Control ..." AfterImage: Leaking Control Flow Data and Tracking Load Operations via the Hardware Prefetcher Yun Chen, Lingfeng Pei, and Trevor E. Carlson (National University of Singapore, Singapore) Research into processor-based side-channels has seen both a large number and a large variety of disclosed vulnerabilities that can leak critical, private data to malicious attackers. While most previous works require speculative execution and the use of cache primitives to transmit data, our new approach, called AfterImage, requires neither, capitalizing on vulnerabilities in Intel’s IP-stride prefetcher to both expose and transmit victim data. By training this prefetcher with attacker-known values, and watching for changes to the prefetcher state when execution returns to the attacker, it is now possible to monitor and leak critical data from a large number of common userspace applications and kernel routines without speculation and additional cache accesses. To demonstrate the novel capabilities of AfterImage, we (1) present proof-of-concept attacks that leak data across different isolation levels, (2) present an end-to-end attack that leaks an entire RSA key from a modern, timing-balanced algorithm, and also (3) show how AfterImage can significantly improve the effectiveness of other attacks, such as power side-channel attacks, by using this technique as a high-precision marker. In addition to an extensive evaluation of these and other cache-based attacks, we also present a full reverse-engineering of the Intel IP-stride prefetcher which was required to enable AfterImage, and describe how AfterImage can be used as a covert channel. Finally, we present several mitigation techniques that can be used to block this side-channel on machines today. Taken together, this work explores a full set of techniques to utilize the prefetcher to leak previously protected information between different protection domains (SGX, kernel and other user spaces) and across many important applications, including security and non-security-related workloads. @InProceedings{ASPLOS23p16, author = {Yun Chen and Lingfeng Pei and Trevor E. Carlson}, title = {AfterImage: Leaking Control Flow Data and Tracking Load Operations via the Hardware Prefetcher}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {16--32}, doi = {10.1145/3575693.3575719}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Chadwick, Jason	ASPLOS '23: "Qompress: Efficient Compilation ..." Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction Andrew Litteken, Lennart Maximilian Seifert, Jason Chadwick, Natalia Nottingham, Frederic T. Chong, and Jonathan M. Baker (University of Chicago, USA) Quantum computing is in an era of limited resources. Current hardware lacks high fidelity gates, long coherence times, and the number of computational units required to perform meaningful computation. Contemporary quantum devices typically use a binary system, where each qubit exists in a superposition of the 0 and 1 states. However, it is often possible to access the 2 or even 3 states in the same physical unit by manipulating the system in different ways. In this work, we consider automatically encoding two qubits into one four-state ququart via a compression scheme. We use quantum optimal control to design efficient proof-of-concept gates that fully replicate standard qubit computation on these encoded qubits. We extend qubit compilation schemes to efficiently route qubits on an arbitrary mixed-radix system consisting of both qubits and ququarts, reducing communication and minimizing excess circuit execution time introduced by longer-duration ququart gates. In conjunction with these compilation strategies, we introduce several methods to find beneficial compressions, reducing circuit error due to computation and communication by up to 50 @InProceedings{ASPLOS23p646, author = {Andrew Litteken and Lennart Maximilian Seifert and Jason Chadwick and Natalia Nottingham and Frederic T. Chong and Jonathan M. Baker}, title = {Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {646--659}, doi = {10.1145/3575693.3575726}, year = {2023}, } Publisher's Version
	Chakkaravarthy, Manoj	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Chakraborty, Soham	ASPLOS '23: "Probabilistic Concurrency ..." Probabilistic Concurrency Testing for Weak Memory Programs Mingyu Gao, Soham Chakraborty, and Burcu Kulahcioglu Ozkan (Delft University of Technology, Netherlands) The Probabilistic Concurrency Testing (PCT) algorithm that provides theoretical guarantees on the probability of detecting concurrency bugs does not apply to weak memory programs. The PCT algorithm builds on the interleaving semantics of sequential consistency, which does not hold for weak memory concurrency. It is because weak memory concurrency allows additional behaviors that cannot be produced by any interleaving execution. In this paper, we generalize PCT to address weak memory concurrency and present Probabilistic Concurrency Testing for Weak Memory (PCTWM). We empirically evaluate PCTWM on a set of well-known weak memory program benchmarks in comparison to the state-of-the-art weak memory testing tool C11Tester. Our results show that PCTWM can detect concurrency bugs more frequently than C11Tester. @InProceedings{ASPLOS23p603, author = {Mingyu Gao and Soham Chakraborty and Burcu Kulahcioglu Ozkan}, title = {Probabilistic Concurrency Testing for Weak Memory Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {603--616}, doi = {10.1145/3575693.3575729}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Chang, Rui	ASPLOS '23: "VDom: Fast and Unlimited Virtual ..." VDom: Fast and Unlimited Virtual Domains on Multiple Architectures Ziqi Yuan, Siyu Hong, Rui Chang, Yajin Zhou, Wenbo Shen, and Kui Ren (Zhejiang University, China) Hardware memory domain primitives, such as Intel MPK and ARM Memory Domain, have been used for efficient in-process memory isolation. However, they can only provide a limited number of memory domains (16 domains), which cannot satisfy the compelling need for more isolated domains inside the address space of a process. Existing solutions to virtualize memory domains are either intrusive (need the modification to existing hardware), or incur a large performance overhead. In this paper, we propose VDom, a fast and scalable memory domain virtualization system that supports unlimited memory domains. VDom leverages separate address spaces to provide an unlimited number of virtual domains, and optimizes related memory management operations. To map virtual domains to hardware domains, we design a domain virtualization algorithm, which manages address spaces and domain maps for threads to efficiently access other domains that are unmapped in the current address space. According to our evaluation on real Intel and ARM platforms, on real-world server applications (httpd and MySQL), VDom incurs less than 2.65% performance overhead, which is lower than the overheads of the state-of-the-art software approaches (libmpk and EPK). In random domain access tests, VDom is comparable to EPK and has significantly higher efficiency than libmpk. @InProceedings{ASPLOS23p905, author = {Ziqi Yuan and Siyu Hong and Rui Chang and Yajin Zhou and Wenbo Shen and Kui Ren}, title = {VDom: Fast and Unlimited Virtual Domains on Multiple Architectures}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {905--919}, doi = {10.1145/3575693.3575735}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Chen, Geng	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Chen, Haibo	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version ASPLOS '23: "BeeHive: Sub-second Elasticity ..." BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution Ziming Zhao, Mingyu Wu, Jiawei Tang, Binyu Zang, Zhaoguo Wang, and Haibo Chen (Shanghai Jiao Tong University, China; Shanghai AI Laboratory, China; Engineering Research Center for Domain-specific Operating Systems of the Ministry of Education of China, China) Function-as-a-service (FaaS), an emerging cloud computing paradigm, is expected to provide strong elasticity due to its promise to auto-scale fine-grained functions rapidly. Although appealing for applications with good parallelism and dynamic workload, this paper shows that it is non-trivial to adapt existing monolithic applications (like web services) to FaaS due to their complexity. To bridge the gap between complicated web services and FaaS, this paper proposes a runtime-based Semi-FaaS execution model, which dynamically extracts time-consuming code snippets (closures) from applications and offloads them to FaaS platforms for execution. It further proposes BeeHive, an offloading framework for Semi-FaaS, which relies on the managed runtime to provide a fallback-based execution model and addresses the performance issues in traditional offloading mechanisms for FaaS. Meanwhile, the runtime system of BeeHive selects offloading candidates in a user-transparent way and supports efficient object sharing, memory management, and failure recovery in a distributed environment. The evaluation using various web applications suggests that the Semi-FaaS execution supported by BeeHive can reach sub-second resource provisioning on commercialized FaaS platforms like AWS Lambda, which is up to two orders of magnitude better than other alternative scaling approaches in cloud computing. @InProceedings{ASPLOS23p74, author = {Ziming Zhao and Mingyu Wu and Jiawei Tang and Binyu Zang and Zhaoguo Wang and Haibo Chen}, title = {BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {74--87}, doi = {10.1145/3575693.3575752}, year = {2023}, } Publisher's Version
	Chen, Hongtao	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version
	Chen, Jingji	ASPLOS '23: "Khuzdul: Efficient and Scalable ..." Khuzdul: Efficient and Scalable Distributed Graph Pattern Mining Engine Jingji Chen and Xuehai Qian (Purdue University, USA) This paper proposes Khuzdul, a distributed execution engine with a well-defined abstraction that can be integrated with existing single-machine graph pattern mining (GPM) systems to provide efficiency and scalability at the same time. The key novelty is the extendable embedding abstraction which can express pattern enumeration algorithms, allow fine-grained task scheduling, and enable low-cost GPM-specific data reuse to reduce communication cost. The effective BFS-DFS hybrid exploration generates sufficient concurrent tasks for communication-computation overlapping with bounded memory consumption. Two scalable distributed GPM systems are implemented by porting Automine and GraphPi on Khuzdul. Our evaluation shows that Khuzdul based systems significantly outperform state-of-the-art distributed GPM systems with partitioned graphs by up to 75.5× (on average 19.0×), achieve similar or even better performance compared with the fastest distributed GPM systems with replicated graph, and scale to massive graphs with more than one hundred billion edges with a commodity cluster. @InProceedings{ASPLOS23p413, author = {Jingji Chen and Xuehai Qian}, title = {Khuzdul: Efficient and Scalable Distributed Graph Pattern Mining Engine}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {413--426}, doi = {10.1145/3575693.3575743}, year = {2023}, } Publisher's Version
	Chen, Kang	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version
	Chen, Taijie	ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Chen, Tianqi	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Chen, Yun	ASPLOS '23: "AfterImage: Leaking Control ..." AfterImage: Leaking Control Flow Data and Tracking Load Operations via the Hardware Prefetcher Yun Chen, Lingfeng Pei, and Trevor E. Carlson (National University of Singapore, Singapore) Research into processor-based side-channels has seen both a large number and a large variety of disclosed vulnerabilities that can leak critical, private data to malicious attackers. While most previous works require speculative execution and the use of cache primitives to transmit data, our new approach, called AfterImage, requires neither, capitalizing on vulnerabilities in Intel’s IP-stride prefetcher to both expose and transmit victim data. By training this prefetcher with attacker-known values, and watching for changes to the prefetcher state when execution returns to the attacker, it is now possible to monitor and leak critical data from a large number of common userspace applications and kernel routines without speculation and additional cache accesses. To demonstrate the novel capabilities of AfterImage, we (1) present proof-of-concept attacks that leak data across different isolation levels, (2) present an end-to-end attack that leaks an entire RSA key from a modern, timing-balanced algorithm, and also (3) show how AfterImage can significantly improve the effectiveness of other attacks, such as power side-channel attacks, by using this technique as a high-precision marker. In addition to an extensive evaluation of these and other cache-based attacks, we also present a full reverse-engineering of the Intel IP-stride prefetcher which was required to enable AfterImage, and describe how AfterImage can be used as a covert channel. Finally, we present several mitigation techniques that can be used to block this side-channel on machines today. Taken together, this work explores a full set of techniques to utilize the prefetcher to leak previously protected information between different protection domains (SGX, kernel and other user spaces) and across many important applications, including security and non-security-related workloads. @InProceedings{ASPLOS23p16, author = {Yun Chen and Lingfeng Pei and Trevor E. Carlson}, title = {AfterImage: Leaking Control Flow Data and Tracking Load Operations via the Hardware Prefetcher}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {16--32}, doi = {10.1145/3575693.3575719}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Cheng, Peng	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Chirkov, Grigory	ASPLOS '23: "SMAPPIC: Scalable Multi-FPGA ..." SMAPPIC: Scalable Multi-FPGA Architecture Prototype Platform in the Cloud Grigory Chirkov and David Wentzlaff (Princeton University, USA) Traditionally, architecture prototypes are built on top of FPGA infrastructure, with two associated problems. First, very large FPGAs are prohibitively expensive for most people and institutions. Second, the burden of FPGA development adds to an already uneasy life of researchers, especially those who focus on software. Large designs that do not fit into a single FPGA exacerbate these issues even more. This work presents SMAPPIC — the first open-source prototype platform for shared memory multi-die architectures on cloud FPGAs. SMAPPIC leverages the OpenPiton/BYOC infrastructure and AWS F1 instances to make FPGA-based prototypes of System-on-Chips, processor cores, accelerators, cache subsystems, etc., cheap, scalable, and straightforward. SMAPPIC enables many use cases that are not possible or significantly more complicated in existing software and FPGA tools. This work has the potential to accelerate the rate of innovation in computer engineering fields in the nearest future. @InProceedings{ASPLOS23p733, author = {Grigory Chirkov and David Wentzlaff}, title = {SMAPPIC: Scalable Multi-FPGA Architecture Prototype Platform in the Cloud}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {733--746}, doi = {10.1145/3575693.3575753}, year = {2023}, } Publisher's Version
	Cho, Mingun	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Choe, Wonkyo	ASPLOS '23: "STI: Turbocharge NLP Inference ..." STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining Liwei Guo, Wonkyo Choe, and Felix Xiaozhu Lin (University of Virginia, USA) Natural Language Processing (NLP) inference is seeing increasing adoption by mobile applications, where on-device inference is desirable for crucially preserving user data privacy and avoiding network roundtrips. Yet, the unprecedented size of an NLP model stresses both latency and memory, creating a tension between the two key resources of a mobile device. To meet a target latency, holding the whole model in memory launches execution as soon as possible but increases one app’s memory footprints by several times, limiting its benefits to only a few inferences before being recycled by mobile memory management. On the other hand, loading the model from storage on demand incurs IO as long as a few seconds, far exceeding the delay range satisfying to a user; pipelining layerwise model loading and execution does not hide IO either, due to the high skewness between IO and computation delays. To this end, we propose Speedy Transformer Inference (STI). Built on the key idea of maximizing IO/compute resource utilization on the most important parts of a model, STI reconciles the latency v.s. memory tension via two novel techniques. First, model sharding. STI manages model parameters as independently tunable shards, and profiles their importance to accuracy. Second, elastic pipeline planning with a preload buffer. STI instantiates an IO/compute pipeline and uses a small buffer for preload shards to bootstrap execution without stalling at early stages; it judiciously selects, tunes, and assembles shards per their importance for resource-elastic execution, maximizing inference accuracy. Atop two commodity SoCs, we build STI and evaluate it against a wide range of NLP tasks, under a practical range of target latencies, and on both CPU and GPU. We demonstrate that STI delivers high accuracies with 1–2 orders of magnitude lower memory, outperforming competitive baselines. @InProceedings{ASPLOS23p791, author = {Liwei Guo and Wonkyo Choe and Felix Xiaozhu Lin}, title = {STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {791--803}, doi = {10.1145/3575693.3575698}, year = {2023}, } Publisher's Version
	Chong, Frederic T.	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version ASPLOS '23: "Qompress: Efficient Compilation ..." Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction Andrew Litteken, Lennart Maximilian Seifert, Jason Chadwick, Natalia Nottingham, Frederic T. Chong, and Jonathan M. Baker (University of Chicago, USA) Quantum computing is in an era of limited resources. Current hardware lacks high fidelity gates, long coherence times, and the number of computational units required to perform meaningful computation. Contemporary quantum devices typically use a binary system, where each qubit exists in a superposition of the 0 and 1 states. However, it is often possible to access the 2 or even 3 states in the same physical unit by manipulating the system in different ways. In this work, we consider automatically encoding two qubits into one four-state ququart via a compression scheme. We use quantum optimal control to design efficient proof-of-concept gates that fully replicate standard qubit computation on these encoded qubits. We extend qubit compilation schemes to efficiently route qubits on an arbitrary mixed-radix system consisting of both qubits and ququarts, reducing communication and minimizing excess circuit execution time introduced by longer-duration ququart gates. In conjunction with these compilation strategies, we introduce several methods to find beneficial compressions, reducing circuit error due to computation and communication by up to 50 @InProceedings{ASPLOS23p646, author = {Andrew Litteken and Lennart Maximilian Seifert and Jason Chadwick and Natalia Nottingham and Frederic T. Chong and Jonathan M. Baker}, title = {Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {646--659}, doi = {10.1145/3575693.3575726}, year = {2023}, } Publisher's Version ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Chu, Xiaomeng	ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Chung, I-Hsin	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Costa, Renato	ASPLOS '23: "Compiling Distributed System ..." Compiling Distributed System Models with PGo Finn Hackett, Shayan Hosseini, Renato Costa, Matthew Do, and Ivan Beschastnikh (University of British Columbia, Canada) Distributed systems are difficult to design and implement correctly. In response, both research and industry are exploring applications of formal methods to distributed systems. A key challenge in this domain is the missing link between the formal design of a system and its implementation. Today, practitioners bridge this link through manual effort. We present a language called Modular PlusCal that extends PlusCal by cleanly separating the model of a system from a model of its environment. We then present a compiler tool-chain called PGo that automatically translates MPCal models to TLA+ for model checking, and that also compiles MPCal models to runnable Go code. PGo provides system designers with a new ability to model and check their designs, and then re-use their modeling efforts to mechanically extract runnable implementations of their designs. Our evaluation shows that the PGo approach works for complex models: we model check, compile, and evaluate the performance of MPCal systems based on Raft, CRDTs, and primary-backup. Compared to previous work, PGo requires less time to develop a checked model and derive a fully working implementation. With PGo we created a formally checked Raft model and its corresponding implementation in under 1 person-month, which is 3× less time than Ivy. Our evaluation shows that a PGo-based Raft KV store with three nodes has 41% higher throughput than a Raft KV store based on Ivy, the highest performing verified Raft-based KV store from related work. A PGo-based CRDT set has a latency within 2 × of a CRDT set implementation from SoundCloud called Roshi. @InProceedings{ASPLOS23p159, author = {Finn Hackett and Shayan Hosseini and Renato Costa and Matthew Do and Ivan Beschastnikh}, title = {Compiling Distributed System Models with PGo}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {159--175}, doi = {10.1145/3575693.3575695}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Cowan, Meghan	ASPLOS '23: "MSCCLang: Microsoft Collective ..." MSCCLang: Microsoft Collective Communication Language Meghan Cowan, Saeed Maleki, Madanlal Musuvathi, Olli Saarikivi, and Yifan Xiong (Microsoft Research, USA; Microsoft Research, China) Machine learning models with millions or billions of parameters are increasingly trained and served on large multi-GPU systems. As models grow in size and execute on more GPUs, collective communication becomes a bottleneck. Custom collective algorithms optimized for both particular network topologies and application-specific communication patterns can alleviate this bottleneck and help these applications scale. However, implementing correct and efficient custom algorithms is challenging. This paper introduces MSCCLang, a system for programmable GPU communication. MSCCLang provides a domain specific language for writing collective communication algorithms and an optimizing compiler for lowering them to an executable form, which can be executed efficiently and flexibly in an interpreter-based runtime. We used MSCCLang to write novel collective algorithms for AllReduce and AllToAll that are up to 1.9× and 1.3× faster than hand-optimized implementations, respectively. @InProceedings{ASPLOS23p502, author = {Meghan Cowan and Saeed Maleki and Madanlal Musuvathi and Olli Saarikivi and Yifan Xiong}, title = {MSCCLang: Microsoft Collective Communication Language}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {502--514}, doi = {10.1145/3575693.3575724}, year = {2023}, } Publisher's Version
	Dally, William	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Das, Poulami	ASPLOS '23: "FrozenQubits: Boosting Fidelity ..." FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes Ramin Ayanzadeh, Narges Alavisamani, Poulami Das, and Moinuddin Qureshi (Georgia Institute of Technology, USA) Quantum Approximate Optimization Algorithm (QAOA) is one of the leading candidates for demonstrating the quantum advantage using near-term quantum computers. Unfortunately, high device error rates limit us from reliably running QAOA circuits for problems with more than a few qubits. In QAOA, the problem graph is translated into a quantum circuit such that every edge corresponds to two 2-qubit CNOT operations in each layer of the circuit. As CNOTs are extremely error-prone, the fidelity of QAOA circuits is dictated by the number of edges in the problem graph. We observe that the majority of graphs corresponding to real-world applications follow a “power-law” distribution, where some hotspot nodes have significantly higher number of connections. We leverage this insight and propose “FrozenQubits” that freezes the hotspot nodes or qubits and intelligently partitions the state-space of the given problem into several smaller sub-spaces, which are then solved independently. The corresponding QAOA sub-circuits are significantly less vulnerable to gate and decoherence errors due to the reduced number of CNOT operations in each sub-circuit. Unlike prior circuit-cutting approaches, FrozenQubits does not require any exponentially complex postprocessing step. Our evaluations with 5,300 QAOA circuits on eight different quantum computers from IBM show that FrozenQubits can improve the quality of solutions by 8.73x on average (and by up to 57x), albeit while utilizing 2x more quantum resources. @InProceedings{ASPLOS23p311, author = {Ramin Ayanzadeh and Narges Alavisamani and Poulami Das and Moinuddin Qureshi}, title = {FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {311--324}, doi = {10.1145/3575693.3575741}, year = {2023}, } Publisher's Version Info
	D'Elia, Daniele Cono	ASPLOS '23: "Where Did My Variable Go? ..." Where Did My Variable Go? Poking Holes in Incomplete Debug Information Cristian Assaiante, Daniele Cono D'Elia, Giuseppe Antonio Di Luna, and Leonardo Querzoni (Sapienza University of Rome, Italy) The availability of debug information for optimized executables can largely ease crucial tasks such as crash analysis. Source-level debuggers use this information to display program state in terms of source code, allowing users to reason on it even when optimizations alter program structure extensively. A few recent endeavors have proposed effective methodologies for identifying incorrect instances of debug information, which can mislead users by presenting them with an inconsistent program state. In this work, we identify and study a related important problem: the completeness of debug information. Unlike correctness issues for which an unoptimized executable can serve as reference, we find there is no analogous oracle to deem when the cause behind an unreported part of program state is an unavoidable effect of optimization or a compiler implementation defect. In this scenario, we argue that empirically derived conjectures on the expected availability of debug information can serve as an effective means to expose classes of these defects. We propose three conjectures involving variable values and study how often synthetic programs compiled with different configurations of the popular gcc and LLVM compilers deviate from them. We then discuss techniques to pinpoint the optimizations behind such violations and minimize bug reports accordingly. Our experiments revealed, among others, 24 bugs already confirmed by the developers of the gcc-gdb and clang-lldb ecosystems. @InProceedings{ASPLOS23p935, author = {Cristian Assaiante and Daniele Cono D'Elia and Giuseppe Antonio Di Luna and Leonardo Querzoni}, title = {Where Did My Variable Go? Poking Holes in Incomplete Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {935--947}, doi = {10.1145/3575693.3575720}, year = {2023}, } Publisher's Version
	Delimitrou, Christina	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Dhanotia, Abhishek	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Di Luna, Giuseppe Antonio	ASPLOS '23: "Where Did My Variable Go? ..." Where Did My Variable Go? Poking Holes in Incomplete Debug Information Cristian Assaiante, Daniele Cono D'Elia, Giuseppe Antonio Di Luna, and Leonardo Querzoni (Sapienza University of Rome, Italy) The availability of debug information for optimized executables can largely ease crucial tasks such as crash analysis. Source-level debuggers use this information to display program state in terms of source code, allowing users to reason on it even when optimizations alter program structure extensively. A few recent endeavors have proposed effective methodologies for identifying incorrect instances of debug information, which can mislead users by presenting them with an inconsistent program state. In this work, we identify and study a related important problem: the completeness of debug information. Unlike correctness issues for which an unoptimized executable can serve as reference, we find there is no analogous oracle to deem when the cause behind an unreported part of program state is an unavoidable effect of optimization or a compiler implementation defect. In this scenario, we argue that empirically derived conjectures on the expected availability of debug information can serve as an effective means to expose classes of these defects. We propose three conjectures involving variable values and study how often synthetic programs compiled with different configurations of the popular gcc and LLVM compilers deviate from them. We then discuss techniques to pinpoint the optimizations behind such violations and minimize bug reports accordingly. Our experiments revealed, among others, 24 bugs already confirmed by the developers of the gcc-gdb and clang-lldb ecosystems. @InProceedings{ASPLOS23p935, author = {Cristian Assaiante and Daniele Cono D'Elia and Giuseppe Antonio Di Luna and Leonardo Querzoni}, title = {Where Did My Variable Go? Poking Holes in Incomplete Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {935--947}, doi = {10.1145/3575693.3575720}, year = {2023}, } Publisher's Version
	Ding, Yaoyao	ASPLOS '23: "Hidet: Task-Mapping Programming ..." Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding, Cody Hao Yu, Bojian Zheng, Yizhi Liu, Yida Wang, and Gennady Pekhimenko (University of Toronto, Canada; Vector Institute, Canada; Amazon Web Services, USA) As deep learning models nowadays are widely adopted by both cloud services and edge devices, reducing the latency of deep learning model inferences becomes crucial to provide efficient model serving. However, it is challenging to develop efficient tensor programs for deep learning operators due to the high complexity of modern accelerators (e.g., NVIDIA GPUs and Google TPUs) and the rapidly growing number of operators. Deep learning compilers, such as Apache TVM, adopt declarative scheduling primitives to lower the bar of developing tensor programs. However, we show that this approach is insufficient to cover state-of-the-art tensor program optimizations (e.g., double buffering). In this paper, we propose to embed the scheduling process into tensor programs and use dedicated mappings, called task mappings, to define the computation assignment and ordering directly in the tensor programs. This new approach greatly enriches the expressible optimizations by allowing developers to manipulate tensor programs at a much finer granularity (e.g., allowing program-statement-level optimizations). We call the proposed method the task-mapping programming paradigm. In addition, we propose a new post-scheduling fusion optimization that allows developers to focus on scheduling every single operator and automates the fusion after scheduling. It greatly reduces the engineering efforts for operator fusion. Our proposed paradigm also constructs an efficient hardware-centric schedule space, which is agnostic to the program input size and greatly reduces the tuning time. With the proposed paradigm, we implement a deep learning compiler Hidet. Extensive experiments on modern convolution and transformer models show that Hidet outperforms state-of-the-art DNN inference framework, ONNX Runtime, and compiler, TVM equipped with scheduler AutoTVM and Ansor, by up to 1.48x (1.22x on average). It also reduces the tuning time by 20x and 11x compared with AutoTVM and Ansor, respectively. We open-sourced hidet at https://www.github.com/hidet-org/hidet. @InProceedings{ASPLOS23p370, author = {Yaoyao Ding and Cody Hao Yu and Bojian Zheng and Yizhi Liu and Yida Wang and Gennady Pekhimenko}, title = {Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {370--384}, doi = {10.1145/3575693.3575702}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Do, Matthew	ASPLOS '23: "Compiling Distributed System ..." Compiling Distributed System Models with PGo Finn Hackett, Shayan Hosseini, Renato Costa, Matthew Do, and Ivan Beschastnikh (University of British Columbia, Canada) Distributed systems are difficult to design and implement correctly. In response, both research and industry are exploring applications of formal methods to distributed systems. A key challenge in this domain is the missing link between the formal design of a system and its implementation. Today, practitioners bridge this link through manual effort. We present a language called Modular PlusCal that extends PlusCal by cleanly separating the model of a system from a model of its environment. We then present a compiler tool-chain called PGo that automatically translates MPCal models to TLA+ for model checking, and that also compiles MPCal models to runnable Go code. PGo provides system designers with a new ability to model and check their designs, and then re-use their modeling efforts to mechanically extract runnable implementations of their designs. Our evaluation shows that the PGo approach works for complex models: we model check, compile, and evaluate the performance of MPCal systems based on Raft, CRDTs, and primary-backup. Compared to previous work, PGo requires less time to develop a checked model and derive a fully working implementation. With PGo we created a formally checked Raft model and its corresponding implementation in under 1 person-month, which is 3× less time than Ivy. Our evaluation shows that a PGo-based Raft KV store with three nodes has 41% higher throughput than a Raft KV store based on Ivy, the highest performing verified Raft-based KV store from related work. A PGo-based CRDT set has a latency within 2 × of a CRDT set implementation from SoundCloud called Roshi. @InProceedings{ASPLOS23p159, author = {Finn Hackett and Shayan Hosseini and Renato Costa and Matthew Do and Ivan Beschastnikh}, title = {Compiling Distributed System Models with PGo}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {159--175}, doi = {10.1145/3575693.3575695}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Dong, Wenqian	ASPLOS '23: "Betty: Enabling Large-Scale ..." Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning Shuangyan Yang, Minjia Zhang, Wenqian Dong, and Dong Li (University of California at Merced, USA; Microsoft Research, USA; Florida International University, USA) The Graph Neural Network (GNN) is showing outstanding results in improving the performance of graph-based applications. Recent studies demonstrate that GNN performance can be boosted via using more advanced aggregators, deeper aggregation depth, larger sampling rate, etc. While leading to promising results, the improvements come at a cost of significantly increased memory footprint, easily exceeding GPU memory capacity. In this paper, we introduce a method, Betty, to make GNN training more scalable and accessible via batch-level partitioning. Different from DNN training, a mini-batch in GNN has complex dependencies between input features and output labels, making batch-level partitioning difficult. Betty introduces two noveltechniques, redundancy-embedded graph (REG) partitioning and memory-aware partitioning, to effectively mitigate the redundancy and load imbalances issues across the partitions. Our evaluation of large-scale real-world datasets shows that Betty can significantly mitigate the memory bottleneck, enabling scalable GNN training with much deeper aggregation depths, larger sampling rate, larger training batch sizes, together with more advanced aggregators, with a few as a single GPU. @InProceedings{ASPLOS23p103, author = {Shuangyan Yang and Minjia Zhang and Wenqian Dong and Dong Li}, title = {Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {103--117}, doi = {10.1145/3575693.3575725}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Dong, Yiwen	ASPLOS '23: "Compilation Consistency Modulo ..." Compilation Consistency Modulo Debug Information Theodore Luo Wang, Yongqiang Tian, Yiwen Dong, Zhenyang Xu, and Chengnian Sun (University of Waterloo, Canada) Compilation Consistency Modulo Debug Information (CCMD) is an essential compiler property that a production compiler should support: the compiler should emit the same machine code regardless of enabling debug information. CCMD is vital to developers’ experiences with debugging a production binary containing no debug information. To debug such a binary, developers need build another binary with the same compiler flags and enable debug information. Without CCMD, the machine code in the latter binary will be different, which can confuse the debugger, hide the bug, or even cause a miscompilation (as GCC once did with the Linux Kernel). This paper is the first to introduce to the research community the validation of CCMD, a new research problem that has been overlooked for decades despite its importance. More importantly, we propose the first testing technique Dfusor to automatically validate CCMD for C compilers. At the high level, given a compilable program P as a seed, Dfusor automatically generates compilable program variants via multiple effective program transformations. Such variants can cause a compiler to emit more debug information than it would when compiling P, thus exercising more code paths in the compiler and increasing the chance to find CCMD bugs. Our extensive evaluations of Dfusor demonstrate that Dfusor can produce variants that exhibit significant increases in the quantity and complexity of the emitted debug information, and thus has found new, real bugs in GCC and LLVM. With a sample of 100 variants derived from distinct seed programs, Dfusor introduces 214% more debug information entries and 36% more distinct debug information entries in the variants than the seeds, and improves the code coverage of GCC and Clang by up to 6.00% and 6.82%. More importantly, Dfusor has found CCMD bugs; within 10 months of development and intermittent testing, Dfusor has found 23 bugs (9 in GCC and 14 in Clang), with 3 confirmed and 18 fixed. @InProceedings{ASPLOS23p146, author = {Theodore Luo Wang and Yongqiang Tian and Yiwen Dong and Zhenyang Xu and Chengnian Sun}, title = {Compilation Consistency Modulo Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {146--158}, doi = {10.1145/3575693.3575740}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Duan, Zhuohui	ASPLOS '23: "Revisiting Log-Structured ..." Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems Zhuohui Duan, Jiabo Yao, Haikun Liu, Xiaofei Liao, Hai Jin, and Yu Zhang (Huazhong University of Science and Technology, China) We present MioDB, a novel LSM-tree based key-value (KV) store system designed to fully exploit the advantages of byte-addressable non-volatile memories (NVMs). Our experimental studies reveal that the performance bottleneck of LSM-tree based KV stores using NVMs mainly stems from (1) costly data serialization/deserialization across memory and storage, and (2) unbalanced speed between memory-to-disk data flushing and on-disk data compaction. They may cause unpredictable performance degradation due to write stalls and write amplification. To address these problems, we advocate byte-addressable and persistent skip lists to replace the on-disk data structure of LSM-tree, and design four novel techniques to make the best use of fast NVMs. First, we propose one-piece flushing to minimize the cost of data serialization from DRAM to NVM. Second, we exploit an elastic NVM buffer with multiple levels and zero-copy compaction to eliminate write stalls and reduce write amplification. Third, we propose parallel compaction to orchestrate data flushing and compactions across all levels of LSM-trees. Finally, MioDB increases the depth of LSM-tree and exploits bloom filters to improve the read performance. Our extensive experimental studies demonstrate that MioDB achieves 17.1× and 21.7× lower 99.9th percentile latency, 8.3× and 2.5× higher random write throughput, and up to 5× and 4.9× lower write amplification compared with the state-of-the-art NoveLSM and MatrixKV, respectively. @InProceedings{ASPLOS23p674, author = {Zhuohui Duan and Jiabo Yao and Haikun Liu and Xiaofei Liao and Hai Jin and Yu Zhang}, title = {Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {674--687}, doi = {10.1145/3575693.3575715}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Earnest, Nathan	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version
	Eeckhout, Lieven	ASPLOS '23: "NUBA: Non-Uniform Bandwidth ..." NUBA: Non-Uniform Bandwidth GPUs Xia Zhao, Magnus Jahre, Yuhua Tang, Guangda Zhang, and Lieven Eeckhout (Academy of Military Sciences, China; NTNU, Norway; National University of Defense Technology, China; Ghent University, Belgium) The parallel execution model of GPUs enables scaling to hundreds of thousands of threads, which is a key capability that many modern high-performance applications exploit. GPU vendors are hence increasing the compute and memory resources with every GPU generation — resulting in the need to efficiently stitch together a plethora of Symmetric Multiprocessors (SMs), Last-Level Cache (LLC) slices and memory controllers while maximizing bandwidth and keeping power consumption and design complexity in check. Conventional GPUs are Uniform Bandwidth Architectures (UBAs) as they provide equal bandwidth between all SMs and all LLC slices. UBA GPUs require a uniform high-bandwidth Network-on-Chip (NoC), and our key observation is that provisioning a NoC to match the LLC slice bandwidth incurs a hefty power and complexity overhead. We propose the Non-Uniform Bandwidth Architecture (NUBA), a GPU system architecture aimed at fully utilizing LLC slice bandwidth. A NUBA GPU consists of partitions that each feature a few SMs and LLC slices as well as a memory controller — hence exposing the complete LLC bandwidth to the SMs within a partition since they can be connected with point-to-point links — and a NoC between partitions — to enable access to remote data.Exploiting the potential of NUBA GPUs however requires carefully co-designing system software, the compiler and architectural policies. The critical system software component is our Local-And-Balanced (LAB) page placement policy which enables the GPU driver to place data in local partitions while avoiding load imbalance. Moreover, we propose Model-Driven Replication (MDR) which identifies read-only shared data with data-flow analysis at compile time. At run time, MDR leverages an architectural mechanism that replicates read-only shared data across LLC slices when this can be done without pressuring cache capacity. With LAB and MDR, our NUBA GPU improves average performance by 23.1% and 22.2% (and up to 183.9% and 182.4%) compared to iso-resource memory-side and SM-side UBA GPUs, respectively. When the NUBA concept is leveraged to reduce overhead while maintaining similar performance, NUBA reduces NoC power consumption by 12.1× and 9.4×, respectively. @InProceedings{ASPLOS23p544, author = {Xia Zhao and Magnus Jahre and Yuhua Tang and Guangda Zhang and Lieven Eeckhout}, title = {NUBA: Non-Uniform Bandwidth GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {544--559}, doi = {10.1145/3575693.3575745}, year = {2023}, } Publisher's Version
	Emer, Joel S.	ASPLOS '23: "WACO: Learning Workload-Aware ..." WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program Jaeyeon Won, Charith Mendis, Joel S. Emer, and Saman Amarasinghe (Massachusetts Institute of Technology, USA; University of Illinois at Urbana-Champaign, USA; NVIDIA, USA) In this paper, we present WACO, a novel method of co-optimizing the format and the schedule of a given sparsity pattern in a sparse tensor program. A core challenge in this paper is the design of a lightweight cost model that accurately predicts the runtime of a sparse tensor program by considering the sparsity pattern, the format, and the schedule. The key idea in addressing this is exploiting a sparse convolutional network to learn meaningful features of the sparsity pattern and embedding a coupled behavior between the format and the schedule using a specially designed schedule template. In addition, within the enormous search space of co-optimization, our novel search strategy, an approximate nearest neighbor search, efficiently and accurately retrieves the best format and schedule for a given sparsity pattern. We evaluated WACO for four different algorithms (SpMV, SpMM, SDDMM, and MTTKRP) on a CPU using 726 different sparsity patterns. Our experimental results showed that WACO outperformed four state-of-the-art baselines, Intel MKL, BestFormat, TACO with a default schedule, and ASpT. Compared to the best of four baselines, WACO achieved 1.43×, 1.18×, 1.14×, and 1.27× average speedups on SpMV, SpMM, SDDMM, and MTTKRP, respectively. @InProceedings{ASPLOS23p920, author = {Jaeyeon Won and Charith Mendis and Joel S. Emer and Saman Amarasinghe}, title = {WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {920--934}, doi = {10.1145/3575693.3575742}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Eom, Young Ik	ASPLOS '23: "Prism: Optimizing Key-Value ..." Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices Yongju Song, Wook-Hee Kim, Sumit Kumar Monga, Changwoo Min, and Young Ik Eom (Sungkyunkwan University, South Korea; Konkuk University, South Korea; Virginia Tech, USA) As data generation has been on an upward trend, storing vast volumes of data cost-effectively as well as efficiently accessing them is paramount. At the same time, today's storage landscape continues to diversify, from high-bandwidth storage devices such as NVMe SSDs to low-latency non-volatile memory (e.g., Intel Optane DCPMM). These heterogeneous storage devices have the potential to deliver high performance in terms of bandwidth and latency with cost efficiency, while achieving the performance and cost targets together still remains a challenging problem. We provide our solution, Prism, a novel key-value store that utilizes modern heterogeneous storage devices. Prism uses heterogeneous storage devices synergistically to harness the advantages of each storage device while suppressing their downsides. We devise new techniques to balance the latency-bandwidth tradeoff when reading from SSD. For ensuring multicore scalability and crash consistency of data across heterogeneous storage media, Prism proposes cross-storage concurrency control and cross-storage crash consistency protocols. Our evaluation shows that Prism outperforms state-of-the-art key-value stores by up to 13.1× with significantly lower tail latency. @InProceedings{ASPLOS23p588, author = {Yongju Song and Wook-Hee Kim and Sumit Kumar Monga and Changwoo Min and Young Ik Eom}, title = {Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {588--602}, doi = {10.1145/3575693.3575722}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Ernst, Daniel	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Fayyazi, Arash	ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Feng, Siyuan	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Feng, Yangyang	ASPLOS '23: "Mobius: Fine Tuning Large-Scale ..." Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers Yangyang Feng, Minhui Xie, Zijie Tian, Shuo Wang, Youyou Lu, and Jiwu Shu (Tsinghua University, China) Fine-tuning on cheap commodity GPU servers makes large-scale deep learning models benefit more people. However, the low inter-GPU communication bandwidth and pressing communication contention on the commodity GPU server obstruct training efficiency. In this paper, we present Mobius, a communication-efficient system for fine tuning large-scale models on commodity GPU servers. The key idea is a novel pipeline parallelism scheme enabling heterogeneous memory for large-scale model training, while bringing fewer communications than existing systems. Mobius partitions the model into stages and carefully schedules them between GPU memory and DRAM to overlap communication with computation. It formulates pipeline execution into a mixed-integer program problem to find the optimal pipeline partition. It also features a new stage-to-GPU mapping method termed cross mapping, to minimize communication contention. Experiments on various scale models and GPU topologies show that Mobius significantly reduces the training time by 3.8-5.1× compared with the prior art. @InProceedings{ASPLOS23p489, author = {Yangyang Feng and Minhui Xie and Zijie Tian and Shuo Wang and Youyou Lu and Jiwu Shu}, title = {Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {489--501}, doi = {10.1145/3575693.3575703}, year = {2023}, } Publisher's Version
	Fingler, Henrique	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Fonseca, Pedro	ASPLOS '23: "KIT: Testing OS-Level Virtualization ..." KIT: Testing OS-Level Virtualization for Functional Interference Bugs Congyu Liu, Sishuai Gong, and Pedro Fonseca (Purdue University, USA) Container isolation is implemented through OS-level virtualization, such as Linux namespaces. Unfortunately, these mechanisms are extremely challenging to implement correctly and, in practice, suffer from functional interference bugs, which compromise container security. In particular, functional interference bugs allow an attacker to extract information from another container running on the same machine or impact its integrity by modifying kernel resources that are incorrectly isolated. Despite their impact, functional interference bugs in OS-level virtualization have received limited attention in part due to the challenges in detecting them. Instead of causing memory errors or crashes, many functional interference bugs involve hard-to-catch logic errors that silently produce semantically incorrect results. This paper proposes KIT, a dynamic testing framework that discovers functional interference bugs in OS-level virtualization mechanisms, such as Linux namespaces. The key idea of KIT is to detect inter-container functional interference by comparing the system call traces of a container across two executions, where it runs with and without the preceding execution of another container. To achieve high efficiency and accuracy, KIT includes two critical components: an efficient algorithm to generate test cases that exercise inter-container data flows and a system call trace analysis framework that detects functional interference bugs and clusters bug reports. KIT discovered 9 functional interference bugs in Linux kernel 5.13, of which 6 have been confirmed. All bugs are caused by logic errors, showing that this approach is able to detect hard-to-catch semantic bugs. @InProceedings{ASPLOS23p427, author = {Congyu Liu and Sishuai Gong and Pedro Fonseca}, title = {KIT: Testing OS-Level Virtualization for Functional Interference Bugs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {427--441}, doi = {10.1145/3575693.3575731}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Fontoura, Marcus	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Fu, Ming	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Gan, Yu	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Ganger, Gregory R.	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Gao, Chao	ASPLOS '23: "CommonGraph: Graph Analytics ..." CommonGraph: Graph Analytics on Evolving Data Mahbod Afarin, Chao Gao, Shafiur Rahman, Nael Abu-Ghazaleh, and Rajiv Gupta (University of California at Riverside, USA) We consider the problem of graph analytics on evolving graphs (i.e., graphs that change over time). In this scenario, a query typically needs to be applied to different snapshots of the graph over an extended time window, for example to track the evolution of a property over time. Solving a query independently on multiple snapshots is inefficient due to repeated execution of subcomputation common to multiple snapshots. At the same time, we show that using streaming, where we start from the earliest snapshot and stream the changes to the graph incrementally updating the query results one snapshot at a time is also inefficient. We propose CommonGraph, an approach for efficient processing of queries on evolving graphs. We first observe that deletion operations are significantly more expensive than addition operations for many graph queries (those that are monotonic). CommonGraph converts all deletions to additions by finding a common graph that exists across all snapshots. After computing the query on this graph, to reach any snapshot, we simply need to add the missing edges and incrementally update the query results. CommonGraph also allows sharing of common additions among snapshots that require them, and breaks the sequential dependency inherent in the traditional streaming approach where snapshots are processed in sequence, enabling additional opportunities for parallelism. We incorporate the CommonGraph approach by extending the KickStarter streaming framework. We implement optimizations that enable efficient handling of edge additions without resorting to expensive in place graph mutations, significantly reducing the streaming overhead, and enabling direct reuse of shared edges among different snapshots. CommonGraph achieves 1.38x-8.17x improvement in performance over Kickstarter across multiple benchmarks. @InProceedings{ASPLOS23p133, author = {Mahbod Afarin and Chao Gao and Shafiur Rahman and Nael Abu-Ghazaleh and Rajiv Gupta}, title = {CommonGraph: Graph Analytics on Evolving Data}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {133--145}, doi = {10.1145/3575693.3575713}, year = {2023}, } Publisher's Version
	Gao, Mingyu	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version ASPLOS '23: "Probabilistic Concurrency ..." Probabilistic Concurrency Testing for Weak Memory Programs Mingyu Gao, Soham Chakraborty, and Burcu Kulahcioglu Ozkan (Delft University of Technology, Netherlands) The Probabilistic Concurrency Testing (PCT) algorithm that provides theoretical guarantees on the probability of detecting concurrency bugs does not apply to weak memory programs. The PCT algorithm builds on the interleaving semantics of sequential consistency, which does not hold for weak memory concurrency. It is because weak memory concurrency allows additional behaviors that cannot be produced by any interleaving execution. In this paper, we generalize PCT to address weak memory concurrency and present Probabilistic Concurrency Testing for Weak Memory (PCTWM). We empirically evaluate PCTWM on a set of well-known weak memory program benchmarks in comparison to the state-of-the-art weak memory testing tool C11Tester. Our results show that PCTWM can detect concurrency bugs more frequently than C11Tester. @InProceedings{ASPLOS23p603, author = {Mingyu Gao and Soham Chakraborty and Burcu Kulahcioglu Ozkan}, title = {Probabilistic Concurrency Testing for Weak Memory Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {603--616}, doi = {10.1145/3575693.3575729}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Gao, Yizhao	ASPLOS '23: "DPACS: Hardware Accelerated ..." DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design Yizhao Gao, Baoheng Zhang, Xiaojuan Qi, and Hayden Kwok-Hay So (University of Hong Kong, Hong Kong) By eliminating compute operations intelligently based on the run time input, dynamic pruning (DP) promises to improve deep neural network inference speed substantially without incurring a major impact on their accuracy. Although many DP algorithms with good pruning performance have been proposed, it remains a challenge to translate these theoretical reductions in compute operations into satisfactory end-to-end speedups in practical real-world implementations. The overhead of identifying operations to be pruned during run time, the need to efficiently process the resulting dynamic dataflow, and the non-trivial memory I/O bottleneck that emerges as the number of compute operations reduces, have all contributed to the challenge of implementing practical DP systems. In this paper, the design and implementation of DPACS are presented to address these challenges. DPACS utilizes a hardware-aware dynamic spatial and channel pruning algorithm in conjunction with a dynamic dataflow engine in hardware to facilitate efficient processing of the pruned network. A channel mask precomputation scheme is designed to reduce memory I/O, and a dedicated inter-layer pipeline is used to achieve efficient indexing and dataflow of sparse activation. Extensive design space exploration has been performed using two architectural variations implemented on FPGA to accelerate multiple networks from the ResNet family on the ImageNet and CIFAR10 dataset across a wide range of pruning ratios. Across the spectrum of configurations, DPACS is able to achieve 1.1× to 3.9× end-to-end speedup over a baseline hardware implementation without pruning. Analysis of the tradeoff among accuracy, compute, and memory I/O performance highlights the importance of algorithm-architecture codesign in developing DP systems. @InProceedings{ASPLOS23p237, author = {Yizhao Gao and Baoheng Zhang and Xiaojuan Qi and Hayden Kwok-Hay So}, title = {DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {237--251}, doi = {10.1145/3575693.3575728}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Garland, Michael	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Gelado, Isaac	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Glova, Alvin Oliver	ASPLOS '23: "A Prediction System Service ..." A Prediction System Service Zhizhou Zhang, Alvin Oliver Glova, Timothy Sherwood, and Jonathan Balkind (University of California at Santa Barbara, USA) To better facilitate application performance programming we propose a software optimization strategy enabled by a novel low-latency Prediction System Service (PSS). Rather than relying on nuanced domain-specific knowledge or slapdash heuristics, a system service for prediction encourages programmers to spend their time uncovering new levers for optimization rather than worrying about the details of their control. The core idea is to write optimizations that improve performance in specific cases, or under specific tunings, and leave the decision of how and when exactly to apply those optimizations to the system to learn through feedback-directed learning. Such a prediction service can be implemented in any number of ways, including as a shared library that can be easily reused by software written in different programming languages, and opens the door to both new software optimization patterns and hardware design possibilities. As a demonstration of the utility of this approach, we show that three very different application-targeted optimization scenarios can each benefit from even a very straightforward perceptron-based implementation of the PSS as long as the service latency can be held low. First, we show that PSS can be used to more intelligently guide hardware lock elision with resulting speedups over a baseline implementation by 34% on average. Second, we show that a PSS can find good configuration parameters for PyPy’s Just-In-Time (JIT) compiler resulting in 15% speedup on average. Last, we show PSS can guide the page reclamation task within a kernel memory management subsystem to reduce the average memory latency by 33% on average. In all three cases, this new optimization pattern with service support is able to meet or beat the best-known hand-crafted methods with a fraction of the complexity. @InProceedings{ASPLOS23p48, author = {Zhizhou Zhang and Alvin Oliver Glova and Timothy Sherwood and Jonathan Balkind}, title = {A Prediction System Service}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {48--60}, doi = {10.1145/3575693.3575714}, year = {2023}, } Publisher's Version
	Gong, Sishuai	ASPLOS '23: "KIT: Testing OS-Level Virtualization ..." KIT: Testing OS-Level Virtualization for Functional Interference Bugs Congyu Liu, Sishuai Gong, and Pedro Fonseca (Purdue University, USA) Container isolation is implemented through OS-level virtualization, such as Linux namespaces. Unfortunately, these mechanisms are extremely challenging to implement correctly and, in practice, suffer from functional interference bugs, which compromise container security. In particular, functional interference bugs allow an attacker to extract information from another container running on the same machine or impact its integrity by modifying kernel resources that are incorrectly isolated. Despite their impact, functional interference bugs in OS-level virtualization have received limited attention in part due to the challenges in detecting them. Instead of causing memory errors or crashes, many functional interference bugs involve hard-to-catch logic errors that silently produce semantically incorrect results. This paper proposes KIT, a dynamic testing framework that discovers functional interference bugs in OS-level virtualization mechanisms, such as Linux namespaces. The key idea of KIT is to detect inter-container functional interference by comparing the system call traces of a container across two executions, where it runs with and without the preceding execution of another container. To achieve high efficiency and accuracy, KIT includes two critical components: an efficient algorithm to generate test cases that exercise inter-container data flows and a system call trace analysis framework that detects functional interference bugs and clusters bug reports. KIT discovered 9 functional interference bugs in Linux kernel 5.13, of which 6 have been confirmed. All bugs are caused by logic errors, showing that this approach is able to detect hard-to-catch semantic bugs. @InProceedings{ASPLOS23p427, author = {Congyu Liu and Sishuai Gong and Pedro Fonseca}, title = {KIT: Testing OS-Level Virtualization for Functional Interference Bugs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {427--441}, doi = {10.1145/3575693.3575731}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Gu, Diandian	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Guerraoui, Rachid	ASPLOS '23: "uBFT: Microsecond-Scale BFT ..." uBFT: Microsecond-Scale BFT using Disaggregated Memory Marcos K. Aguilera, Naama Ben-David, Rachid Guerraoui, Antoine Murat, Athanasios Xygkis, and Igor Zablotchi (VMware Research, USA; EPFL, Switzerland; Massachusetts Institute of Technology, USA) We propose uBFT, the first State Machine Replication (SMR) system to achieve microsecond-scale latency in data centers, while using only 2f+1 replicas to tolerate f Byzantine failures. The Byzantine Fault Tolerance (BFT) provided by uBFT is essential as pure crashes appear to be a mere illusion with real-life systems reportedly failing in many unexpected ways. uBFT relies on a small non-tailored trusted computing base—disaggregated memory—and consumes a practically bounded amount of memory. uBFT is based on a novel abstraction called Consistent Tail Broadcast, which we use to prevent equivocation while bounding memory. We implement uBFT using RDMA-based disaggregated memory and obtain an end-to-end latency of as little as 10 us. This is at least 50× faster than MinBFT, a state-of-the-art 2f+1 BFT SMR based on Intel’s SGX. We use uBFT to replicate two KV-stores (Memcached and Redis), as well as a financial order matching engine (Liquibook). These applications have low latency (up to 20 us) and become Byzantine tolerant with as little as 10 us more. The price for uBFT is a small amount of reliable disaggregated memory (less than 1 MiB), which in our prototype consists of a small number of memory servers connected through RDMA and replicated for fault tolerance. @InProceedings{ASPLOS23p862, author = {Marcos K. Aguilera and Naama Ben-David and Rachid Guerraoui and Antoine Murat and Athanasios Xygkis and Igor Zablotchi}, title = {uBFT: Microsecond-Scale BFT using Disaggregated Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {862--877}, doi = {10.1145/3575693.3575732}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Gunawi, Haryadi S.	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version
	Guo, Liwei	ASPLOS '23: "STI: Turbocharge NLP Inference ..." STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining Liwei Guo, Wonkyo Choe, and Felix Xiaozhu Lin (University of Virginia, USA) Natural Language Processing (NLP) inference is seeing increasing adoption by mobile applications, where on-device inference is desirable for crucially preserving user data privacy and avoiding network roundtrips. Yet, the unprecedented size of an NLP model stresses both latency and memory, creating a tension between the two key resources of a mobile device. To meet a target latency, holding the whole model in memory launches execution as soon as possible but increases one app’s memory footprints by several times, limiting its benefits to only a few inferences before being recycled by mobile memory management. On the other hand, loading the model from storage on demand incurs IO as long as a few seconds, far exceeding the delay range satisfying to a user; pipelining layerwise model loading and execution does not hide IO either, due to the high skewness between IO and computation delays. To this end, we propose Speedy Transformer Inference (STI). Built on the key idea of maximizing IO/compute resource utilization on the most important parts of a model, STI reconciles the latency v.s. memory tension via two novel techniques. First, model sharding. STI manages model parameters as independently tunable shards, and profiles their importance to accuracy. Second, elastic pipeline planning with a preload buffer. STI instantiates an IO/compute pipeline and uses a small buffer for preload shards to bootstrap execution without stalling at early stages; it judiciously selects, tunes, and assembles shards per their importance for resource-elastic execution, maximizing inference accuracy. Atop two commodity SoCs, we build STI and evaluate it against a wide range of NLP tasks, under a practical range of target latencies, and on both CPU and GPU. We demonstrate that STI delivers high accuracies with 1–2 orders of magnitude lower memory, outperforming competitive baselines. @InProceedings{ASPLOS23p791, author = {Liwei Guo and Wonkyo Choe and Felix Xiaozhu Lin}, title = {STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {791--803}, doi = {10.1145/3575693.3575698}, year = {2023}, } Publisher's Version
	Guo, Minyi	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Guo, Tianhao	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Gupta, Rajiv	ASPLOS '23: "CommonGraph: Graph Analytics ..." CommonGraph: Graph Analytics on Evolving Data Mahbod Afarin, Chao Gao, Shafiur Rahman, Nael Abu-Ghazaleh, and Rajiv Gupta (University of California at Riverside, USA) We consider the problem of graph analytics on evolving graphs (i.e., graphs that change over time). In this scenario, a query typically needs to be applied to different snapshots of the graph over an extended time window, for example to track the evolution of a property over time. Solving a query independently on multiple snapshots is inefficient due to repeated execution of subcomputation common to multiple snapshots. At the same time, we show that using streaming, where we start from the earliest snapshot and stream the changes to the graph incrementally updating the query results one snapshot at a time is also inefficient. We propose CommonGraph, an approach for efficient processing of queries on evolving graphs. We first observe that deletion operations are significantly more expensive than addition operations for many graph queries (those that are monotonic). CommonGraph converts all deletions to additions by finding a common graph that exists across all snapshots. After computing the query on this graph, to reach any snapshot, we simply need to add the missing edges and incrementally update the query results. CommonGraph also allows sharing of common additions among snapshots that require them, and breaks the sequential dependency inherent in the traditional streaming approach where snapshots are processed in sequence, enabling additional opportunities for parallelism. We incorporate the CommonGraph approach by extending the KickStarter streaming framework. We implement optimizations that enable efficient handling of edge additions without resorting to expensive in place graph mutations, significantly reducing the streaming overhead, and enabling direct reuse of shared edges among different snapshots. CommonGraph achieves 1.38x-8.17x improvement in performance over Kickstarter across multiple benchmarks. @InProceedings{ASPLOS23p133, author = {Mahbod Afarin and Chao Gao and Shafiur Rahman and Nael Abu-Ghazaleh and Rajiv Gupta}, title = {CommonGraph: Graph Analytics on Evolving Data}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {133--145}, doi = {10.1145/3575693.3575713}, year = {2023}, } Publisher's Version
	Gupta, Udit	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Gyurgyik, Christophe	ASPLOS '23: "Stepwise Debugging for Hardware ..." Stepwise Debugging for Hardware Accelerators Griffin Berlstein, Rachit Nigam, Christophe Gyurgyik, and Adrian Sampson (Cornell University, USA) High-level programming models for hardware design let domain experts quickly produce specialized accelerators. However, tools for debugging these accelerators remain tied to low-level hardware description languages (HDLs). High-level descriptions contain control-flow information that is lost in HDL code. We describe Cider, a stepwise debugger that exploits this information to provide software-like debugging abstractions for languages that compile to hardware. Cider uses Calyx, an intermediate language for accelerator generators that preserves control information. Cider provides breakpoints, watchpoints, state inspection, and source-level position mapping. Using case studies that examine one new and two preexisting accelerator generators, we demonstrate how Cider helps find and localize previously unreported bugs. By directly simulating a control-rich representation, Cider avoids wasting effort on inactive parts of the design and, despite being largely unoptimized, performs competitively with open-source HDL simulators. @InProceedings{ASPLOS23p778, author = {Griffin Berlstein and Rachit Nigam and Christophe Gyurgyik and Adrian Sampson}, title = {Stepwise Debugging for Hardware Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {778--790}, doi = {10.1145/3575693.3575717}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hackett, Finn	ASPLOS '23: "Compiling Distributed System ..." Compiling Distributed System Models with PGo Finn Hackett, Shayan Hosseini, Renato Costa, Matthew Do, and Ivan Beschastnikh (University of British Columbia, Canada) Distributed systems are difficult to design and implement correctly. In response, both research and industry are exploring applications of formal methods to distributed systems. A key challenge in this domain is the missing link between the formal design of a system and its implementation. Today, practitioners bridge this link through manual effort. We present a language called Modular PlusCal that extends PlusCal by cleanly separating the model of a system from a model of its environment. We then present a compiler tool-chain called PGo that automatically translates MPCal models to TLA+ for model checking, and that also compiles MPCal models to runnable Go code. PGo provides system designers with a new ability to model and check their designs, and then re-use their modeling efforts to mechanically extract runnable implementations of their designs. Our evaluation shows that the PGo approach works for complex models: we model check, compile, and evaluate the performance of MPCal systems based on Raft, CRDTs, and primary-backup. Compared to previous work, PGo requires less time to develop a checked model and derive a fully working implementation. With PGo we created a formally checked Raft model and its corresponding implementation in under 1 person-month, which is 3× less time than Ivy. Our evaluation shows that a PGo-based Raft KV store with three nodes has 41% higher throughput than a Raft KV store based on Ivy, the highest performing verified Raft-based KV store from related work. A PGo-based CRDT set has a latency within 2 × of a CRDT set implementation from SoundCloud called Roshi. @InProceedings{ASPLOS23p159, author = {Finn Hackett and Shayan Hosseini and Renato Costa and Matthew Do and Ivan Beschastnikh}, title = {Compiling Distributed System Models with PGo}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {159--175}, doi = {10.1145/3575693.3575695}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Han, Sungsoo	ASPLOS '23: "ShakeFlow: Functional Hardware ..." ShakeFlow: Functional Hardware Description with Latency-Insensitive Interface Combinators Sungsoo Han, Minseong Jang, and Jeehoon Kang (KAIST, South Korea) Functional programming’s benefits for hardware description have long been recognized in the literature. In particular, functional hardware description languages provide combinators such as maps and filters to facilitate the compositional description of circuits. However, it is challenging to apply functional programming with combinators to complex circuits with latency-insensitive interfaces such as valid/ready interfaces due to the cyclic nature of their forward and backward ports. In this work, we present ShakeFlow: the first functional hardware description language supporting latency-insensitive interface combinators. ShakeFlow provides extensible support for custom interfaces and combinators and a compiler to synthesizable Verilog and FIRRTL. We port a part of the BaseJump STL library and the Corundum 100Gbps NIC from (System)Verilog to ShakeFlow, reducing the code size by 38% and 26%, respectively. By experimenting with Corundum, we demonstrate that ShakeFlow is capable of designing realistic circuits, and porting to ShakeFlow does not incur significant resource and performance overhead. @InProceedings{ASPLOS23p702, author = {Sungsoo Han and Minseong Jang and Jeehoon Kang}, title = {ShakeFlow: Functional Hardware Description with Latency-Insensitive Interface Combinators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {702--717}, doi = {10.1145/3575693.3575701}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Han, Zhenhua	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hanafy, Walid	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Härtig, Hermann	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	He, Yongchao	ASPLOS '23: "A Generic Service to Provide ..." A Generic Service to Provide In-Network Aggregation for Key-Value Streams Yongchao He, Wenfei Wu, Yanfang Le, Ming Liu, and ChonLam Lao (Tsinghua University, China; Peking University, China; Intel, USA; University of Wisconsin-Madison, USA; Harvard University, USA) Key-value stream aggregation is a common operation in distributed systems, which requires intensive computation and network resources. We propose a generic in-network aggregation service for key-value streams, ASK, to accelerate the aggregation operations in diverse distributed applications. ASK is a switch-host co-designed system, where the programmable switch provides a best-effort aggregation service, and the host runs a daemon to interact with applications. ASK makes in-depth optimization tailored to traffic characteristics, hardware restrictions, and network unreliable natures: it vectorizes multiple key-value tuples’ aggregation of one packet in one switch pipeline pass, which improves the per-host’s goodput; it develops a lightweight reliability mechanism for key-value stream’s asynchronous aggregation, which guarantees computation correctness; it designs a hot-key agnostic prioritization for key-skewed workloads, which improves the switch memory utilization. We prototype ASK and use it to support Spark and BytePS. The evaluation shows that ASK could accelerate pure key-value aggregation tasks by up to 155 times and big data jobs by 3-5 times, and be backward compatible with existing INA-empowered distributed training solutions with the same speedup. @InProceedings{ASPLOS23p33, author = {Yongchao He and Wenfei Wu and Yanfang Le and Ming Liu and ChonLam Lao}, title = {A Generic Service to Provide In-Network Aggregation for Key-Value Streams}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {33--47}, doi = {10.1145/3575693.3575708}, year = {2023}, } Publisher's Version
	Hildenbrand, David	ASPLOS '23: "Copy-on-Pin: The Missing Piece ..." Copy-on-Pin: The Missing Piece for Correct Copy-on-Write David Hildenbrand, Martin Schulz, and Nadav Amit (TU Munich, Germany; Red Hat, Germany; VMware, USA) Operating systems utilize Copy-on-Write (COW) to conserve memory and improve performance. During the last two decades, a series of COW-related bugs - which compromised security, corrupted memory and degraded performance - was found. The majority of these bugs are related to page "pinning", which operating systems employ to access process memory efficiently and to perform direct I/O. Unfortunately, the true cause of these bugs is not well understood, resulting in incomplete bug fixes. We show this by: (1) surveying previously reported pinning-related COW bugs; (2) uncovering new such bugs in Linux, FreeBSD, and NetBSD; and (3) showing that they occur because the COW logic does not consider page pinnings correctly, resulting in incorrect behavior (e.g., I/O of stale data). We then address the underlying problem by deriving when/how shared pages must be copied and under which conditions pinned pages can be shared to maintain correctness. Based on this assessment, we introduce the "Copy-on-Pin (COP)" scheme, an extension of the COW mechanism that handles pinned pages correctly by ensuring pinned pages and shared pages are mutually exclusive. However, we find that a naive implementation of this scheme hampers performance and increases complexity if pages are copied only when strictly necessary. To compensate, we introduce a relaxed-COP design, which does not require precise tracking of page sharing, maintains correctness without increasing complexity, and (while potentially needlessly copying pages in some corner cases) marginally improves performance. Our relaxed-COP solution has been integrated into Linux 5.19. @InProceedings{ASPLOS23p176, author = {David Hildenbrand and Martin Schulz and Nadav Amit}, title = {Copy-on-Pin: The Missing Piece for Correct Copy-on-Write}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {176--191}, doi = {10.1145/3575693.3575716}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hill, Mark D.	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Hoffmann, Henry	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version
	Hong, Siyu	ASPLOS '23: "VDom: Fast and Unlimited Virtual ..." VDom: Fast and Unlimited Virtual Domains on Multiple Architectures Ziqi Yuan, Siyu Hong, Rui Chang, Yajin Zhou, Wenbo Shen, and Kui Ren (Zhejiang University, China) Hardware memory domain primitives, such as Intel MPK and ARM Memory Domain, have been used for efficient in-process memory isolation. However, they can only provide a limited number of memory domains (16 domains), which cannot satisfy the compelling need for more isolated domains inside the address space of a process. Existing solutions to virtualize memory domains are either intrusive (need the modification to existing hardware), or incur a large performance overhead. In this paper, we propose VDom, a fast and scalable memory domain virtualization system that supports unlimited memory domains. VDom leverages separate address spaces to provide an unlimited number of virtual domains, and optimizes related memory management operations. To map virtual domains to hardware domains, we design a domain virtualization algorithm, which manages address spaces and domain maps for threads to efficiently access other domains that are unmapped in the current address space. According to our evaluation on real Intel and ARM platforms, on real-world server applications (httpd and MySQL), VDom incurs less than 2.65% performance overhead, which is lower than the overheads of the state-of-the-art software approaches (libmpk and EPK). In random domain access tests, VDom is comparable to EPK and has significantly higher efficiency than libmpk. @InProceedings{ASPLOS23p905, author = {Ziqi Yuan and Siyu Hong and Rui Chang and Yajin Zhou and Wenbo Shen and Kui Ren}, title = {VDom: Fast and Unlimited Virtual Domains on Multiple Architectures}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {905--919}, doi = {10.1145/3575693.3575735}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hosseini, Shayan	ASPLOS '23: "Compiling Distributed System ..." Compiling Distributed System Models with PGo Finn Hackett, Shayan Hosseini, Renato Costa, Matthew Do, and Ivan Beschastnikh (University of British Columbia, Canada) Distributed systems are difficult to design and implement correctly. In response, both research and industry are exploring applications of formal methods to distributed systems. A key challenge in this domain is the missing link between the formal design of a system and its implementation. Today, practitioners bridge this link through manual effort. We present a language called Modular PlusCal that extends PlusCal by cleanly separating the model of a system from a model of its environment. We then present a compiler tool-chain called PGo that automatically translates MPCal models to TLA+ for model checking, and that also compiles MPCal models to runnable Go code. PGo provides system designers with a new ability to model and check their designs, and then re-use their modeling efforts to mechanically extract runnable implementations of their designs. Our evaluation shows that the PGo approach works for complex models: we model check, compile, and evaluate the performance of MPCal systems based on Raft, CRDTs, and primary-backup. Compared to previous work, PGo requires less time to develop a checked model and derive a fully working implementation. With PGo we created a formally checked Raft model and its corresponding implementation in under 1 person-month, which is 3× less time than Ivy. Our evaluation shows that a PGo-based Raft KV store with three nodes has 41% higher throughput than a Raft KV store based on Ivy, the highest performing verified Raft-based KV store from related work. A PGo-based CRDT set has a latency within 2 × of a CRDT set implementation from SoundCloud called Roshi. @InProceedings{ASPLOS23p159, author = {Finn Hackett and Shayan Hosseini and Renato Costa and Matthew Do and Ivan Beschastnikh}, title = {Compiling Distributed System Models with PGo}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {159--175}, doi = {10.1145/3575693.3575695}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hou, Bohan	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Hsu, Lisa	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Hu, Bodun	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hu, Qinghao	ASPLOS '23: "Lucid: A Non-intrusive, Scalable ..." Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs Qinghao Hu, Meng Zhang, Peng Sun, Yonggang Wen, and Tianwei Zhang (Nanyang Technological University, Singapore; Shanghai AI Laboratory, China; SenseTime, China) While recent deep learning workload schedulers exhibit excellent performance, it is arduous to deploy them in practice due to some substantial defects, including inflexible intrusive manner, exorbitant integration and maintenance cost, limited scalability, as well as opaque decision processes. Motivated by these issues, we design and implement Lucid, a non-intrusive deep learning workload scheduler based on interpretable models. It consists of three innovative modules. First, a two-dimensional optimized profiler is introduced for efficient job metric collection and timely debugging job feedback. Second, Lucid utilizes an indolent packing strategy to circumvent interference. Third, Lucid orchestrates resources based on estimated job priority values and sharing scores to achieve efficient scheduling. Additionally, Lucid promotes model performance maintenance and system transparent adjustment via a well-designed system optimizer. Our evaluation shows that Lucid reduces the average job completion time by up to 1.3× compared with state-of-the-art preemptive scheduler Tiresias. Furthermore, it provides explicit system interpretations and excellent scalability for practical deployment. @InProceedings{ASPLOS23p457, author = {Qinghao Hu and Meng Zhang and Peng Sun and Yonggang Wen and Tianwei Zhang}, title = {Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {457--472}, doi = {10.1145/3575693.3575705}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hu, Weifang	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Huang, Gang	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Huang, Haochen	ASPLOS '23: "Protecting Data Integrity ..." Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code Haochen Huang, Bingyu Shen, Li Zhong, and Yuanyuan Zhou (University of California at San Diego, USA) Database-backed web applications persist a large amount of production data and have high requirements for integrity. To protect data integrity against application code bugs and operator mistakes, most RDBMSes allow application developers to specify various types of integrity constraints. Unfortunately, applications (e.g., e-commerce web apps) often do not take full advantage of this capability and miss specifying many database constraints, resulting in many severe consequences, such as crashing the order placement page and corrupting the store inventory data. In this paper, we focus on the problem of missing database constraints in web applications. We first study several widely used open-source e-commerce and communication applications, and observe that all these applications have missed integrity constraints and many were added later as afterthoughts after issues occurred. Motivated by our observations, we build a tool called CFinder to automatically infer missing database constraints from application source code by cleverly leveraging the observation that many source code patterns usually imply certain data integrity constraints. By analyzing application source code automatically, CFinder can extract such constraints and check against their database schemas to detect missing ones. We evaluate CFinder with eight widelydeployed web applications, including one commercial company with millions of users. Overall, our tool identifies 210 previously unknown missing constraints. We have reported 92 of them to the developers of these applications, so far 75 are confirmed. Our tool achieves a precision of 78% and a recall of 79%. @InProceedings{ASPLOS23p632, author = {Haochen Huang and Bingyu Shen and Li Zhong and Yuanyuan Zhou}, title = {Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {632--645}, doi = {10.1145/3575693.3575699}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Huang, Jian	ASPLOS '23: "LeaFTL: A Learning-Based Flash ..." LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives Jinghan Sun, Shaobo Li, Yunxin Sun, Chao Sun, Dejan Vucinic, and Jian Huang (University of Illinois at Urbana-Champaign, USA; ETH Zurich, Switzerland; Western Digital Research, USA) In modern solid-state drives (SSDs), the indexing of flash pages is a critical component in their storage controllers. It not only affects the data access performance, but also determines the efficiency of the precious in-device DRAM resource. A variety of address mapping schemes and optimizations have been proposed. However, most of them were developed with human-driven heuristics. In this paper, we present a learning-based flash translation layer (FTL), named LeaFTL, which learns the address mapping to tolerate dynamic data access patterns via linear regression at runtime. By grouping a large set of mapping entries into a learned segment, it significantly reduces the memory footprint of the address mapping table, which further benefits the data caching in SSD controllers. LeaFTL also employs various optimization techniques, including out-of-band metadata verification to tolerate mispredictions, optimized flash allocation, and dynamic compaction of learned index segments. We implement LeaFTL with both a validated SSD simulator and a real open-channel SSD board. Our evaluation with various storage workloads demonstrates that LeaFTL saves the memory consumption of the mapping table by 2.9× and improves the storage performance by 1.4× on average, in comparison with state-of-the-art FTL schemes. @InProceedings{ASPLOS23p442, author = {Jinghan Sun and Shaobo Li and Yunxin Sun and Chao Sun and Dejan Vucinic and Jian Huang}, title = {LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {442--456}, doi = {10.1145/3575693.3575744}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Hwu, Wen-mei	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Irwin, David	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Jahre, Magnus	ASPLOS '23: "NUBA: Non-Uniform Bandwidth ..." NUBA: Non-Uniform Bandwidth GPUs Xia Zhao, Magnus Jahre, Yuhua Tang, Guangda Zhang, and Lieven Eeckhout (Academy of Military Sciences, China; NTNU, Norway; National University of Defense Technology, China; Ghent University, Belgium) The parallel execution model of GPUs enables scaling to hundreds of thousands of threads, which is a key capability that many modern high-performance applications exploit. GPU vendors are hence increasing the compute and memory resources with every GPU generation — resulting in the need to efficiently stitch together a plethora of Symmetric Multiprocessors (SMs), Last-Level Cache (LLC) slices and memory controllers while maximizing bandwidth and keeping power consumption and design complexity in check. Conventional GPUs are Uniform Bandwidth Architectures (UBAs) as they provide equal bandwidth between all SMs and all LLC slices. UBA GPUs require a uniform high-bandwidth Network-on-Chip (NoC), and our key observation is that provisioning a NoC to match the LLC slice bandwidth incurs a hefty power and complexity overhead. We propose the Non-Uniform Bandwidth Architecture (NUBA), a GPU system architecture aimed at fully utilizing LLC slice bandwidth. A NUBA GPU consists of partitions that each feature a few SMs and LLC slices as well as a memory controller — hence exposing the complete LLC bandwidth to the SMs within a partition since they can be connected with point-to-point links — and a NoC between partitions — to enable access to remote data.Exploiting the potential of NUBA GPUs however requires carefully co-designing system software, the compiler and architectural policies. The critical system software component is our Local-And-Balanced (LAB) page placement policy which enables the GPU driver to place data in local partitions while avoiding load imbalance. Moreover, we propose Model-Driven Replication (MDR) which identifies read-only shared data with data-flow analysis at compile time. At run time, MDR leverages an architectural mechanism that replicates read-only shared data across LLC slices when this can be done without pressuring cache capacity. With LAB and MDR, our NUBA GPU improves average performance by 23.1% and 22.2% (and up to 183.9% and 182.4%) compared to iso-resource memory-side and SM-side UBA GPUs, respectively. When the NUBA concept is leveraged to reduce overhead while maintaining similar performance, NUBA reduces NoC power consumption by 12.1× and 9.4×, respectively. @InProceedings{ASPLOS23p544, author = {Xia Zhao and Magnus Jahre and Yuhua Tang and Guangda Zhang and Lieven Eeckhout}, title = {NUBA: Non-Uniform Bandwidth GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {544--559}, doi = {10.1145/3575693.3575745}, year = {2023}, } Publisher's Version
	Jang, Hongsun	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Jang, Minseong	ASPLOS '23: "ShakeFlow: Functional Hardware ..." ShakeFlow: Functional Hardware Description with Latency-Insensitive Interface Combinators Sungsoo Han, Minseong Jang, and Jeehoon Kang (KAIST, South Korea) Functional programming’s benefits for hardware description have long been recognized in the literature. In particular, functional hardware description languages provide combinators such as maps and filters to facilitate the compositional description of circuits. However, it is challenging to apply functional programming with combinators to complex circuits with latency-insensitive interfaces such as valid/ready interfaces due to the cyclic nature of their forward and backward ports. In this work, we present ShakeFlow: the first functional hardware description language supporting latency-insensitive interface combinators. ShakeFlow provides extensible support for custom interfaces and combinators and a compiler to synthesizable Verilog and FIRRTL. We port a part of the BaseJump STL library and the Corundum 100Gbps NIC from (System)Verilog to ShakeFlow, reducing the code size by 38% and 26%, respectively. By experimenting with Corundum, we demonstrate that ShakeFlow is capable of designing realistic circuits, and porting to ShakeFlow does not incur significant resource and performance overhead. @InProceedings{ASPLOS23p702, author = {Sungsoo Han and Minseong Jang and Jeehoon Kang}, title = {ShakeFlow: Functional Hardware Description with Latency-Insensitive Interface Combinators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {702--717}, doi = {10.1145/3575693.3575701}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Javadi-Abhari, Ali	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Jayarajan, Anand	ASPLOS '23: "TiLT: A Time-Centric Approach ..." TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization Anand Jayarajan, Wei Zhao, Yudi Sun, and Gennady Pekhimenko (University of Toronto, Canada) Stream processing engines (SPEs) are widely used for large scale streaming analytics over unbounded time-ordered data streams. Modern day streaming analytics applications exhibit diverse compute characteristics and demand strict latency and throughput requirements. Over the years, there has been significant attention in building hardware-efficient stream processing engines (SPEs) that support several query optimization, parallelization, and execution strategies to meet the performance requirements of large scale streaming analytics applications. However, in this work, we observe that these strategies often fail to generalize well on many real-world streaming analytics applications due to several inherent design limitations of current SPEs. We further argue that these limitations stem from the shortcomings of the fundamental design choices and the query representation model followed in modern SPEs. To address these challenges, we first propose TiLT, a novel intermediate representation (IR) that offers a highly expressive temporal query language amenable to effective query optimization and parallelization strategies. We subsequently build a compiler backend for TiLT that applies such optimizations on streaming queries and generates hardware-efficient code to achieve high performance on multi-core stream query executions. We demonstrate that TiLT achieves up to 326× (20.49× on average) higher throughput compared to state-of-the-art SPEs (e.g., Trill) across eight real-world streaming analytics applications. TiLT source code is available at https://github.com/ampersand-projects/tilt.git. @InProceedings{ASPLOS23p818, author = {Anand Jayarajan and Wei Zhao and Yudi Sun and Gennady Pekhimenko}, title = {TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {818--832}, doi = {10.1145/3575693.3575704}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Jeon, Jekyeom	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Ji, Jianmin	ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Jin, Hai	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version ASPLOS '23: "Revisiting Log-Structured ..." Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems Zhuohui Duan, Jiabo Yao, Haikun Liu, Xiaofei Liao, Hai Jin, and Yu Zhang (Huazhong University of Science and Technology, China) We present MioDB, a novel LSM-tree based key-value (KV) store system designed to fully exploit the advantages of byte-addressable non-volatile memories (NVMs). Our experimental studies reveal that the performance bottleneck of LSM-tree based KV stores using NVMs mainly stems from (1) costly data serialization/deserialization across memory and storage, and (2) unbalanced speed between memory-to-disk data flushing and on-disk data compaction. They may cause unpredictable performance degradation due to write stalls and write amplification. To address these problems, we advocate byte-addressable and persistent skip lists to replace the on-disk data structure of LSM-tree, and design four novel techniques to make the best use of fast NVMs. First, we propose one-piece flushing to minimize the cost of data serialization from DRAM to NVM. Second, we exploit an elastic NVM buffer with multiple levels and zero-copy compaction to eliminate write stalls and reduce write amplification. Third, we propose parallel compaction to orchestrate data flushing and compactions across all levels of LSM-trees. Finally, MioDB increases the depth of LSM-tree and exploits bloom filters to improve the read performance. Our extensive experimental studies demonstrate that MioDB achieves 17.1× and 21.7× lower 99.9th percentile latency, 8.3× and 2.5× higher random write throughput, and up to 5× and 4.9× lower write amplification compared with the state-of-the-art NoveLSM and MatrixKV, respectively. @InProceedings{ASPLOS23p674, author = {Zhuohui Duan and Jiabo Yao and Haikun Liu and Xiaofei Liao and Hai Jin and Yu Zhang}, title = {Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {674--687}, doi = {10.1145/3575693.3575715}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Jin, Hongyi	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Jin, Xin	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Jung, Jaehoon	ASPLOS '23: "DeepUM: Tensor Migration and ..." DeepUM: Tensor Migration and Prefetching in Unified Memory Jaehoon Jung, Jinpyo Kim, and Jaejin Lee (Moreh, South Korea; Seoul National University, South Korea) Deep neural networks (DNNs) are continuing to get wider and deeper. As a result, it requires a tremendous amount of GPU memory and computing power. In this paper, we propose a framework called DeepUM that exploits CUDA Unified Memory (UM) to allow GPU memory oversubscription for DNNs. While UM allows memory oversubscription using a page fault mechanism, page migration introduces enormous overhead. DeepUM uses a new correlation prefetching technique to hide the page migration overhead. It is fully automatic and transparent to users. We also propose two optimization techniques to minimize the GPU fault handling time. We evaluate the performance of DeepUM using nine large-scale DNNs from MLPerf, PyTorch examples, and Hugging Face and compare its performance with six state-of-the-art GPU memory swapping approaches. The evaluation result indicates that DeepUM is very effective for GPU memory oversubscription and can handle larger models that other approaches fail to handle. @InProceedings{ASPLOS23p207, author = {Jaehoon Jung and Jinpyo Kim and Jaejin Lee}, title = {DeepUM: Tensor Migration and Prefetching in Unified Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {207--221}, doi = {10.1145/3575693.3575736}, year = {2023}, } Publisher's Version
	Jung, Jaewon	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Kamath, Aditya K	ASPLOS '23: "Scoped Buffered Persistency ..." Scoped Buffered Persistency Model for GPUs Shweta Pandey, Aditya K Kamath, and Arkaprava Basu (IISc Bangalore, India; University of Washington, USA) While the implications of persistent memory (PM) on CPU hardware and software are well-explored, the same is not true for GPUs (Graphics Processing Units). A recent work, GPM, demonstrated how GPU programs can benefit from the fine-grain persistence of PM. However, in the absence of a persistency model, one cannot reason about the correctness of PM-aware GPU programs. Persistency models define the order in which writes to PM are persisted. We explore persistency models for GPUs. We explore persistency models for GPUs. We demonstrate that CPU persistency models fall short for GPUs. We qualitatively and quantitatively argue that GPU persistency models should support scopes and buffering of writes to PM to leverage parallelism while adapting to higher NVM latencies. We formally specify a GPU persistency model that supports both scopes and buffers. We detail how GPU architecture can efficiently realize such a model. Finally, we quantitatively demonstrate the usefulness of scopes and buffers for PM-aware GPU programs. @InProceedings{ASPLOS23p688, author = {Shweta Pandey and Aditya K Kamath and Arkaprava Basu}, title = {Scoped Buffered Persistency Model for GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {688--701}, doi = {10.1145/3575693.3575749}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Kaminsky, Michael	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Kang, Jeehoon	ASPLOS '23: "ShakeFlow: Functional Hardware ..." ShakeFlow: Functional Hardware Description with Latency-Insensitive Interface Combinators Sungsoo Han, Minseong Jang, and Jeehoon Kang (KAIST, South Korea) Functional programming’s benefits for hardware description have long been recognized in the literature. In particular, functional hardware description languages provide combinators such as maps and filters to facilitate the compositional description of circuits. However, it is challenging to apply functional programming with combinators to complex circuits with latency-insensitive interfaces such as valid/ready interfaces due to the cyclic nature of their forward and backward ports. In this work, we present ShakeFlow: the first functional hardware description language supporting latency-insensitive interface combinators. ShakeFlow provides extensible support for custom interfaces and combinators and a compiler to synthesizable Verilog and FIRRTL. We port a part of the BaseJump STL library and the Corundum 100Gbps NIC from (System)Verilog to ShakeFlow, reducing the code size by 38% and 26%, respectively. By experimenting with Corundum, we demonstrate that ShakeFlow is capable of designing realistic circuits, and porting to ShakeFlow does not incur significant resource and performance overhead. @InProceedings{ASPLOS23p702, author = {Sungsoo Han and Minseong Jang and Jeehoon Kang}, title = {ShakeFlow: Functional Hardware Description with Latency-Insensitive Interface Combinators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {702--717}, doi = {10.1145/3575693.3575701}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Kannan, Tejas	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version
	Kao, Sheng-Chun	ASPLOS '23: "FLAT: An Optimized Dataflow ..." FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks Sheng-Chun Kao, Suvinay Subramanian, Gaurav Agrawal, Amir Yazdanbakhsh, and Tushar Krishna (Georgia Institute of Technology, USA; Google, USA; Microsoft, USA) Attention mechanisms, primarily designed to capture pairwise correlations between words, have become the backbone of machine learning, expanding beyond natural language processing into other domains. This growth in adaptation comes at the cost of prohibitively large memory requirements and computational complexity, especially at higher number of input elements. This limitation is due to inherently limited data reuse opportunities and quadratic growth in memory footprints, leading to severe memory-boundedness and limited scalability of input elements. This work addresses these challenges by devising a tailored dataflow optimization, called FLAT, for attention mechanisms without altering their functionality. This dataflow processes costly attention operations through a unique fusion mechanism, transforming the memory footprint quadratic growth to merely a linear one. To realize the full potential of this bespoke mechanism, we propose a tiling approach to enhance the data reuse across attention operations. Our method both mitigates the off-chip bandwidth bottleneck as well as reduces the on-chip memory requirement. FLAT delivers 1.94x (1.76x) speedup and 49% and (42%) of energy savings compared to the state-of-the-art Edge (Cloud) accelerators with no customized dataflow optimization. When on-chip resources are scarce (20 KB-200 KB), FLAT yields, on average, 1.5x end-to-end latency reduction across a diverse range of conventional attention-based models with input sequence lengths ranging from 512-token to 64K-token. Our evaluations demonstrate that state-of-the-art DNN dataflow applied to attention operations reach the efficiency limit for inputs above 512 elements. In contrast, FLAT unblocks transformer models for inputs with up to 64K elements. @InProceedings{ASPLOS23p295, author = {Sheng-Chun Kao and Suvinay Subramanian and Gaurav Agrawal and Amir Yazdanbakhsh and Tushar Krishna}, title = {FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {295--310}, doi = {10.1145/3575693.3575747}, year = {2023}, } Publisher's Version
	Kastner, Ryan	ASPLOS '23: "Junkyard Computing: Repurposing ..." Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon Jennifer Switzer, Gabriel Marcano, Ryan Kastner, and Pat Pannuto (University of California at San Diego, USA) 1.5 billion smartphones are sold annually, and most are decommissioned less than two years later. Most of these unwanted smartphones are neither discarded nor recycled but languish in junk drawers and storage units. This computational stockpile represents a substantial wasted potential: modern smartphones have increasingly high-performance and energy-efficient processors, extensive networking capabilities, and a reliable built-in power supply. This project studies the ability to reuse smartphones as "junkyard computers." Junkyard computers grow global computing capacity by extending device lifetimes, which supplants the manufacture of new devices. We show that the capabilities of even decade-old smartphones are within those demanded by modern cloud microservices and discuss how to combine phones to perform increasingly complex tasks. We describe how current operation-focused metrics do not capture the actual carbon costs of compute. We propose Computational Carbon Intensity---a performance metric that balances the continued service of older devices with the superlinear runtime improvements of newer machines. We use this metric to redefine device service lifetime in terms of carbon efficiency. We develop a cloudlet of reused Pixel 3A phones. We analyze the carbon benefits of deploying large, end-to-end microservice-based applications on these smartphones. Finally, we describe system architectures and associated challenges to scale to cloudlets with hundreds and thousands of smartphones. @InProceedings{ASPLOS23p400, author = {Jennifer Switzer and Gabriel Marcano and Ryan Kastner and Pat Pannuto}, title = {Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {400--412}, doi = {10.1145/3575693.3575710}, year = {2023}, } Publisher's Version
	Kazhamiaka, Fiodar	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Ketkar, Mahesh	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Khan, Sharjeel	ASPLOS '23: "Decker: Attack Surface Reduction ..." Decker: Attack Surface Reduction via On-Demand Code Mapping Chris Porter, Sharjeel Khan, and Santosh Pande (Georgia Institute of Technology, USA) Modern code reuse attacks take full advantage of bloated software. Attackers piece together short sequences of instructions in otherwise benign code to carry out malicious actions. Mitigating these reusable code snippets, known as gadgets, has become one of the prime focuses of attack surface reduction research. While some debloating techniques remove parts of software that contain such gadgets, other methods focus on making them unusable by breaking up chains of them, thereby substantially diminishing the possibility of code reuse attacks. Third-party libraries are another main focus, because they exhibit a high number of vulnerabilities, but recently, techniques have emerged that deal with whole applications. Attack surface reduction efforts have typically tried to eliminate such attacks by subsetting (debloating) the application, e.g. via user-specified inputs, configurations, or features to achieve high gadget reductions. However, such techniques suffer from the limitations of soundness, i.e. the software might crash during no-attack executions on regular inputs, or they may be conservative and leave a large amount of attack surface untackled. In this work we present a general, whole-program attack surface reduction technique called Decker that significantly reduces gadgets which are accessible to an attacker during an execution phase (called a deck) and has minor performance degradation. Decker requires no user inputs and leaves all features intact. It uses static analysis to determine key function sets that should be enabled/disabled at runtime. The runtime system then enables these function sets at the specified program points during execution. On SPEC CPU 2017, our framework achieves 73.2% total gadget reduction with 5.2% average slowdown. On 10 GNU coreutils applications, it achieves 87.2% reduction and negligible slowdown. On the nginx server it achieves 80.3% reduction with 2% slowdown. We also provide a gadget chain-breaking case study, including detailed JOP gadget metrics on both Linux and Windows, and show that our framework breaks the shell-spawning chain in all cases. @InProceedings{ASPLOS23p192, author = {Chris Porter and Sharjeel Khan and Santosh Pande}, title = {Decker: Attack Surface Reduction via On-Demand Code Mapping}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {192--206}, doi = {10.1145/3575693.3575734}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Kim, Hyung-Jin	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Kim, Jinpyo	ASPLOS '23: "DeepUM: Tensor Migration and ..." DeepUM: Tensor Migration and Prefetching in Unified Memory Jaehoon Jung, Jinpyo Kim, and Jaejin Lee (Moreh, South Korea; Seoul National University, South Korea) Deep neural networks (DNNs) are continuing to get wider and deeper. As a result, it requires a tremendous amount of GPU memory and computing power. In this paper, we propose a framework called DeepUM that exploits CUDA Unified Memory (UM) to allow GPU memory oversubscription for DNNs. While UM allows memory oversubscription using a page fault mechanism, page migration introduces enormous overhead. DeepUM uses a new correlation prefetching technique to hide the page migration overhead. It is fully automatic and transparent to users. We also propose two optimization techniques to minimize the GPU fault handling time. We evaluate the performance of DeepUM using nine large-scale DNNs from MLPerf, PyTorch examples, and Hugging Face and compare its performance with six state-of-the-art GPU memory swapping approaches. The evaluation result indicates that DeepUM is very effective for GPU memory oversubscription and can handle larger models that other approaches fail to handle. @InProceedings{ASPLOS23p207, author = {Jaehoon Jung and Jinpyo Kim and Jaejin Lee}, title = {DeepUM: Tensor Migration and Prefetching in Unified Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {207--221}, doi = {10.1145/3575693.3575736}, year = {2023}, } Publisher's Version
	Kim, Thomas	ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Kim, Wook-Hee	ASPLOS '23: "Prism: Optimizing Key-Value ..." Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices Yongju Song, Wook-Hee Kim, Sumit Kumar Monga, Changwoo Min, and Young Ik Eom (Sungkyunkwan University, South Korea; Konkuk University, South Korea; Virginia Tech, USA) As data generation has been on an upward trend, storing vast volumes of data cost-effectively as well as efficiently accessing them is paramount. At the same time, today's storage landscape continues to diversify, from high-bandwidth storage devices such as NVMe SSDs to low-latency non-volatile memory (e.g., Intel Optane DCPMM). These heterogeneous storage devices have the potential to deliver high performance in terms of bandwidth and latency with cost efficiency, while achieving the performance and cost targets together still remains a challenging problem. We provide our solution, Prism, a novel key-value store that utilizes modern heterogeneous storage devices. Prism uses heterogeneous storage devices synergistically to harness the advantages of each storage device while suppressing their downsides. We devise new techniques to balance the latency-bandwidth tradeoff when reading from SSD. For ensuring multicore scalability and crash consistency of data across heterogeneous storage media, Prism proposes cross-storage concurrency control and cross-storage crash consistency protocols. Our evaluation shows that Prism outperforms state-of-the-art key-value stores by up to 13.1× with significantly lower tail latency. @InProceedings{ASPLOS23p588, author = {Yongju Song and Wook-Hee Kim and Sumit Kumar Monga and Changwoo Min and Young Ik Eom}, title = {Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {588--602}, doi = {10.1145/3575693.3575722}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Kim, Youngsok	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Krishna, Tushar	ASPLOS '23: "FLAT: An Optimized Dataflow ..." FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks Sheng-Chun Kao, Suvinay Subramanian, Gaurav Agrawal, Amir Yazdanbakhsh, and Tushar Krishna (Georgia Institute of Technology, USA; Google, USA; Microsoft, USA) Attention mechanisms, primarily designed to capture pairwise correlations between words, have become the backbone of machine learning, expanding beyond natural language processing into other domains. This growth in adaptation comes at the cost of prohibitively large memory requirements and computational complexity, especially at higher number of input elements. This limitation is due to inherently limited data reuse opportunities and quadratic growth in memory footprints, leading to severe memory-boundedness and limited scalability of input elements. This work addresses these challenges by devising a tailored dataflow optimization, called FLAT, for attention mechanisms without altering their functionality. This dataflow processes costly attention operations through a unique fusion mechanism, transforming the memory footprint quadratic growth to merely a linear one. To realize the full potential of this bespoke mechanism, we propose a tiling approach to enhance the data reuse across attention operations. Our method both mitigates the off-chip bandwidth bottleneck as well as reduces the on-chip memory requirement. FLAT delivers 1.94x (1.76x) speedup and 49% and (42%) of energy savings compared to the state-of-the-art Edge (Cloud) accelerators with no customized dataflow optimization. When on-chip resources are scarce (20 KB-200 KB), FLAT yields, on average, 1.5x end-to-end latency reduction across a diverse range of conventional attention-based models with input sequence lengths ranging from 512-token to 64K-token. Our evaluations demonstrate that state-of-the-art DNN dataflow applied to attention operations reach the efficiency limit for inputs above 512 elements. In contrast, FLAT unblocks transformer models for inputs with up to 64K elements. @InProceedings{ASPLOS23p295, author = {Sheng-Chun Kao and Suvinay Subramanian and Gaurav Agrawal and Amir Yazdanbakhsh and Tushar Krishna}, title = {FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {295--310}, doi = {10.1145/3575693.3575747}, year = {2023}, } Publisher's Version
	Kuang, Haozhao	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Kumar, Snehasish	ASPLOS '23: "Propeller: A Profile Guided, ..." Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications Han Shen, Krzysztof Pszeniczny, Rahman Lavaee, Snehasish Kumar, Sriraman Tallam, and Xinliang David Li (Google, USA; Google, Switzerland) While profile guided optimizations (PGO) and link time optimiza-tions (LTO) have been widely adopted, post link optimizations (PLO)have languished until recently when researchers demonstrated that late injection of profiles can yield significant performance improvements. However, the disassembly-driven, monolithic design of post link optimizers face scaling challenges with large binaries andis at odds with distributed build systems. To reconcile and enable post link optimizations within a distributed build environment, we propose Propeller, a relinking optimizer for warehouse scale work-loads. To enable flexible code layout optimizations, we introduce basic block sections, a novel linker abstraction. Propeller uses basic block sections to enable a new approach to PLO without disassembly. Propeller achieves scalability by relinking the binary using precise profiles instead of rewriting the binary. The overhead of relinking is lowered by caching and leveraging distributed compiler actions during code generation. Propeller has been deployed to production at Google with over tens of millions of cores executing Propeller optimized code at any time. An evaluation of internal warehouse-scale applications show Propeller improves performance by 1.1% to 8% beyond PGO and ThinLTO. Compiler tools such as Clang improve by 7% while MySQL improves by 1%. Compared to the state of the art binary optimizer, Propeller achieves comparable performance while lowering memory overheads by 30%-70% on large benchmarks. @InProceedings{ASPLOS23p617, author = {Han Shen and Krzysztof Pszeniczny and Rahman Lavaee and Snehasish Kumar and Sriraman Tallam and Xinliang David Li}, title = {Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {617--631}, doi = {10.1145/3575693.3575727}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Kurniawan, Daniar H.	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version
	Lai, Ruihang	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Lao, ChonLam	ASPLOS '23: "A Generic Service to Provide ..." A Generic Service to Provide In-Network Aggregation for Key-Value Streams Yongchao He, Wenfei Wu, Yanfang Le, Ming Liu, and ChonLam Lao (Tsinghua University, China; Peking University, China; Intel, USA; University of Wisconsin-Madison, USA; Harvard University, USA) Key-value stream aggregation is a common operation in distributed systems, which requires intensive computation and network resources. We propose a generic in-network aggregation service for key-value streams, ASK, to accelerate the aggregation operations in diverse distributed applications. ASK is a switch-host co-designed system, where the programmable switch provides a best-effort aggregation service, and the host runs a daemon to interact with applications. ASK makes in-depth optimization tailored to traffic characteristics, hardware restrictions, and network unreliable natures: it vectorizes multiple key-value tuples’ aggregation of one packet in one switch pipeline pass, which improves the per-host’s goodput; it develops a lightweight reliability mechanism for key-value stream’s asynchronous aggregation, which guarantees computation correctness; it designs a hot-key agnostic prioritization for key-skewed workloads, which improves the switch memory utilization. We prototype ASK and use it to support Spark and BytePS. The evaluation shows that ASK could accelerate pure key-value aggregation tasks by up to 155 times and big data jobs by 3-5 times, and be backward compatible with existing INA-empowered distributed training solutions with the same speedup. @InProceedings{ASPLOS23p33, author = {Yongchao He and Wenfei Wu and Yanfang Le and Ming Liu and ChonLam Lao}, title = {A Generic Service to Provide In-Network Aggregation for Key-Value Streams}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {33--47}, doi = {10.1145/3575693.3575708}, year = {2023}, } Publisher's Version
	Lavaee, Rahman	ASPLOS '23: "Propeller: A Profile Guided, ..." Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications Han Shen, Krzysztof Pszeniczny, Rahman Lavaee, Snehasish Kumar, Sriraman Tallam, and Xinliang David Li (Google, USA; Google, Switzerland) While profile guided optimizations (PGO) and link time optimiza-tions (LTO) have been widely adopted, post link optimizations (PLO)have languished until recently when researchers demonstrated that late injection of profiles can yield significant performance improvements. However, the disassembly-driven, monolithic design of post link optimizers face scaling challenges with large binaries andis at odds with distributed build systems. To reconcile and enable post link optimizations within a distributed build environment, we propose Propeller, a relinking optimizer for warehouse scale work-loads. To enable flexible code layout optimizations, we introduce basic block sections, a novel linker abstraction. Propeller uses basic block sections to enable a new approach to PLO without disassembly. Propeller achieves scalability by relinking the binary using precise profiles instead of rewriting the binary. The overhead of relinking is lowered by caching and leveraging distributed compiler actions during code generation. Propeller has been deployed to production at Google with over tens of millions of cores executing Propeller optimized code at any time. An evaluation of internal warehouse-scale applications show Propeller improves performance by 1.1% to 8% beyond PGO and ThinLTO. Compiler tools such as Clang improve by 7% while MySQL improves by 1%. Compared to the state of the art binary optimizer, Propeller achieves comparable performance while lowering memory overheads by 30%-70% on large benchmarks. @InProceedings{ASPLOS23p617, author = {Han Shen and Krzysztof Pszeniczny and Rahman Lavaee and Snehasish Kumar and Sriraman Tallam and Xinliang David Li}, title = {Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {617--631}, doi = {10.1145/3575693.3575727}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Le, Yanfang	ASPLOS '23: "A Generic Service to Provide ..." A Generic Service to Provide In-Network Aggregation for Key-Value Streams Yongchao He, Wenfei Wu, Yanfang Le, Ming Liu, and ChonLam Lao (Tsinghua University, China; Peking University, China; Intel, USA; University of Wisconsin-Madison, USA; Harvard University, USA) Key-value stream aggregation is a common operation in distributed systems, which requires intensive computation and network resources. We propose a generic in-network aggregation service for key-value streams, ASK, to accelerate the aggregation operations in diverse distributed applications. ASK is a switch-host co-designed system, where the programmable switch provides a best-effort aggregation service, and the host runs a daemon to interact with applications. ASK makes in-depth optimization tailored to traffic characteristics, hardware restrictions, and network unreliable natures: it vectorizes multiple key-value tuples’ aggregation of one packet in one switch pipeline pass, which improves the per-host’s goodput; it develops a lightweight reliability mechanism for key-value stream’s asynchronous aggregation, which guarantees computation correctness; it designs a hot-key agnostic prioritization for key-skewed workloads, which improves the switch memory utilization. We prototype ASK and use it to support Spark and BytePS. The evaluation shows that ASK could accelerate pure key-value aggregation tasks by up to 155 times and big data jobs by 3-5 times, and be backward compatible with existing INA-empowered distributed training solutions with the same speedup. @InProceedings{ASPLOS23p33, author = {Yongchao He and Wenfei Wu and Yanfang Le and Ming Liu and ChonLam Lao}, title = {A Generic Service to Provide In-Network Aggregation for Key-Value Streams}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {33--47}, doi = {10.1145/3575693.3575708}, year = {2023}, } Publisher's Version
	Lee, Benjamin	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Lee, Jaejin	ASPLOS '23: "DeepUM: Tensor Migration and ..." DeepUM: Tensor Migration and Prefetching in Unified Memory Jaehoon Jung, Jinpyo Kim, and Jaejin Lee (Moreh, South Korea; Seoul National University, South Korea) Deep neural networks (DNNs) are continuing to get wider and deeper. As a result, it requires a tremendous amount of GPU memory and computing power. In this paper, we propose a framework called DeepUM that exploits CUDA Unified Memory (UM) to allow GPU memory oversubscription for DNNs. While UM allows memory oversubscription using a page fault mechanism, page migration introduces enormous overhead. DeepUM uses a new correlation prefetching technique to hide the page migration overhead. It is fully automatic and transparent to users. We also propose two optimization techniques to minimize the GPU fault handling time. We evaluate the performance of DeepUM using nine large-scale DNNs from MLPerf, PyTorch examples, and Hugging Face and compare its performance with six state-of-the-art GPU memory swapping approaches. The evaluation result indicates that DeepUM is very effective for GPU memory oversubscription and can handle larger models that other approaches fail to handle. @InProceedings{ASPLOS23p207, author = {Jaehoon Jung and Jinpyo Kim and Jaejin Lee}, title = {DeepUM: Tensor Migration and Prefetching in Unified Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {207--221}, doi = {10.1145/3575693.3575736}, year = {2023}, } Publisher's Version
	Lee, Jinho	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Lee, Scott	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Lei, Kelun	ASPLOS '23: "VClinic: A Portable and Efficient ..." VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers Xin You, Hailong Yang, Kelun Lei, Zhongzhi Luan, and Depei Qian (Beihang University, China) Fine-grained value profilers reveal a promising way to accurately detect value-related software inefficiencies with binary instrumentation. Due to the architecture-dependent implementation details of binary instrumentation, existing value profilers suffer from poor portability as well as high engineering efforts to achieve efficiency across platforms. In this paper, we propose VClinic, a portable and efficient fine-grained value profiling framework for analyzing highly optimized binaries on both X86 and ARM platforms. VClinic exploits operand-centric two-level designs in its implementation to provide the common building blocks required for value profilers. By constructing four representative value profilers with VClinic, we demonstrate that VClinic can ease the development of value profilers with portability and efficiency across platforms. Guided by the value profilers built upon VClinic, we can achieve up to 89.94% and 74.66% speedup for real-world programs on X86 and ARM platforms, respectively. @InProceedings{ASPLOS23p892, author = {Xin You and Hailong Yang and Kelun Lei and Zhongzhi Luan and Depei Qian}, title = {VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {892--904}, doi = {10.1145/3575693.3576934}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Leng, Jingwen	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Levien, Raph	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Levine, Reese	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Li, Chao	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Li, Dong	ASPLOS '23: "Betty: Enabling Large-Scale ..." Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning Shuangyan Yang, Minjia Zhang, Wenqian Dong, and Dong Li (University of California at Merced, USA; Microsoft Research, USA; Florida International University, USA) The Graph Neural Network (GNN) is showing outstanding results in improving the performance of graph-based applications. Recent studies demonstrate that GNN performance can be boosted via using more advanced aggregators, deeper aggregation depth, larger sampling rate, etc. While leading to promising results, the improvements come at a cost of significantly increased memory footprint, easily exceeding GPU memory capacity. In this paper, we introduce a method, Betty, to make GNN training more scalable and accessible via batch-level partitioning. Different from DNN training, a mini-batch in GNN has complex dependencies between input features and output labels, making batch-level partitioning difficult. Betty introduces two noveltechniques, redundancy-embedded graph (REG) partitioning and memory-aware partitioning, to effectively mitigate the redundancy and load imbalances issues across the partitions. Our evaluation of large-scale real-world datasets shows that Betty can significantly mitigate the memory bottleneck, enabling scalable GNN training with much deeper aggregation depths, larger sampling rate, larger training batch sizes, together with more advanced aggregators, with a few as a single GPU. @InProceedings{ASPLOS23p103, author = {Shuangyan Yang and Minjia Zhang and Wenqian Dong and Dong Li}, title = {Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {103--117}, doi = {10.1145/3575693.3575725}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Li, Huaicheng	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version ASPLOS '23: "RAIZN: Redundant Array of ..." RAIZN: Redundant Array of Independent Zoned Namespaces Thomas Kim, Jekyeom Jeon, Nikhil Arora, Huaicheng Li, Michael Kaminsky, David G. Andersen, Gregory R. Ganger, George Amvrosiadis, and Matias Bjørling (Carnegie Mellon University, USA; Enriched Ag, USA; Western Digital, Denmark) Zoned Namespace (ZNS) SSDs are the latest evolution of host-managed flash storage, enabling improved performance at a lower cost-per-byte than traditional block interface (conventional) SSDs. To date, there is no support for arranging these new devices in arrays that offer increased throughput and reliability (RAID). We identify key challenges in designing redundant ZNS SSD arrays, such as managing metadata updates and persisting partial stripe writes in the absence of overwrite support from the device. We present RAIZN, a logical volume manager that exposes a ZNS interface and stripes data and parity across ZNS SSDs. RAIZN provides more stable throughput and lower tail latencies than an mdraid array of conventional SSDs based on the same hardware platform. RAIZN achieves superior performance because device-level garbage collection slows down conventional SSDs. We confirm that the benefits of RAIZN translate to higher layers by adapting the F2FS file system, RocksDB key-value store, and MySQL database to work with ZNS and leverage its benefits by closely controlling garbage collection. Compared to arrays of conventional SSDs experiencing on-device garbage collection, RAIZN leverages the ZNS interface to maintain consistent performance with up to 14× higher throughput and lower tail latency. @InProceedings{ASPLOS23p660, author = {Thomas Kim and Jekyeom Jeon and Nikhil Arora and Huaicheng Li and Michael Kaminsky and David G. Andersen and Gregory R. Ganger and George Amvrosiadis and Matias Bjørling}, title = {RAIZN: Redundant Array of Independent Zoned Namespaces}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {660--673}, doi = {10.1145/3575693.3575746}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Li, Jiaxiang	ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Li, Jinyang	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Li, Shaobo	ASPLOS '23: "LeaFTL: A Learning-Based Flash ..." LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives Jinghan Sun, Shaobo Li, Yunxin Sun, Chao Sun, Dejan Vucinic, and Jian Huang (University of Illinois at Urbana-Champaign, USA; ETH Zurich, Switzerland; Western Digital Research, USA) In modern solid-state drives (SSDs), the indexing of flash pages is a critical component in their storage controllers. It not only affects the data access performance, but also determines the efficiency of the precious in-device DRAM resource. A variety of address mapping schemes and optimizations have been proposed. However, most of them were developed with human-driven heuristics. In this paper, we present a learning-based flash translation layer (FTL), named LeaFTL, which learns the address mapping to tolerate dynamic data access patterns via linear regression at runtime. By grouping a large set of mapping entries into a learned segment, it significantly reduces the memory footprint of the address mapping table, which further benefits the data caching in SSD controllers. LeaFTL also employs various optimization techniques, including out-of-band metadata verification to tolerate mispredictions, optimized flash allocation, and dynamic compaction of learned index segments. We implement LeaFTL with both a validated SSD simulator and a real open-channel SSD board. Our evaluation with various storage workloads demonstrates that LeaFTL saves the memory consumption of the mapping table by 2.9× and improves the storage performance by 1.4× on average, in comparison with state-of-the-art FTL schemes. @InProceedings{ASPLOS23p442, author = {Jinghan Sun and Shaobo Li and Yunxin Sun and Chao Sun and Dejan Vucinic and Jian Huang}, title = {LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {442--456}, doi = {10.1145/3575693.3575744}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Li, Xinliang David	ASPLOS '23: "Propeller: A Profile Guided, ..." Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications Han Shen, Krzysztof Pszeniczny, Rahman Lavaee, Snehasish Kumar, Sriraman Tallam, and Xinliang David Li (Google, USA; Google, Switzerland) While profile guided optimizations (PGO) and link time optimiza-tions (LTO) have been widely adopted, post link optimizations (PLO)have languished until recently when researchers demonstrated that late injection of profiles can yield significant performance improvements. However, the disassembly-driven, monolithic design of post link optimizers face scaling challenges with large binaries andis at odds with distributed build systems. To reconcile and enable post link optimizations within a distributed build environment, we propose Propeller, a relinking optimizer for warehouse scale work-loads. To enable flexible code layout optimizations, we introduce basic block sections, a novel linker abstraction. Propeller uses basic block sections to enable a new approach to PLO without disassembly. Propeller achieves scalability by relinking the binary using precise profiles instead of rewriting the binary. The overhead of relinking is lowered by caching and leveraging distributed compiler actions during code generation. Propeller has been deployed to production at Google with over tens of millions of cores executing Propeller optimized code at any time. An evaluation of internal warehouse-scale applications show Propeller improves performance by 1.1% to 8% beyond PGO and ThinLTO. Compiler tools such as Clang improve by 7% while MySQL improves by 1%. Compared to the state of the art binary optimizer, Propeller achieves comparable performance while lowering memory overheads by 30%-70% on large benchmarks. @InProceedings{ASPLOS23p617, author = {Han Shen and Krzysztof Pszeniczny and Rahman Lavaee and Snehasish Kumar and Sriraman Tallam and Xinliang David Li}, title = {Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {617--631}, doi = {10.1145/3575693.3575727}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Li, Yong	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Li, Yueying	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Li, Zhiyao	ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Liang, Mingyu	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Liang, Qianlin	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Liao, Xiaofei	ASPLOS '23: "Revisiting Log-Structured ..." Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems Zhuohui Duan, Jiabo Yao, Haikun Liu, Xiaofei Liao, Hai Jin, and Yu Zhang (Huazhong University of Science and Technology, China) We present MioDB, a novel LSM-tree based key-value (KV) store system designed to fully exploit the advantages of byte-addressable non-volatile memories (NVMs). Our experimental studies reveal that the performance bottleneck of LSM-tree based KV stores using NVMs mainly stems from (1) costly data serialization/deserialization across memory and storage, and (2) unbalanced speed between memory-to-disk data flushing and on-disk data compaction. They may cause unpredictable performance degradation due to write stalls and write amplification. To address these problems, we advocate byte-addressable and persistent skip lists to replace the on-disk data structure of LSM-tree, and design four novel techniques to make the best use of fast NVMs. First, we propose one-piece flushing to minimize the cost of data serialization from DRAM to NVM. Second, we exploit an elastic NVM buffer with multiple levels and zero-copy compaction to eliminate write stalls and reduce write amplification. Third, we propose parallel compaction to orchestrate data flushing and compactions across all levels of LSM-trees. Finally, MioDB increases the depth of LSM-tree and exploits bloom filters to improve the read performance. Our extensive experimental studies demonstrate that MioDB achieves 17.1× and 21.7× lower 99.9th percentile latency, 8.3× and 2.5× higher random write throughput, and up to 5× and 4.9× lower write amplification compared with the state-of-the-art NoveLSM and MatrixKV, respectively. @InProceedings{ASPLOS23p674, author = {Zhuohui Duan and Jiabo Yao and Haikun Liu and Xiaofei Liao and Hai Jin and Yu Zhang}, title = {Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {674--687}, doi = {10.1145/3575693.3575715}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Lin, Felix Xiaozhu	ASPLOS '23: "STI: Turbocharge NLP Inference ..." STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining Liwei Guo, Wonkyo Choe, and Felix Xiaozhu Lin (University of Virginia, USA) Natural Language Processing (NLP) inference is seeing increasing adoption by mobile applications, where on-device inference is desirable for crucially preserving user data privacy and avoiding network roundtrips. Yet, the unprecedented size of an NLP model stresses both latency and memory, creating a tension between the two key resources of a mobile device. To meet a target latency, holding the whole model in memory launches execution as soon as possible but increases one app’s memory footprints by several times, limiting its benefits to only a few inferences before being recycled by mobile memory management. On the other hand, loading the model from storage on demand incurs IO as long as a few seconds, far exceeding the delay range satisfying to a user; pipelining layerwise model loading and execution does not hide IO either, due to the high skewness between IO and computation delays. To this end, we propose Speedy Transformer Inference (STI). Built on the key idea of maximizing IO/compute resource utilization on the most important parts of a model, STI reconciles the latency v.s. memory tension via two novel techniques. First, model sharding. STI manages model parameters as independently tunable shards, and profiles their importance to accuracy. Second, elastic pipeline planning with a preload buffer. STI instantiates an IO/compute pipeline and uses a small buffer for preload shards to bootstrap execution without stalling at early stages; it judiciously selects, tunes, and assembles shards per their importance for resource-elastic execution, maximizing inference accuracy. Atop two commodity SoCs, we build STI and evaluate it against a wide range of NLP tasks, under a practical range of target latencies, and on both CPU and GPU. We demonstrate that STI delivers high accuracies with 1–2 orders of magnitude lower memory, outperforming competitive baselines. @InProceedings{ASPLOS23p791, author = {Liwei Guo and Wonkyo Choe and Felix Xiaozhu Lin}, title = {STI: Turbocharge NLP Inference at the Edge via Elastic Pipelining}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {791--803}, doi = {10.1145/3575693.3575698}, year = {2023}, } Publisher's Version
	Lin, Jinkun	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Lin, Sophia Fuhui	ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Lin, Wei	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Lin, Wuwei	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Lis, Mieszko	ASPLOS '23: "HuffDuff: Stealing Pruned ..." HuffDuff: Stealing Pruned DNNs from Sparse Accelerators Dingqing Yang, Prashant J. Nair, and Mieszko Lis (University of British Columbia, Canada) Deep learning models are a valuable “secret sauce” that confers a significant competitive advantage. Many models are never visible to the user and even publicly known state-of-the-art models are either completely proprietary or only accessible via access-controlled APIs. Increasingly, these models run directly on the edge, often using a low-power DNN accelerator. This makes models particularly vulnerable, as an attacker with physical access can exploit side channels like off-chip memory access volumes. Indeed, prior work has shown that this channel can be used to steal dense DNNs from edge devices by correlating data transfer volumes with layer geometry. Unfortunately, prior techniques become intractable when the model is sparse in either weights or activations because off-chip transfers no longer correspond exactly to layer dimensions. Could it be that the many mobile-class sparse accelerators are inherently safe from this style of attack? In this paper, we show that it is feasible to steal a pruned DNN model architecture from a mobile-class sparse accelerator using the DRAM access volume channel. We describe HuffDuff, an attack scheme with two novel techniques that leverage (i) the boundary effect present in CONV layers, and (ii) the timing side channel of on-the-fly activation compression. Together, these techniques dramatically reduce the space of possible model architectures up to 94 orders of magnitude, resulting in fewer than 100 candidate models — a number that can be feasibly tested. Finally, we sample network instances from our solution space and show that (i) our solutions reach the victim accuracy under the iso-footprint constraint, and (ii) significantly improve black-box targeted attack success rates. @InProceedings{ASPLOS23p385, author = {Dingqing Yang and Prashant J. Nair and Mieszko Lis}, title = {HuffDuff: Stealing Pruned DNNs from Sparse Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {385--399}, doi = {10.1145/3575693.3575738}, year = {2023}, } Publisher's Version
	Litteken, Andrew	ASPLOS '23: "Qompress: Efficient Compilation ..." Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction Andrew Litteken, Lennart Maximilian Seifert, Jason Chadwick, Natalia Nottingham, Frederic T. Chong, and Jonathan M. Baker (University of Chicago, USA) Quantum computing is in an era of limited resources. Current hardware lacks high fidelity gates, long coherence times, and the number of computational units required to perform meaningful computation. Contemporary quantum devices typically use a binary system, where each qubit exists in a superposition of the 0 and 1 states. However, it is often possible to access the 2 or even 3 states in the same physical unit by manipulating the system in different ways. In this work, we consider automatically encoding two qubits into one four-state ququart via a compression scheme. We use quantum optimal control to design efficient proof-of-concept gates that fully replicate standard qubit computation on these encoded qubits. We extend qubit compilation schemes to efficiently route qubits on an arbitrary mixed-radix system consisting of both qubits and ququarts, reducing communication and minimizing excess circuit execution time introduced by longer-duration ququart gates. In conjunction with these compilation strategies, we introduce several methods to find beneficial compressions, reducing circuit error due to computation and communication by up to 50 @InProceedings{ASPLOS23p646, author = {Andrew Litteken and Lennart Maximilian Seifert and Jason Chadwick and Natalia Nottingham and Frederic T. Chong and Jonathan M. Baker}, title = {Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {646--659}, doi = {10.1145/3575693.3575726}, year = {2023}, } Publisher's Version
	Liu, Congyu	ASPLOS '23: "KIT: Testing OS-Level Virtualization ..." KIT: Testing OS-Level Virtualization for Functional Interference Bugs Congyu Liu, Sishuai Gong, and Pedro Fonseca (Purdue University, USA) Container isolation is implemented through OS-level virtualization, such as Linux namespaces. Unfortunately, these mechanisms are extremely challenging to implement correctly and, in practice, suffer from functional interference bugs, which compromise container security. In particular, functional interference bugs allow an attacker to extract information from another container running on the same machine or impact its integrity by modifying kernel resources that are incorrectly isolated. Despite their impact, functional interference bugs in OS-level virtualization have received limited attention in part due to the challenges in detecting them. Instead of causing memory errors or crashes, many functional interference bugs involve hard-to-catch logic errors that silently produce semantically incorrect results. This paper proposes KIT, a dynamic testing framework that discovers functional interference bugs in OS-level virtualization mechanisms, such as Linux namespaces. The key idea of KIT is to detect inter-container functional interference by comparing the system call traces of a container across two executions, where it runs with and without the preceding execution of another container. To achieve high efficiency and accuracy, KIT includes two critical components: an efficient algorithm to generate test cases that exercise inter-container data flows and a system call trace analysis framework that detects functional interference bugs and clusters bug reports. KIT discovered 9 functional interference bugs in Linux kernel 5.13, of which 6 have been confirmed. All bugs are caused by logic errors, showing that this approach is able to detect hard-to-catch semantic bugs. @InProceedings{ASPLOS23p427, author = {Congyu Liu and Sishuai Gong and Pedro Fonseca}, title = {KIT: Testing OS-Level Virtualization for Functional Interference Bugs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {427--441}, doi = {10.1145/3575693.3575731}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Liu, Haikun	ASPLOS '23: "Revisiting Log-Structured ..." Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems Zhuohui Duan, Jiabo Yao, Haikun Liu, Xiaofei Liao, Hai Jin, and Yu Zhang (Huazhong University of Science and Technology, China) We present MioDB, a novel LSM-tree based key-value (KV) store system designed to fully exploit the advantages of byte-addressable non-volatile memories (NVMs). Our experimental studies reveal that the performance bottleneck of LSM-tree based KV stores using NVMs mainly stems from (1) costly data serialization/deserialization across memory and storage, and (2) unbalanced speed between memory-to-disk data flushing and on-disk data compaction. They may cause unpredictable performance degradation due to write stalls and write amplification. To address these problems, we advocate byte-addressable and persistent skip lists to replace the on-disk data structure of LSM-tree, and design four novel techniques to make the best use of fast NVMs. First, we propose one-piece flushing to minimize the cost of data serialization from DRAM to NVM. Second, we exploit an elastic NVM buffer with multiple levels and zero-copy compaction to eliminate write stalls and reduce write amplification. Third, we propose parallel compaction to orchestrate data flushing and compactions across all levels of LSM-trees. Finally, MioDB increases the depth of LSM-tree and exploits bloom filters to improve the read performance. Our extensive experimental studies demonstrate that MioDB achieves 17.1× and 21.7× lower 99.9th percentile latency, 8.3× and 2.5× higher random write throughput, and up to 5× and 4.9× lower write amplification compared with the state-of-the-art NoveLSM and MatrixKV, respectively. @InProceedings{ASPLOS23p674, author = {Zhuohui Duan and Jiabo Yao and Haikun Liu and Xiaofei Liao and Hai Jin and Yu Zhang}, title = {Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {674--687}, doi = {10.1145/3575693.3575715}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Liu, Jiawei	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Liu, Ming	ASPLOS '23: "A Generic Service to Provide ..." A Generic Service to Provide In-Network Aggregation for Key-Value Streams Yongchao He, Wenfei Wu, Yanfang Le, Ming Liu, and ChonLam Lao (Tsinghua University, China; Peking University, China; Intel, USA; University of Wisconsin-Madison, USA; Harvard University, USA) Key-value stream aggregation is a common operation in distributed systems, which requires intensive computation and network resources. We propose a generic in-network aggregation service for key-value streams, ASK, to accelerate the aggregation operations in diverse distributed applications. ASK is a switch-host co-designed system, where the programmable switch provides a best-effort aggregation service, and the host runs a daemon to interact with applications. ASK makes in-depth optimization tailored to traffic characteristics, hardware restrictions, and network unreliable natures: it vectorizes multiple key-value tuples’ aggregation of one packet in one switch pipeline pass, which improves the per-host’s goodput; it develops a lightweight reliability mechanism for key-value stream’s asynchronous aggregation, which guarantees computation correctness; it designs a hot-key agnostic prioritization for key-skewed workloads, which improves the switch memory utilization. We prototype ASK and use it to support Spark and BytePS. The evaluation shows that ASK could accelerate pure key-value aggregation tasks by up to 155 times and big data jobs by 3-5 times, and be backward compatible with existing INA-empowered distributed training solutions with the same speedup. @InProceedings{ASPLOS23p33, author = {Yongchao He and Wenfei Wu and Yanfang Le and Ming Liu and ChonLam Lao}, title = {A Generic Service to Provide In-Network Aggregation for Key-Value Streams}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {33--47}, doi = {10.1145/3575693.3575708}, year = {2023}, } Publisher's Version
	Liu, Shuo	ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Liu, Xiangwen	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Liu, Xuanzhe	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Liu, Yizhi	ASPLOS '23: "Hidet: Task-Mapping Programming ..." Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding, Cody Hao Yu, Bojian Zheng, Yizhi Liu, Yida Wang, and Gennady Pekhimenko (University of Toronto, Canada; Vector Institute, Canada; Amazon Web Services, USA) As deep learning models nowadays are widely adopted by both cloud services and edge devices, reducing the latency of deep learning model inferences becomes crucial to provide efficient model serving. However, it is challenging to develop efficient tensor programs for deep learning operators due to the high complexity of modern accelerators (e.g., NVIDIA GPUs and Google TPUs) and the rapidly growing number of operators. Deep learning compilers, such as Apache TVM, adopt declarative scheduling primitives to lower the bar of developing tensor programs. However, we show that this approach is insufficient to cover state-of-the-art tensor program optimizations (e.g., double buffering). In this paper, we propose to embed the scheduling process into tensor programs and use dedicated mappings, called task mappings, to define the computation assignment and ordering directly in the tensor programs. This new approach greatly enriches the expressible optimizations by allowing developers to manipulate tensor programs at a much finer granularity (e.g., allowing program-statement-level optimizations). We call the proposed method the task-mapping programming paradigm. In addition, we propose a new post-scheduling fusion optimization that allows developers to focus on scheduling every single operator and automates the fusion after scheduling. It greatly reduces the engineering efforts for operator fusion. Our proposed paradigm also constructs an efficient hardware-centric schedule space, which is agnostic to the program input size and greatly reduces the tuning time. With the proposed paradigm, we implement a deep learning compiler Hidet. Extensive experiments on modern convolution and transformer models show that Hidet outperforms state-of-the-art DNN inference framework, ONNX Runtime, and compiler, TVM equipped with scheduler AutoTVM and Ansor, by up to 1.48x (1.22x on average). It also reduces the tuning time by 20x and 11x compared with AutoTVM and Ansor, respectively. We open-sourced hidet at https://www.github.com/hidet-org/hidet. @InProceedings{ASPLOS23p370, author = {Yaoyao Ding and Cody Hao Yu and Bojian Zheng and Yizhi Liu and Yida Wang and Gennady Pekhimenko}, title = {Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {370--384}, doi = {10.1145/3575693.3575702}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Lu, Shuwen	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Lu, Youyou	ASPLOS '23: "Mobius: Fine Tuning Large-Scale ..." Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers Yangyang Feng, Minhui Xie, Zijie Tian, Shuo Wang, Youyou Lu, and Jiwu Shu (Tsinghua University, China) Fine-tuning on cheap commodity GPU servers makes large-scale deep learning models benefit more people. However, the low inter-GPU communication bandwidth and pressing communication contention on the commodity GPU server obstruct training efficiency. In this paper, we present Mobius, a communication-efficient system for fine tuning large-scale models on commodity GPU servers. The key idea is a novel pipeline parallelism scheme enabling heterogeneous memory for large-scale model training, while bringing fewer communications than existing systems. Mobius partitions the model into stages and carefully schedules them between GPU memory and DRAM to overlap communication with computation. It formulates pipeline execution into a mixed-integer program problem to find the optimal pipeline partition. It also features a new stage-to-GPU mapping method termed cross mapping, to minimize communication contention. Experiments on various scale models and GPU topologies show that Mobius significantly reduces the training time by 3.8-5.1× compared with the prior art. @InProceedings{ASPLOS23p489, author = {Yangyang Feng and Minhui Xie and Zijie Tian and Shuo Wang and Youyou Lu and Jiwu Shu}, title = {Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {489--501}, doi = {10.1145/3575693.3575703}, year = {2023}, } Publisher's Version
	Luan, Zhongzhi	ASPLOS '23: "VClinic: A Portable and Efficient ..." VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers Xin You, Hailong Yang, Kelun Lei, Zhongzhi Luan, and Depei Qian (Beihang University, China) Fine-grained value profilers reveal a promising way to accurately detect value-related software inefficiencies with binary instrumentation. Due to the architecture-dependent implementation details of binary instrumentation, existing value profilers suffer from poor portability as well as high engineering efforts to achieve efficiency across platforms. In this paper, we propose VClinic, a portable and efficient fine-grained value profiling framework for analyzing highly optimized binaries on both X86 and ARM platforms. VClinic exploits operand-centric two-level designs in its implementation to provide the common building blocks required for value profilers. By constructing four representative value profilers with VClinic, we demonstrate that VClinic can ease the development of value profilers with portability and efficiency across platforms. Guided by the value profilers built upon VClinic, we can achieve up to 89.94% and 74.66% speedup for real-world programs on X86 and ARM platforms, respectively. @InProceedings{ASPLOS23p892, author = {Xin You and Hailong Yang and Kelun Lei and Zhongzhi Luan and Depei Qian}, title = {VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {892--904}, doi = {10.1145/3575693.3576934}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Ma, Weiliang	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Ma, Xiaosong	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Maeng, Kiwan	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Mailthody, Vikram Sharma	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Maleki, Saeed	ASPLOS '23: "MSCCLang: Microsoft Collective ..." MSCCLang: Microsoft Collective Communication Language Meghan Cowan, Saeed Maleki, Madanlal Musuvathi, Olli Saarikivi, and Yifan Xiong (Microsoft Research, USA; Microsoft Research, China) Machine learning models with millions or billions of parameters are increasingly trained and served on large multi-GPU systems. As models grow in size and execute on more GPUs, collective communication becomes a bottleneck. Custom collective algorithms optimized for both particular network topologies and application-specific communication patterns can alleviate this bottleneck and help these applications scale. However, implementing correct and efficient custom algorithms is challenging. This paper introduces MSCCLang, a system for programmable GPU communication. MSCCLang provides a domain specific language for writing collective communication algorithms and an optimizing compiler for lowering them to an executable form, which can be executed efficiently and flexibly in an interpreter-based runtime. We used MSCCLang to write novel collective algorithms for AllReduce and AllToAll that are up to 1.9× and 1.3× faster than hand-optimized implementations, respectively. @InProceedings{ASPLOS23p502, author = {Meghan Cowan and Saeed Maleki and Madanlal Musuvathi and Olli Saarikivi and Yifan Xiong}, title = {MSCCLang: Microsoft Collective Communication Language}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {502--514}, doi = {10.1145/3575693.3575724}, year = {2023}, } Publisher's Version
	Marcano, Gabriel	ASPLOS '23: "Junkyard Computing: Repurposing ..." Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon Jennifer Switzer, Gabriel Marcano, Ryan Kastner, and Pat Pannuto (University of California at San Diego, USA) 1.5 billion smartphones are sold annually, and most are decommissioned less than two years later. Most of these unwanted smartphones are neither discarded nor recycled but languish in junk drawers and storage units. This computational stockpile represents a substantial wasted potential: modern smartphones have increasingly high-performance and energy-efficient processors, extensive networking capabilities, and a reliable built-in power supply. This project studies the ability to reuse smartphones as "junkyard computers." Junkyard computers grow global computing capacity by extending device lifetimes, which supplants the manufacture of new devices. We show that the capabilities of even decade-old smartphones are within those demanded by modern cloud microservices and discuss how to combine phones to perform increasingly complex tasks. We describe how current operation-focused metrics do not capture the actual carbon costs of compute. We propose Computational Carbon Intensity---a performance metric that balances the continued service of older devices with the superlinear runtime improvements of newer machines. We use this metric to redefine device service lifetime in terms of carbon efficiency. We develop a cloudlet of reused Pixel 3A phones. We analyze the carbon benefits of deploying large, end-to-end microservice-based applications on these smartphones. Finally, we describe system architectures and associated challenges to scale to cloudlets with hundreds and thousands of smartphones. @InProceedings{ASPLOS23p400, author = {Jennifer Switzer and Gabriel Marcano and Ryan Kastner and Pat Pannuto}, title = {Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {400--412}, doi = {10.1145/3575693.3575710}, year = {2023}, } Publisher's Version
	Masood, Amna	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Mendis, Charith	ASPLOS '23: "WACO: Learning Workload-Aware ..." WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program Jaeyeon Won, Charith Mendis, Joel S. Emer, and Saman Amarasinghe (Massachusetts Institute of Technology, USA; University of Illinois at Urbana-Champaign, USA; NVIDIA, USA) In this paper, we present WACO, a novel method of co-optimizing the format and the schedule of a given sparsity pattern in a sparse tensor program. A core challenge in this paper is the design of a lightweight cost model that accurately predicts the runtime of a sparse tensor program by considering the sparsity pattern, the format, and the schedule. The key idea in addressing this is exploiting a sparse convolutional network to learn meaningful features of the sparsity pattern and embedding a coupled behavior between the format and the schedule using a specially designed schedule template. In addition, within the enormous search space of co-optimization, our novel search strategy, an approximate nearest neighbor search, efficiently and accurately retrieves the best format and schedule for a given sparsity pattern. We evaluated WACO for four different algorithms (SpMV, SpMM, SDDMM, and MTTKRP) on a CPU using 726 different sparsity patterns. Our experimental results showed that WACO outperformed four state-of-the-art baselines, Intel MKL, BestFormat, TACO with a default schedule, and ASpT. Compared to the best of four baselines, WACO achieved 1.43×, 1.18×, 1.14×, and 1.27× average speedups on SpMV, SpMM, SDDMM, and MTTKRP, respectively. @InProceedings{ASPLOS23p920, author = {Jaeyeon Won and Charith Mendis and Joel S. Emer and Saman Amarasinghe}, title = {WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {920--934}, doi = {10.1145/3575693.3575742}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Min, Changwoo	ASPLOS '23: "Prism: Optimizing Key-Value ..." Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices Yongju Song, Wook-Hee Kim, Sumit Kumar Monga, Changwoo Min, and Young Ik Eom (Sungkyunkwan University, South Korea; Konkuk University, South Korea; Virginia Tech, USA) As data generation has been on an upward trend, storing vast volumes of data cost-effectively as well as efficiently accessing them is paramount. At the same time, today's storage landscape continues to diversify, from high-bandwidth storage devices such as NVMe SSDs to low-latency non-volatile memory (e.g., Intel Optane DCPMM). These heterogeneous storage devices have the potential to deliver high performance in terms of bandwidth and latency with cost efficiency, while achieving the performance and cost targets together still remains a challenging problem. We provide our solution, Prism, a novel key-value store that utilizes modern heterogeneous storage devices. Prism uses heterogeneous storage devices synergistically to harness the advantages of each storage device while suppressing their downsides. We devise new techniques to balance the latency-bandwidth tradeoff when reading from SSD. For ensuring multicore scalability and crash consistency of data across heterogeneous storage media, Prism proposes cross-storage concurrency control and cross-storage crash consistency protocols. Our evaluation shows that Prism outperforms state-of-the-art key-value stores by up to 13.1× with significantly lower tail latency. @InProceedings{ASPLOS23p588, author = {Yongju Song and Wook-Hee Kim and Sumit Kumar Monga and Changwoo Min and Young Ik Eom}, title = {Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {588--602}, doi = {10.1145/3575693.3575722}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Min, Seungwon	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Monga, Sumit Kumar	ASPLOS '23: "Prism: Optimizing Key-Value ..." Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices Yongju Song, Wook-Hee Kim, Sumit Kumar Monga, Changwoo Min, and Young Ik Eom (Sungkyunkwan University, South Korea; Konkuk University, South Korea; Virginia Tech, USA) As data generation has been on an upward trend, storing vast volumes of data cost-effectively as well as efficiently accessing them is paramount. At the same time, today's storage landscape continues to diversify, from high-bandwidth storage devices such as NVMe SSDs to low-latency non-volatile memory (e.g., Intel Optane DCPMM). These heterogeneous storage devices have the potential to deliver high performance in terms of bandwidth and latency with cost efficiency, while achieving the performance and cost targets together still remains a challenging problem. We provide our solution, Prism, a novel key-value store that utilizes modern heterogeneous storage devices. Prism uses heterogeneous storage devices synergistically to harness the advantages of each storage device while suppressing their downsides. We devise new techniques to balance the latency-bandwidth tradeoff when reading from SSD. For ensuring multicore scalability and crash consistency of data across heterogeneous storage media, Prism proposes cross-storage concurrency control and cross-storage crash consistency protocols. Our evaluation shows that Prism outperforms state-of-the-art key-value stores by up to 13.1× with significantly lower tail latency. @InProceedings{ASPLOS23p588, author = {Yongju Song and Wook-Hee Kim and Sumit Kumar Monga and Changwoo Min and Young Ik Eom}, title = {Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {588--602}, doi = {10.1145/3575693.3575722}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Murat, Antoine	ASPLOS '23: "uBFT: Microsecond-Scale BFT ..." uBFT: Microsecond-Scale BFT using Disaggregated Memory Marcos K. Aguilera, Naama Ben-David, Rachid Guerraoui, Antoine Murat, Athanasios Xygkis, and Igor Zablotchi (VMware Research, USA; EPFL, Switzerland; Massachusetts Institute of Technology, USA) We propose uBFT, the first State Machine Replication (SMR) system to achieve microsecond-scale latency in data centers, while using only 2f+1 replicas to tolerate f Byzantine failures. The Byzantine Fault Tolerance (BFT) provided by uBFT is essential as pure crashes appear to be a mere illusion with real-life systems reportedly failing in many unexpected ways. uBFT relies on a small non-tailored trusted computing base—disaggregated memory—and consumes a practically bounded amount of memory. uBFT is based on a novel abstraction called Consistent Tail Broadcast, which we use to prevent equivocation while bounding memory. We implement uBFT using RDMA-based disaggregated memory and obtain an end-to-end latency of as little as 10 us. This is at least 50× faster than MinBFT, a state-of-the-art 2f+1 BFT SMR based on Intel’s SGX. We use uBFT to replicate two KV-stores (Memcached and Redis), as well as a financial order matching engine (Liquibook). These applications have low latency (up to 20 us) and become Byzantine tolerant with as little as 10 us more. The price for uBFT is a small amount of reliable disaggregated memory (less than 1 MiB), which in our prototype consists of a small number of memory servers connected through RDMA and replicated for fault tolerance. @InProceedings{ASPLOS23p862, author = {Marcos K. Aguilera and Naama Ben-David and Rachid Guerraoui and Antoine Murat and Athanasios Xygkis and Igor Zablotchi}, title = {uBFT: Microsecond-Scale BFT using Disaggregated Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {862--877}, doi = {10.1145/3575693.3575732}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Murillo, Jorge	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Musuvathi, Madanlal	ASPLOS '23: "MSCCLang: Microsoft Collective ..." MSCCLang: Microsoft Collective Communication Language Meghan Cowan, Saeed Maleki, Madanlal Musuvathi, Olli Saarikivi, and Yifan Xiong (Microsoft Research, USA; Microsoft Research, China) Machine learning models with millions or billions of parameters are increasingly trained and served on large multi-GPU systems. As models grow in size and execute on more GPUs, collective communication becomes a bottleneck. Custom collective algorithms optimized for both particular network topologies and application-specific communication patterns can alleviate this bottleneck and help these applications scale. However, implementing correct and efficient custom algorithms is challenging. This paper introduces MSCCLang, a system for programmable GPU communication. MSCCLang provides a domain specific language for writing collective communication algorithms and an optimizing compiler for lowering them to an executable form, which can be executed efficiently and flexibly in an interpreter-based runtime. We used MSCCLang to write novel collective algorithms for AllReduce and AllToAll that are up to 1.9× and 1.3× faster than hand-optimized implementations, respectively. @InProceedings{ASPLOS23p502, author = {Meghan Cowan and Saeed Maleki and Madanlal Musuvathi and Olli Saarikivi and Yifan Xiong}, title = {MSCCLang: Microsoft Collective Communication Language}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {502--514}, doi = {10.1145/3575693.3575724}, year = {2023}, } Publisher's Version
	Nair, Prashant J.	ASPLOS '23: "HuffDuff: Stealing Pruned ..." HuffDuff: Stealing Pruned DNNs from Sparse Accelerators Dingqing Yang, Prashant J. Nair, and Mieszko Lis (University of British Columbia, Canada) Deep learning models are a valuable “secret sauce” that confers a significant competitive advantage. Many models are never visible to the user and even publicly known state-of-the-art models are either completely proprietary or only accessible via access-controlled APIs. Increasingly, these models run directly on the edge, often using a low-power DNN accelerator. This makes models particularly vulnerable, as an attacker with physical access can exploit side channels like off-chip memory access volumes. Indeed, prior work has shown that this channel can be used to steal dense DNNs from edge devices by correlating data transfer volumes with layer geometry. Unfortunately, prior techniques become intractable when the model is sparse in either weights or activations because off-chip transfers no longer correspond exactly to layer dimensions. Could it be that the many mobile-class sparse accelerators are inherently safe from this style of attack? In this paper, we show that it is feasible to steal a pruned DNN model architecture from a mobile-class sparse accelerator using the DRAM access volume channel. We describe HuffDuff, an attack scheme with two novel techniques that leverage (i) the boundary effect present in CONV layers, and (ii) the timing side channel of on-the-fly activation compression. Together, these techniques dramatically reduce the space of possible model architectures up to 94 orders of magnitude, resulting in fewer than 100 candidate models — a number that can be feasibly tested. Finally, we sample network instances from our solution space and show that (i) our solutions reach the victim accuracy under the iso-footprint constraint, and (ii) significantly improve black-box targeted attack success rates. @InProceedings{ASPLOS23p385, author = {Dingqing Yang and Prashant J. Nair and Mieszko Lis}, title = {HuffDuff: Stealing Pruned DNNs from Sparse Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {385--399}, doi = {10.1145/3575693.3575738}, year = {2023}, } Publisher's Version
	Neto, David	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Newburn, C. J.	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Nigam, Rachit	ASPLOS '23: "Stepwise Debugging for Hardware ..." Stepwise Debugging for Hardware Accelerators Griffin Berlstein, Rachit Nigam, Christophe Gyurgyik, and Adrian Sampson (Cornell University, USA) High-level programming models for hardware design let domain experts quickly produce specialized accelerators. However, tools for debugging these accelerators remain tied to low-level hardware description languages (HDLs). High-level descriptions contain control-flow information that is lost in HDL code. We describe Cider, a stepwise debugger that exploits this information to provide software-like debugging abstractions for languages that compile to hardware. Cider uses Calyx, an intermediate language for accelerator generators that preserves control information. Cider provides breakpoints, watchpoints, state inspection, and source-level position mapping. Using case studies that examine one new and two preexisting accelerator generators, we demonstrate how Cider helps find and localize previously unreported bugs. By directly simulating a control-rich representation, Cider avoids wasting effort on inactive parts of the design and, despite being largely unoptimized, performs competitively with open-source HDL simulators. @InProceedings{ASPLOS23p778, author = {Griffin Berlstein and Rachit Nigam and Christophe Gyurgyik and Adrian Sampson}, title = {Stepwise Debugging for Hardware Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {778--790}, doi = {10.1145/3575693.3575717}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Niu, Dimin	ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Nottingham, Natalia	ASPLOS '23: "Qompress: Efficient Compilation ..." Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction Andrew Litteken, Lennart Maximilian Seifert, Jason Chadwick, Natalia Nottingham, Frederic T. Chong, and Jonathan M. Baker (University of Chicago, USA) Quantum computing is in an era of limited resources. Current hardware lacks high fidelity gates, long coherence times, and the number of computational units required to perform meaningful computation. Contemporary quantum devices typically use a binary system, where each qubit exists in a superposition of the 0 and 1 states. However, it is often possible to access the 2 or even 3 states in the same physical unit by manipulating the system in different ways. In this work, we consider automatically encoding two qubits into one four-state ququart via a compression scheme. We use quantum optimal control to design efficient proof-of-concept gates that fully replicate standard qubit computation on these encoded qubits. We extend qubit compilation schemes to efficiently route qubits on an arbitrary mixed-radix system consisting of both qubits and ququarts, reducing communication and minimizing excess circuit execution time introduced by longer-duration ququart gates. In conjunction with these compilation strategies, we introduce several methods to find beneficial compressions, reducing circuit error due to computation and communication by up to 50 @InProceedings{ASPLOS23p646, author = {Andrew Litteken and Lennart Maximilian Seifert and Jason Chadwick and Natalia Nottingham and Frederic T. Chong and Jonathan M. Baker}, title = {Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {646--659}, doi = {10.1145/3575693.3575726}, year = {2023}, } Publisher's Version
	Novakovic, Stanko	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Olukotun, Kunle	ASPLOS '23: "Sigma: Compiling Einstein ..." Sigma: Compiling Einstein Summations to Locality-Aware Dataflow Tian Zhao, Alexander Rucker, and Kunle Olukotun (Stanford University, USA) Most dataflow accelerator compilers achieve high performance by mapping each node in a dataflow program to a dedicated hardware element on a dataflow accelerator. However, this approach misses critical data reuse optimizations required to exploit the data bandwidth from fine-grained memory elements, e.g., FIFOs and pipeline registers. Moreover, writing performant dataflow programs requires users to have domain expertise in the underlying dataflow accelerators. To address these issues, we designed Sigma, a novel compiler that supports high-level programming constructs such as Einstein summations, index notations, and tensors, finds opportunities for data reuse from high-level dataflow graphs, and exploits on-chip data bandwidth from fine-grained memory elements. Sigma targeting a research dataflow accelerator demonstrates a 5.4x speedup and 44.6x area-normalized speedup over Nvidia's V100 accelerator, and a 7.1x speedup over hand-written dataflow programs. @InProceedings{ASPLOS23p718, author = {Tian Zhao and Alexander Rucker and Kunle Olukotun}, title = {Sigma: Compiling Einstein Summations to Locality-Aware Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {718--732}, doi = {10.1145/3575693.3575694}, year = {2023}, } Publisher's Version
	Ozkan, Burcu Kulahcioglu	ASPLOS '23: "Probabilistic Concurrency ..." Probabilistic Concurrency Testing for Weak Memory Programs Mingyu Gao, Soham Chakraborty, and Burcu Kulahcioglu Ozkan (Delft University of Technology, Netherlands) The Probabilistic Concurrency Testing (PCT) algorithm that provides theoretical guarantees on the probability of detecting concurrency bugs does not apply to weak memory programs. The PCT algorithm builds on the interleaving semantics of sequential consistency, which does not hold for weak memory concurrency. It is because weak memory concurrency allows additional behaviors that cannot be produced by any interleaving execution. In this paper, we generalize PCT to address weak memory concurrency and present Probabilistic Concurrency Testing for Weak Memory (PCTWM). We empirically evaluate PCTWM on a set of well-known weak memory program benchmarks in comparison to the state-of-the-art weak memory testing tool C11Tester. Our results show that PCTWM can detect concurrency bugs more frequently than C11Tester. @InProceedings{ASPLOS23p603, author = {Mingyu Gao and Soham Chakraborty and Burcu Kulahcioglu Ozkan}, title = {Probabilistic Concurrency Testing for Weak Memory Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {603--616}, doi = {10.1145/3575693.3575729}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Panda, Aurojit	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Pande, Santosh	ASPLOS '23: "Decker: Attack Surface Reduction ..." Decker: Attack Surface Reduction via On-Demand Code Mapping Chris Porter, Sharjeel Khan, and Santosh Pande (Georgia Institute of Technology, USA) Modern code reuse attacks take full advantage of bloated software. Attackers piece together short sequences of instructions in otherwise benign code to carry out malicious actions. Mitigating these reusable code snippets, known as gadgets, has become one of the prime focuses of attack surface reduction research. While some debloating techniques remove parts of software that contain such gadgets, other methods focus on making them unusable by breaking up chains of them, thereby substantially diminishing the possibility of code reuse attacks. Third-party libraries are another main focus, because they exhibit a high number of vulnerabilities, but recently, techniques have emerged that deal with whole applications. Attack surface reduction efforts have typically tried to eliminate such attacks by subsetting (debloating) the application, e.g. via user-specified inputs, configurations, or features to achieve high gadget reductions. However, such techniques suffer from the limitations of soundness, i.e. the software might crash during no-attack executions on regular inputs, or they may be conservative and leave a large amount of attack surface untackled. In this work we present a general, whole-program attack surface reduction technique called Decker that significantly reduces gadgets which are accessible to an attacker during an execution phase (called a deck) and has minor performance degradation. Decker requires no user inputs and leaves all features intact. It uses static analysis to determine key function sets that should be enabled/disabled at runtime. The runtime system then enables these function sets at the specified program points during execution. On SPEC CPU 2017, our framework achieves 73.2% total gadget reduction with 5.2% average slowdown. On 10 GNU coreutils applications, it achieves 87.2% reduction and negligible slowdown. On the nginx server it achieves 80.3% reduction with 2% slowdown. We also provide a gadget chain-breaking case study, including detailed JOP gadget metrics on both Linux and Windows, and show that our framework breaks the shell-spawning chain in all cases. @InProceedings{ASPLOS23p192, author = {Chris Porter and Sharjeel Khan and Santosh Pande}, title = {Decker: Attack Surface Reduction via On-Demand Code Mapping}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {192--206}, doi = {10.1145/3575693.3575734}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Pandey, Shweta	ASPLOS '23: "Scoped Buffered Persistency ..." Scoped Buffered Persistency Model for GPUs Shweta Pandey, Aditya K Kamath, and Arkaprava Basu (IISc Bangalore, India; University of Washington, USA) While the implications of persistent memory (PM) on CPU hardware and software are well-explored, the same is not true for GPUs (Graphics Processing Units). A recent work, GPM, demonstrated how GPU programs can benefit from the fine-grain persistence of PM. However, in the absence of a persistency model, one cannot reason about the correctness of PM-aware GPU programs. Persistency models define the order in which writes to PM are persisted. We explore persistency models for GPUs. We explore persistency models for GPUs. We demonstrate that CPU persistency models fall short for GPUs. We qualitatively and quantitatively argue that GPU persistency models should support scopes and buffering of writes to PM to leverage parallelism while adapting to higher NVM latencies. We formally specify a GPU persistency model that supports both scopes and buffers. We detail how GPU architecture can efficiently realize such a model. Finally, we quantitatively demonstrate the usefulness of scopes and buffers for PM-aware GPU programs. @InProceedings{ASPLOS23p688, author = {Shweta Pandey and Aditya K Kamath and Arkaprava Basu}, title = {Scoped Buffered Persistency Model for GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {688--701}, doi = {10.1145/3575693.3575749}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Pannuto, Pat	ASPLOS '23: "Junkyard Computing: Repurposing ..." Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon Jennifer Switzer, Gabriel Marcano, Ryan Kastner, and Pat Pannuto (University of California at San Diego, USA) 1.5 billion smartphones are sold annually, and most are decommissioned less than two years later. Most of these unwanted smartphones are neither discarded nor recycled but languish in junk drawers and storage units. This computational stockpile represents a substantial wasted potential: modern smartphones have increasingly high-performance and energy-efficient processors, extensive networking capabilities, and a reliable built-in power supply. This project studies the ability to reuse smartphones as "junkyard computers." Junkyard computers grow global computing capacity by extending device lifetimes, which supplants the manufacture of new devices. We show that the capabilities of even decade-old smartphones are within those demanded by modern cloud microservices and discuss how to combine phones to perform increasingly complex tasks. We describe how current operation-focused metrics do not capture the actual carbon costs of compute. We propose Computational Carbon Intensity---a performance metric that balances the continued service of older devices with the superlinear runtime improvements of newer machines. We use this metric to redefine device service lifetime in terms of carbon efficiency. We develop a cloudlet of reused Pixel 3A phones. We analyze the carbon benefits of deploying large, end-to-end microservice-based applications on these smartphones. Finally, we describe system architectures and associated challenges to scale to cloudlets with hundreds and thousands of smartphones. @InProceedings{ASPLOS23p400, author = {Jennifer Switzer and Gabriel Marcano and Ryan Kastner and Pat Pannuto}, title = {Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {400--412}, doi = {10.1145/3575693.3575710}, year = {2023}, } Publisher's Version
	Park, Jeongmin	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Pedram, Massoud	ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Pei, Lingfeng	ASPLOS '23: "AfterImage: Leaking Control ..." AfterImage: Leaking Control Flow Data and Tracking Load Operations via the Hardware Prefetcher Yun Chen, Lingfeng Pei, and Trevor E. Carlson (National University of Singapore, Singapore) Research into processor-based side-channels has seen both a large number and a large variety of disclosed vulnerabilities that can leak critical, private data to malicious attackers. While most previous works require speculative execution and the use of cache primitives to transmit data, our new approach, called AfterImage, requires neither, capitalizing on vulnerabilities in Intel’s IP-stride prefetcher to both expose and transmit victim data. By training this prefetcher with attacker-known values, and watching for changes to the prefetcher state when execution returns to the attacker, it is now possible to monitor and leak critical data from a large number of common userspace applications and kernel routines without speculation and additional cache accesses. To demonstrate the novel capabilities of AfterImage, we (1) present proof-of-concept attacks that leak data across different isolation levels, (2) present an end-to-end attack that leaks an entire RSA key from a modern, timing-balanced algorithm, and also (3) show how AfterImage can significantly improve the effectiveness of other attacks, such as power side-channel attacks, by using this technique as a high-precision marker. In addition to an extensive evaluation of these and other cache-based attacks, we also present a full reverse-engineering of the Intel IP-stride prefetcher which was required to enable AfterImage, and describe how AfterImage can be used as a covert channel. Finally, we present several mitigation techniques that can be used to block this side-channel on machines today. Taken together, this work explores a full set of techniques to utilize the prefetcher to leak previously protected information between different protection domains (SGX, kernel and other user spaces) and across many important applications, including security and non-security-related workloads. @InProceedings{ASPLOS23p16, author = {Yun Chen and Lingfeng Pei and Trevor E. Carlson}, title = {AfterImage: Leaking Control Flow Data and Tracking Load Operations via the Hardware Prefetcher}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {16--32}, doi = {10.1145/3575693.3575719}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Pekhimenko, Gennady	ASPLOS '23: "Hidet: Task-Mapping Programming ..." Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding, Cody Hao Yu, Bojian Zheng, Yizhi Liu, Yida Wang, and Gennady Pekhimenko (University of Toronto, Canada; Vector Institute, Canada; Amazon Web Services, USA) As deep learning models nowadays are widely adopted by both cloud services and edge devices, reducing the latency of deep learning model inferences becomes crucial to provide efficient model serving. However, it is challenging to develop efficient tensor programs for deep learning operators due to the high complexity of modern accelerators (e.g., NVIDIA GPUs and Google TPUs) and the rapidly growing number of operators. Deep learning compilers, such as Apache TVM, adopt declarative scheduling primitives to lower the bar of developing tensor programs. However, we show that this approach is insufficient to cover state-of-the-art tensor program optimizations (e.g., double buffering). In this paper, we propose to embed the scheduling process into tensor programs and use dedicated mappings, called task mappings, to define the computation assignment and ordering directly in the tensor programs. This new approach greatly enriches the expressible optimizations by allowing developers to manipulate tensor programs at a much finer granularity (e.g., allowing program-statement-level optimizations). We call the proposed method the task-mapping programming paradigm. In addition, we propose a new post-scheduling fusion optimization that allows developers to focus on scheduling every single operator and automates the fusion after scheduling. It greatly reduces the engineering efforts for operator fusion. Our proposed paradigm also constructs an efficient hardware-centric schedule space, which is agnostic to the program input size and greatly reduces the tuning time. With the proposed paradigm, we implement a deep learning compiler Hidet. Extensive experiments on modern convolution and transformer models show that Hidet outperforms state-of-the-art DNN inference framework, ONNX Runtime, and compiler, TVM equipped with scheduler AutoTVM and Ansor, by up to 1.48x (1.22x on average). It also reduces the tuning time by 20x and 11x compared with AutoTVM and Ansor, respectively. We open-sourced hidet at https://www.github.com/hidet-org/hidet. @InProceedings{ASPLOS23p370, author = {Yaoyao Ding and Cody Hao Yu and Bojian Zheng and Yizhi Liu and Yida Wang and Gennady Pekhimenko}, title = {Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {370--384}, doi = {10.1145/3575693.3575702}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "TiLT: A Time-Centric Approach ..." TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization Anand Jayarajan, Wei Zhao, Yudi Sun, and Gennady Pekhimenko (University of Toronto, Canada) Stream processing engines (SPEs) are widely used for large scale streaming analytics over unbounded time-ordered data streams. Modern day streaming analytics applications exhibit diverse compute characteristics and demand strict latency and throughput requirements. Over the years, there has been significant attention in building hardware-efficient stream processing engines (SPEs) that support several query optimization, parallelization, and execution strategies to meet the performance requirements of large scale streaming analytics applications. However, in this work, we observe that these strategies often fail to generalize well on many real-world streaming analytics applications due to several inherent design limitations of current SPEs. We further argue that these limitations stem from the shortcomings of the fundamental design choices and the query representation model followed in modern SPEs. To address these challenges, we first propose TiLT, a novel intermediate representation (IR) that offers a highly expressive temporal query language amenable to effective query optimization and parallelization strategies. We subsequently build a compiler backend for TiLT that applies such optimizations on streaming queries and generates hardware-efficient code to achieve high performance on multi-core stream query executions. We demonstrate that TiLT achieves up to 326× (20.49× on average) higher throughput compared to state-of-the-art SPEs (e.g., Trill) across eight real-world streaming analytics applications. TiLT source code is available at https://github.com/ampersand-projects/tilt.git. @InProceedings{ASPLOS23p818, author = {Anand Jayarajan and Wei Zhao and Yudi Sun and Gennady Pekhimenko}, title = {TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {818--832}, doi = {10.1145/3575693.3575704}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Peng, Jie	ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Porter, Chris	ASPLOS '23: "Decker: Attack Surface Reduction ..." Decker: Attack Surface Reduction via On-Demand Code Mapping Chris Porter, Sharjeel Khan, and Santosh Pande (Georgia Institute of Technology, USA) Modern code reuse attacks take full advantage of bloated software. Attackers piece together short sequences of instructions in otherwise benign code to carry out malicious actions. Mitigating these reusable code snippets, known as gadgets, has become one of the prime focuses of attack surface reduction research. While some debloating techniques remove parts of software that contain such gadgets, other methods focus on making them unusable by breaking up chains of them, thereby substantially diminishing the possibility of code reuse attacks. Third-party libraries are another main focus, because they exhibit a high number of vulnerabilities, but recently, techniques have emerged that deal with whole applications. Attack surface reduction efforts have typically tried to eliminate such attacks by subsetting (debloating) the application, e.g. via user-specified inputs, configurations, or features to achieve high gadget reductions. However, such techniques suffer from the limitations of soundness, i.e. the software might crash during no-attack executions on regular inputs, or they may be conservative and leave a large amount of attack surface untackled. In this work we present a general, whole-program attack surface reduction technique called Decker that significantly reduces gadgets which are accessible to an attacker during an execution phase (called a deck) and has minor performance degradation. Decker requires no user inputs and leaves all features intact. It uses static analysis to determine key function sets that should be enabled/disabled at runtime. The runtime system then enables these function sets at the specified program points during execution. On SPEC CPU 2017, our framework achieves 73.2% total gadget reduction with 5.2% average slowdown. On 10 GNU coreutils applications, it achieves 87.2% reduction and negligible slowdown. On the nginx server it achieves 80.3% reduction with 2% slowdown. We also provide a gadget chain-breaking case study, including detailed JOP gadget metrics on both Linux and Windows, and show that our framework breaks the shell-spawning chain in all cases. @InProceedings{ASPLOS23p192, author = {Chris Porter and Sharjeel Khan and Santosh Pande}, title = {Decker: Attack Surface Reduction via On-Demand Code Mapping}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {192--206}, doi = {10.1145/3575693.3575734}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Pszeniczny, Krzysztof	ASPLOS '23: "Propeller: A Profile Guided, ..." Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications Han Shen, Krzysztof Pszeniczny, Rahman Lavaee, Snehasish Kumar, Sriraman Tallam, and Xinliang David Li (Google, USA; Google, Switzerland) While profile guided optimizations (PGO) and link time optimiza-tions (LTO) have been widely adopted, post link optimizations (PLO)have languished until recently when researchers demonstrated that late injection of profiles can yield significant performance improvements. However, the disassembly-driven, monolithic design of post link optimizers face scaling challenges with large binaries andis at odds with distributed build systems. To reconcile and enable post link optimizations within a distributed build environment, we propose Propeller, a relinking optimizer for warehouse scale work-loads. To enable flexible code layout optimizations, we introduce basic block sections, a novel linker abstraction. Propeller uses basic block sections to enable a new approach to PLO without disassembly. Propeller achieves scalability by relinking the binary using precise profiles instead of rewriting the binary. The overhead of relinking is lowered by caching and leveraging distributed compiler actions during code generation. Propeller has been deployed to production at Google with over tens of millions of cores executing Propeller optimized code at any time. An evaluation of internal warehouse-scale applications show Propeller improves performance by 1.1% to 8% beyond PGO and ThinLTO. Compiler tools such as Clang improve by 7% while MySQL improves by 1%. Compared to the state of the art binary optimizer, Propeller achieves comparable performance while lowering memory overheads by 30%-70% on large benchmarks. @InProceedings{ASPLOS23p617, author = {Han Shen and Krzysztof Pszeniczny and Rahman Lavaee and Snehasish Kumar and Sriraman Tallam and Xinliang David Li}, title = {Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {617--631}, doi = {10.1145/3575693.3575727}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Qi, Xiaojuan	ASPLOS '23: "DPACS: Hardware Accelerated ..." DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design Yizhao Gao, Baoheng Zhang, Xiaojuan Qi, and Hayden Kwok-Hay So (University of Hong Kong, Hong Kong) By eliminating compute operations intelligently based on the run time input, dynamic pruning (DP) promises to improve deep neural network inference speed substantially without incurring a major impact on their accuracy. Although many DP algorithms with good pruning performance have been proposed, it remains a challenge to translate these theoretical reductions in compute operations into satisfactory end-to-end speedups in practical real-world implementations. The overhead of identifying operations to be pruned during run time, the need to efficiently process the resulting dynamic dataflow, and the non-trivial memory I/O bottleneck that emerges as the number of compute operations reduces, have all contributed to the challenge of implementing practical DP systems. In this paper, the design and implementation of DPACS are presented to address these challenges. DPACS utilizes a hardware-aware dynamic spatial and channel pruning algorithm in conjunction with a dynamic dataflow engine in hardware to facilitate efficient processing of the pruned network. A channel mask precomputation scheme is designed to reduce memory I/O, and a dedicated inter-layer pipeline is used to achieve efficient indexing and dataflow of sparse activation. Extensive design space exploration has been performed using two architectural variations implemented on FPGA to accelerate multiple networks from the ResNet family on the ImageNet and CIFAR10 dataset across a wide range of pruning ratios. Across the spectrum of configurations, DPACS is able to achieve 1.1× to 3.9× end-to-end speedup over a baseline hardware implementation without pruning. Analysis of the tradeoff among accuracy, compute, and memory I/O performance highlights the importance of algorithm-architecture codesign in developing DP systems. @InProceedings{ASPLOS23p237, author = {Yizhao Gao and Baoheng Zhang and Xiaojuan Qi and Hayden Kwok-Hay So}, title = {DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {237--251}, doi = {10.1145/3575693.3575728}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Qian, Depei	ASPLOS '23: "VClinic: A Portable and Efficient ..." VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers Xin You, Hailong Yang, Kelun Lei, Zhongzhi Luan, and Depei Qian (Beihang University, China) Fine-grained value profilers reveal a promising way to accurately detect value-related software inefficiencies with binary instrumentation. Due to the architecture-dependent implementation details of binary instrumentation, existing value profilers suffer from poor portability as well as high engineering efforts to achieve efficiency across platforms. In this paper, we propose VClinic, a portable and efficient fine-grained value profiling framework for analyzing highly optimized binaries on both X86 and ARM platforms. VClinic exploits operand-centric two-level designs in its implementation to provide the common building blocks required for value profilers. By constructing four representative value profilers with VClinic, we demonstrate that VClinic can ease the development of value profilers with portability and efficiency across platforms. Guided by the value profilers built upon VClinic, we can achieve up to 89.94% and 74.66% speedup for real-world programs on X86 and ARM platforms, respectively. @InProceedings{ASPLOS23p892, author = {Xin You and Hailong Yang and Kelun Lei and Zhongzhi Luan and Depei Qian}, title = {VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {892--904}, doi = {10.1145/3575693.3576934}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Qian, Xuehai	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version ASPLOS '23: "Khuzdul: Efficient and Scalable ..." Khuzdul: Efficient and Scalable Distributed Graph Pattern Mining Engine Jingji Chen and Xuehai Qian (Purdue University, USA) This paper proposes Khuzdul, a distributed execution engine with a well-defined abstraction that can be integrated with existing single-machine graph pattern mining (GPM) systems to provide efficiency and scalability at the same time. The key novelty is the extendable embedding abstraction which can express pattern enumeration algorithms, allow fine-grained task scheduling, and enable low-cost GPM-specific data reuse to reduce communication cost. The effective BFS-DFS hybrid exploration generates sufficient concurrent tasks for communication-computation overlapping with bounded memory consumption. Two scalable distributed GPM systems are implemented by porting Automine and GraphPi on Khuzdul. Our evaluation shows that Khuzdul based systems significantly outperform state-of-the-art distributed GPM systems with partitioned graphs by up to 75.5× (on average 19.0×), achieve similar or even better performance compared with the fastest distributed GPM systems with replicated graph, and scale to massive graphs with more than one hundred billion edges with a commodity cluster. @InProceedings{ASPLOS23p413, author = {Jingji Chen and Xuehai Qian}, title = {Khuzdul: Efficient and Scalable Distributed Graph Pattern Mining Engine}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {413--426}, doi = {10.1145/3575693.3575743}, year = {2023}, } Publisher's Version
	Querzoni, Leonardo	ASPLOS '23: "Where Did My Variable Go? ..." Where Did My Variable Go? Poking Holes in Incomplete Debug Information Cristian Assaiante, Daniele Cono D'Elia, Giuseppe Antonio Di Luna, and Leonardo Querzoni (Sapienza University of Rome, Italy) The availability of debug information for optimized executables can largely ease crucial tasks such as crash analysis. Source-level debuggers use this information to display program state in terms of source code, allowing users to reason on it even when optimizations alter program structure extensively. A few recent endeavors have proposed effective methodologies for identifying incorrect instances of debug information, which can mislead users by presenting them with an inconsistent program state. In this work, we identify and study a related important problem: the completeness of debug information. Unlike correctness issues for which an unoptimized executable can serve as reference, we find there is no analogous oracle to deem when the cause behind an unreported part of program state is an unavoidable effect of optimization or a compiler implementation defect. In this scenario, we argue that empirically derived conjectures on the expected availability of debug information can serve as an effective means to expose classes of these defects. We propose three conjectures involving variable values and study how often synthetic programs compiled with different configurations of the popular gcc and LLVM compilers deviate from them. We then discuss techniques to pinpoint the optimizations behind such violations and minimize bug reports accordingly. Our experiments revealed, among others, 24 bugs already confirmed by the developers of the gcc-gdb and clang-lldb ecosystems. @InProceedings{ASPLOS23p935, author = {Cristian Assaiante and Daniele Cono D'Elia and Giuseppe Antonio Di Luna and Leonardo Querzoni}, title = {Where Did My Variable Go? Poking Holes in Incomplete Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {935--947}, doi = {10.1145/3575693.3575720}, year = {2023}, } Publisher's Version
	Quinn, Andrew	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Qureshi, Moinuddin	ASPLOS '23: "FrozenQubits: Boosting Fidelity ..." FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes Ramin Ayanzadeh, Narges Alavisamani, Poulami Das, and Moinuddin Qureshi (Georgia Institute of Technology, USA) Quantum Approximate Optimization Algorithm (QAOA) is one of the leading candidates for demonstrating the quantum advantage using near-term quantum computers. Unfortunately, high device error rates limit us from reliably running QAOA circuits for problems with more than a few qubits. In QAOA, the problem graph is translated into a quantum circuit such that every edge corresponds to two 2-qubit CNOT operations in each layer of the circuit. As CNOTs are extremely error-prone, the fidelity of QAOA circuits is dictated by the number of edges in the problem graph. We observe that the majority of graphs corresponding to real-world applications follow a “power-law” distribution, where some hotspot nodes have significantly higher number of connections. We leverage this insight and propose “FrozenQubits” that freezes the hotspot nodes or qubits and intelligently partitions the state-space of the given problem into several smaller sub-spaces, which are then solved independently. The corresponding QAOA sub-circuits are significantly less vulnerable to gate and decoherence errors due to the reduced number of CNOT operations in each sub-circuit. Unlike prior circuit-cutting approaches, FrozenQubits does not require any exponentially complex postprocessing step. Our evaluations with 5,300 QAOA circuits on eight different quantum computers from IBM show that FrozenQubits can improve the quality of solutions by 8.73x on average (and by up to 57x), albeit while utilizing 2x more quantum resources. @InProceedings{ASPLOS23p311, author = {Ramin Ayanzadeh and Narges Alavisamani and Poulami Das and Moinuddin Qureshi}, title = {FrozenQubits: Boosting Fidelity of QAOA by Skipping Hotspot Nodes}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {311--324}, doi = {10.1145/3575693.3575741}, year = {2023}, } Publisher's Version Info
	Qureshi, Zaid	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Rahman, Shafiur	ASPLOS '23: "CommonGraph: Graph Analytics ..." CommonGraph: Graph Analytics on Evolving Data Mahbod Afarin, Chao Gao, Shafiur Rahman, Nael Abu-Ghazaleh, and Rajiv Gupta (University of California at Riverside, USA) We consider the problem of graph analytics on evolving graphs (i.e., graphs that change over time). In this scenario, a query typically needs to be applied to different snapshots of the graph over an extended time window, for example to track the evolution of a property over time. Solving a query independently on multiple snapshots is inefficient due to repeated execution of subcomputation common to multiple snapshots. At the same time, we show that using streaming, where we start from the earliest snapshot and stream the changes to the graph incrementally updating the query results one snapshot at a time is also inefficient. We propose CommonGraph, an approach for efficient processing of queries on evolving graphs. We first observe that deletion operations are significantly more expensive than addition operations for many graph queries (those that are monotonic). CommonGraph converts all deletions to additions by finding a common graph that exists across all snapshots. After computing the query on this graph, to reach any snapshot, we simply need to add the missing edges and incrementally update the query results. CommonGraph also allows sharing of common additions among snapshots that require them, and breaks the sequential dependency inherent in the traditional streaming approach where snapshots are processed in sequence, enabling additional opportunities for parallelism. We incorporate the CommonGraph approach by extending the KickStarter streaming framework. We implement optimizations that enable efficient handling of edge additions without resorting to expensive in place graph mutations, significantly reducing the streaming overhead, and enabling direct reuse of shared edges among different snapshots. CommonGraph achieves 1.38x-8.17x improvement in performance over Kickstarter across multiple benchmarks. @InProceedings{ASPLOS23p133, author = {Mahbod Afarin and Chao Gao and Shafiur Rahman and Nael Abu-Ghazaleh and Rajiv Gupta}, title = {CommonGraph: Graph Analytics on Evolving Data}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {133--145}, doi = {10.1145/3575693.3575713}, year = {2023}, } Publisher's Version
	Rajadnya, Samir	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Ravi, Gokul Subramanian	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version ASPLOS '23: "Better Than Worst-Case Decoding ..." Better Than Worst-Case Decoding for Quantum Error Correction Gokul Subramanian Ravi, Jonathan M. Baker, Arash Fayyazi, Sophia Fuhui Lin, Ali Javadi-Abhari, Massoud Pedram, and Frederic T. Chong (University of Chicago, USA; University of Southern California, USA; IBM, USA) The overheads of classical decoding for quantum error correction in cryogenic quantum systems grow rapidly with the number of logical qubits and their correction code distance. Decoding at room temperature is bottlenecked by refrigerator I/O bandwidth while cryogenic on-chip decoding is limited by area/power/thermal budget. To overcome these overheads, we are motivated by the observation that in the common case (over 90% of the time), error correction 'syndromes' are fairly trivial with high redundancy / sparsity, since the error correction codes are over-provisioned to be able to correct for uncommon worst-case complex scenarios (to ensure substantially low logical error rates). If suitably exploited, these trivial scenarios can be handled with insignificant overhead, thereby alleviating any bottlenecks towards handling the worst-case scenarios by state-of-the-art means. We propose Better Than Worst-Case Decoding for Quantum Error Correction, targeting cryogenic quantum systems and Surface Code, consisting of: On-chip Clique Decoder: An extremely lightweight decoder for correcting trivial common-case errors, designed for the cryogenic domain. The decoder is implemented and evaluated for SFQ logic. Statistical Off-chip Bandwidth Allocation: A statistical confidence-based technique for allocation of off-chip decoding bandwidth, to efficiently handle the rare complex decodes that are not covered by the Clique Decoder. Decode-Overflow Execution Stalling: A method to stall circuit execution, for the worst-case scenarios in which the provisioned off-chip bandwidth is insufficient to complete all requested off-chip decodes. In all, BTWC decoding achieves 70-99+% off-chip bandwidth elimination across a range of logical and physical error rates, without significantly sacrificing the accuracy of a state-of-the-art off-chip decoder. Further, it achieves 10-1000x bandwidth reduction over prior bandwidth reduction techniques, as well as 15-37x resource overhead reduction compared to prior on-chip decoding. @InProceedings{ASPLOS23p88, author = {Gokul Subramanian Ravi and Jonathan M. Baker and Arash Fayyazi and Sophia Fuhui Lin and Ali Javadi-Abhari and Massoud Pedram and Frederic T. Chong}, title = {Better Than Worst-Case Decoding for Quantum Error Correction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {88--102}, doi = {10.1145/3575693.3575733}, year = {2023}, } Publisher's Version
	Ren, Kui	ASPLOS '23: "VDom: Fast and Unlimited Virtual ..." VDom: Fast and Unlimited Virtual Domains on Multiple Architectures Ziqi Yuan, Siyu Hong, Rui Chang, Yajin Zhou, Wenbo Shen, and Kui Ren (Zhejiang University, China) Hardware memory domain primitives, such as Intel MPK and ARM Memory Domain, have been used for efficient in-process memory isolation. However, they can only provide a limited number of memory domains (16 domains), which cannot satisfy the compelling need for more isolated domains inside the address space of a process. Existing solutions to virtualize memory domains are either intrusive (need the modification to existing hardware), or incur a large performance overhead. In this paper, we propose VDom, a fast and scalable memory domain virtualization system that supports unlimited memory domains. VDom leverages separate address spaces to provide an unlimited number of virtual domains, and optimizes related memory management operations. To map virtual domains to hardware domains, we design a domain virtualization algorithm, which manages address spaces and domain maps for threads to efficiently access other domains that are unmapped in the current address space. According to our evaluation on real Intel and ARM platforms, on real-world server applications (httpd and MySQL), VDom incurs less than 2.65% performance overhead, which is lower than the overheads of the state-of-the-art software approaches (libmpk and EPK). In random domain access tests, VDom is comparable to EPK and has significantly higher efficiency than libmpk. @InProceedings{ASPLOS23p905, author = {Ziqi Yuan and Siyu Hong and Rui Chang and Yajin Zhou and Wenbo Shen and Kui Ren}, title = {VDom: Fast and Unlimited Virtual Domains on Multiple Architectures}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {905--919}, doi = {10.1145/3575693.3575735}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Rossbach, Christopher J.	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Rucker, Alexander	ASPLOS '23: "Sigma: Compiling Einstein ..." Sigma: Compiling Einstein Summations to Locality-Aware Dataflow Tian Zhao, Alexander Rucker, and Kunle Olukotun (Stanford University, USA) Most dataflow accelerator compilers achieve high performance by mapping each node in a dataflow program to a dedicated hardware element on a dataflow accelerator. However, this approach misses critical data reuse optimizations required to exploit the data bandwidth from fine-grained memory elements, e.g., FIFOs and pipeline registers. Moreover, writing performant dataflow programs requires users to have domain expertise in the underlying dataflow accelerators. To address these issues, we designed Sigma, a novel compiler that supports high-level programming constructs such as Einstein summations, index notations, and tensors, finds opportunities for data reuse from high-level dataflow graphs, and exploits on-chip data bandwidth from fine-grained memory elements. Sigma targeting a research dataflow accelerator demonstrates a 5.4x speedup and 44.6x area-normalized speedup over Nvidia's V100 accelerator, and a 7.1x speedup over hand-written dataflow programs. @InProceedings{ASPLOS23p718, author = {Tian Zhao and Alexander Rucker and Kunle Olukotun}, title = {Sigma: Compiling Einstein Summations to Locality-Aware Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {718--732}, doi = {10.1145/3575693.3575694}, year = {2023}, } Publisher's Version
	Ruffy, Fabian	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Saarikivi, Olli	ASPLOS '23: "MSCCLang: Microsoft Collective ..." MSCCLang: Microsoft Collective Communication Language Meghan Cowan, Saeed Maleki, Madanlal Musuvathi, Olli Saarikivi, and Yifan Xiong (Microsoft Research, USA; Microsoft Research, China) Machine learning models with millions or billions of parameters are increasingly trained and served on large multi-GPU systems. As models grow in size and execute on more GPUs, collective communication becomes a bottleneck. Custom collective algorithms optimized for both particular network topologies and application-specific communication patterns can alleviate this bottleneck and help these applications scale. However, implementing correct and efficient custom algorithms is challenging. This paper introduces MSCCLang, a system for programmable GPU communication. MSCCLang provides a domain specific language for writing collective communication algorithms and an optimizing compiler for lowering them to an executable form, which can be executed efficiently and flexibly in an interpreter-based runtime. We used MSCCLang to write novel collective algorithms for AllReduce and AllToAll that are up to 1.9× and 1.3× faster than hand-optimized implementations, respectively. @InProceedings{ASPLOS23p502, author = {Meghan Cowan and Saeed Maleki and Madanlal Musuvathi and Olli Saarikivi and Yifan Xiong}, title = {MSCCLang: Microsoft Collective Communication Language}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {502--514}, doi = {10.1145/3575693.3575724}, year = {2023}, } Publisher's Version
	Sampson, Adrian	ASPLOS '23: "Stepwise Debugging for Hardware ..." Stepwise Debugging for Hardware Accelerators Griffin Berlstein, Rachit Nigam, Christophe Gyurgyik, and Adrian Sampson (Cornell University, USA) High-level programming models for hardware design let domain experts quickly produce specialized accelerators. However, tools for debugging these accelerators remain tied to low-level hardware description languages (HDLs). High-level descriptions contain control-flow information that is lost in HDL code. We describe Cider, a stepwise debugger that exploits this information to provide software-like debugging abstractions for languages that compile to hardware. Cider uses Calyx, an intermediate language for accelerator generators that preserves control information. Cider provides breakpoints, watchpoints, state inspection, and source-level position mapping. Using case studies that examine one new and two preexisting accelerator generators, we demonstrate how Cider helps find and localize previously unreported bugs. By directly simulating a control-rich representation, Cider avoids wasting effort on inactive parts of the design and, despite being largely unoptimized, performs competitively with open-source HDL simulators. @InProceedings{ASPLOS23p778, author = {Griffin Berlstein and Rachit Nigam and Christophe Gyurgyik and Adrian Sampson}, title = {Stepwise Debugging for Hardware Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {778--790}, doi = {10.1145/3575693.3575717}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Schulz, Martin	ASPLOS '23: "Copy-on-Pin: The Missing Piece ..." Copy-on-Pin: The Missing Piece for Correct Copy-on-Write David Hildenbrand, Martin Schulz, and Nadav Amit (TU Munich, Germany; Red Hat, Germany; VMware, USA) Operating systems utilize Copy-on-Write (COW) to conserve memory and improve performance. During the last two decades, a series of COW-related bugs - which compromised security, corrupted memory and degraded performance - was found. The majority of these bugs are related to page "pinning", which operating systems employ to access process memory efficiently and to perform direct I/O. Unfortunately, the true cause of these bugs is not well understood, resulting in incomplete bug fixes. We show this by: (1) surveying previously reported pinning-related COW bugs; (2) uncovering new such bugs in Linux, FreeBSD, and NetBSD; and (3) showing that they occur because the COW logic does not consider page pinnings correctly, resulting in incorrect behavior (e.g., I/O of stale data). We then address the underlying problem by deriving when/how shared pages must be copied and under which conditions pinned pages can be shared to maintain correctness. Based on this assessment, we introduce the "Copy-on-Pin (COP)" scheme, an extension of the COW mechanism that handles pinned pages correctly by ensuring pinned pages and shared pages are mutually exclusive. However, we find that a naive implementation of this scheme hampers performance and increases complexity if pages are copied only when strictly necessary. To compensate, we introduce a relaxed-COP design, which does not require precise tracking of page sharing, maintains correctness without increasing complexity, and (while potentially needlessly copying pages in some corner cases) marginally improves performance. Our relaxed-COP solution has been integrated into Linux 5.19. @InProceedings{ASPLOS23p176, author = {David Hildenbrand and Martin Schulz and Nadav Amit}, title = {Copy-on-Pin: The Missing Piece for Correct Copy-on-Write}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {176--191}, doi = {10.1145/3575693.3575716}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Seifert, Lennart Maximilian	ASPLOS '23: "Qompress: Efficient Compilation ..." Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction Andrew Litteken, Lennart Maximilian Seifert, Jason Chadwick, Natalia Nottingham, Frederic T. Chong, and Jonathan M. Baker (University of Chicago, USA) Quantum computing is in an era of limited resources. Current hardware lacks high fidelity gates, long coherence times, and the number of computational units required to perform meaningful computation. Contemporary quantum devices typically use a binary system, where each qubit exists in a superposition of the 0 and 1 states. However, it is often possible to access the 2 or even 3 states in the same physical unit by manipulating the system in different ways. In this work, we consider automatically encoding two qubits into one four-state ququart via a compression scheme. We use quantum optimal control to design efficient proof-of-concept gates that fully replicate standard qubit computation on these encoded qubits. We extend qubit compilation schemes to efficiently route qubits on an arbitrary mixed-radix system consisting of both qubits and ququarts, reducing communication and minimizing excess circuit execution time introduced by longer-duration ququart gates. In conjunction with these compilation strategies, we introduce several methods to find beneficial compressions, reducing circuit error due to computation and communication by up to 50 @InProceedings{ASPLOS23p646, author = {Andrew Litteken and Lennart Maximilian Seifert and Jason Chadwick and Natalia Nottingham and Frederic T. Chong and Jonathan M. Baker}, title = {Qompress: Efficient Compilation for Ququarts Exploiting Partial and Mixed Radix Operations for Communication Reduction}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {646--659}, doi = {10.1145/3575693.3575726}, year = {2023}, } Publisher's Version
	Sha, Yan	ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Shah, Monish	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Shao, Junru	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Shen, Bingyu	ASPLOS '23: "Protecting Data Integrity ..." Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code Haochen Huang, Bingyu Shen, Li Zhong, and Yuanyuan Zhou (University of California at San Diego, USA) Database-backed web applications persist a large amount of production data and have high requirements for integrity. To protect data integrity against application code bugs and operator mistakes, most RDBMSes allow application developers to specify various types of integrity constraints. Unfortunately, applications (e.g., e-commerce web apps) often do not take full advantage of this capability and miss specifying many database constraints, resulting in many severe consequences, such as crashing the order placement page and corrupting the store inventory data. In this paper, we focus on the problem of missing database constraints in web applications. We first study several widely used open-source e-commerce and communication applications, and observe that all these applications have missed integrity constraints and many were added later as afterthoughts after issues occurred. Motivated by our observations, we build a tool called CFinder to automatically infer missing database constraints from application source code by cleverly leveraging the observation that many source code patterns usually imply certain data integrity constraints. By analyzing application source code automatically, CFinder can extract such constraints and check against their database schemas to detect missing ones. We evaluate CFinder with eight widelydeployed web applications, including one commercial company with millions of users. Overall, our tool identifies 210 previously unknown missing constraints. We have reported 92 of them to the developers of these applications, so far 75 are confirmed. Our tool achieves a precision of 78% and a recall of 79%. @InProceedings{ASPLOS23p632, author = {Haochen Huang and Bingyu Shen and Li Zhong and Yuanyuan Zhou}, title = {Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {632--645}, doi = {10.1145/3575693.3575699}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Shen, Haichen	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Shen, Han	ASPLOS '23: "Propeller: A Profile Guided, ..." Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications Han Shen, Krzysztof Pszeniczny, Rahman Lavaee, Snehasish Kumar, Sriraman Tallam, and Xinliang David Li (Google, USA; Google, Switzerland) While profile guided optimizations (PGO) and link time optimiza-tions (LTO) have been widely adopted, post link optimizations (PLO)have languished until recently when researchers demonstrated that late injection of profiles can yield significant performance improvements. However, the disassembly-driven, monolithic design of post link optimizers face scaling challenges with large binaries andis at odds with distributed build systems. To reconcile and enable post link optimizations within a distributed build environment, we propose Propeller, a relinking optimizer for warehouse scale work-loads. To enable flexible code layout optimizations, we introduce basic block sections, a novel linker abstraction. Propeller uses basic block sections to enable a new approach to PLO without disassembly. Propeller achieves scalability by relinking the binary using precise profiles instead of rewriting the binary. The overhead of relinking is lowered by caching and leveraging distributed compiler actions during code generation. Propeller has been deployed to production at Google with over tens of millions of cores executing Propeller optimized code at any time. An evaluation of internal warehouse-scale applications show Propeller improves performance by 1.1% to 8% beyond PGO and ThinLTO. Compiler tools such as Clang improve by 7% while MySQL improves by 1%. Compared to the state of the art binary optimizer, Propeller achieves comparable performance while lowering memory overheads by 30%-70% on large benchmarks. @InProceedings{ASPLOS23p617, author = {Han Shen and Krzysztof Pszeniczny and Rahman Lavaee and Snehasish Kumar and Sriraman Tallam and Xinliang David Li}, title = {Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {617--631}, doi = {10.1145/3575693.3575727}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Shen, Wenbo	ASPLOS '23: "VDom: Fast and Unlimited Virtual ..." VDom: Fast and Unlimited Virtual Domains on Multiple Architectures Ziqi Yuan, Siyu Hong, Rui Chang, Yajin Zhou, Wenbo Shen, and Kui Ren (Zhejiang University, China) Hardware memory domain primitives, such as Intel MPK and ARM Memory Domain, have been used for efficient in-process memory isolation. However, they can only provide a limited number of memory domains (16 domains), which cannot satisfy the compelling need for more isolated domains inside the address space of a process. Existing solutions to virtualize memory domains are either intrusive (need the modification to existing hardware), or incur a large performance overhead. In this paper, we propose VDom, a fast and scalable memory domain virtualization system that supports unlimited memory domains. VDom leverages separate address spaces to provide an unlimited number of virtual domains, and optimizes related memory management operations. To map virtual domains to hardware domains, we design a domain virtualization algorithm, which manages address spaces and domain maps for threads to efficiently access other domains that are unmapped in the current address space. According to our evaluation on real Intel and ARM platforms, on real-world server applications (httpd and MySQL), VDom incurs less than 2.65% performance overhead, which is lower than the overheads of the state-of-the-art software approaches (libmpk and EPK). In random domain access tests, VDom is comparable to EPK and has significantly higher efficiency than libmpk. @InProceedings{ASPLOS23p905, author = {Ziqi Yuan and Siyu Hong and Rui Chang and Yajin Zhou and Wenbo Shen and Kui Ren}, title = {VDom: Fast and Unlimited Virtual Domains on Multiple Architectures}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {905--919}, doi = {10.1145/3575693.3575735}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Shen, Wenting	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Shen, Xipeng	ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Shenoy, Prashant	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Sherwood, Timothy	ASPLOS '23: "A Prediction System Service ..." A Prediction System Service Zhizhou Zhang, Alvin Oliver Glova, Timothy Sherwood, and Jonathan Balkind (University of California at Santa Barbara, USA) To better facilitate application performance programming we propose a software optimization strategy enabled by a novel low-latency Prediction System Service (PSS). Rather than relying on nuanced domain-specific knowledge or slapdash heuristics, a system service for prediction encourages programmers to spend their time uncovering new levers for optimization rather than worrying about the details of their control. The core idea is to write optimizations that improve performance in specific cases, or under specific tunings, and leave the decision of how and when exactly to apply those optimizations to the system to learn through feedback-directed learning. Such a prediction service can be implemented in any number of ways, including as a shared library that can be easily reused by software written in different programming languages, and opens the door to both new software optimization patterns and hardware design possibilities. As a demonstration of the utility of this approach, we show that three very different application-targeted optimization scenarios can each benefit from even a very straightforward perceptron-based implementation of the PSS as long as the service latency can be held low. First, we show that PSS can be used to more intelligently guide hardware lock elision with resulting speedups over a baseline implementation by 34% on average. Second, we show that a PSS can find good configuration parameters for PyPy’s Just-In-Time (JIT) compiler resulting in 15% speedup on average. Last, we show PSS can guide the page reclamation task within a kernel memory management subsystem to reduce the average memory latency by 33% on average. In all three cases, this new optimization pattern with service support is able to meet or beat the best-known hand-crafted methods with a fraction of the complexity. @InProceedings{ASPLOS23p48, author = {Zhizhou Zhang and Alvin Oliver Glova and Timothy Sherwood and Jonathan Balkind}, title = {A Prediction System Service}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {48--60}, doi = {10.1145/3575693.3575714}, year = {2023}, } Publisher's Version
	Shi, Xuanhua	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Shu, Jiwu	ASPLOS '23: "Mobius: Fine Tuning Large-Scale ..." Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers Yangyang Feng, Minhui Xie, Zijie Tian, Shuo Wang, Youyou Lu, and Jiwu Shu (Tsinghua University, China) Fine-tuning on cheap commodity GPU servers makes large-scale deep learning models benefit more people. However, the low inter-GPU communication bandwidth and pressing communication contention on the commodity GPU server obstruct training efficiency. In this paper, we present Mobius, a communication-efficient system for fine tuning large-scale models on commodity GPU servers. The key idea is a novel pipeline parallelism scheme enabling heterogeneous memory for large-scale model training, while bringing fewer communications than existing systems. Mobius partitions the model into stages and carefully schedules them between GPU memory and DRAM to overlap communication with computation. It formulates pipeline execution into a mixed-integer program problem to find the optimal pipeline partition. It also features a new stage-to-GPU mapping method termed cross mapping, to minimize communication contention. Experiments on various scale models and GPU topologies show that Mobius significantly reduces the training time by 3.8-5.1× compared with the prior art. @InProceedings{ASPLOS23p489, author = {Yangyang Feng and Minhui Xie and Zijie Tian and Shuo Wang and Youyou Lu and Jiwu Shu}, title = {Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {489--501}, doi = {10.1145/3575693.3575703}, year = {2023}, } Publisher's Version
	Smith, Kaitlin	ASPLOS '23: "Navigating the Dynamic Noise ..." Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET Gokul Subramanian Ravi, Kaitlin Smith, Jonathan M. Baker, Tejas Kannan, Nathan Earnest, Ali Javadi-Abhari, Henry Hoffmann, and Frederic T. Chong (University of Chicago, USA; IBM, USA) In the Noisy Intermediate Scale Quantum (NISQ) era, the dynamic nature of quantum systems causes noise sources to constantly vary over time. Transient errors from the dynamic NISQ noise landscape are challenging to comprehend and are especially detrimental to classes of applications that are iterative and/or long-running, and therefore their timely mitigation is important for quantum advantage in real-world applications. The most popular examples of iterative long-running quantum applications are variational quantum algorithms (VQAs). Iteratively, VQA’s classical optimizer evaluates circuit candidates on an objective function and picks the best circuits towards achieving the application’s target. Noise fluctuation can cause a significant transient impact on the objective function estimation of the VQA iterations’ tuning candidates. This can severely affect VQA tuning and, by extension, its accuracy and convergence. This paper proposes QISMET: Quantum Iteration Skipping to Mitigate Error Transients, to navigate the dynamic noise landscape of VQAs. QISMET actively avoids instances of high fluctuating noise which are predicted to have a significant transient error impact on specific VQA iterations. To achieve this, QISMET estimates transient error in VQA iterations and designs a controller to keep the VQA tuning faithful to the transient-free scenario. By doing so, QISMET efficiently mitigates a large portion of the transient noise impact on VQAs and is able to improve the fidelity by 1.3x-3x over a traditional VQA baseline, with 1.6-2.4x improvement over alternative approaches, across different applications and machines. @InProceedings{ASPLOS23p515, author = {Gokul Subramanian Ravi and Kaitlin Smith and Jonathan M. Baker and Tejas Kannan and Nathan Earnest and Ali Javadi-Abhari and Henry Hoffmann and Frederic T. Chong}, title = {Navigating the Dynamic Noise Landscape of Variational Quantum Algorithms with QISMET}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {515--529}, doi = {10.1145/3575693.3575739}, year = {2023}, } Publisher's Version
	So, Hayden Kwok-Hay	ASPLOS '23: "DPACS: Hardware Accelerated ..." DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design Yizhao Gao, Baoheng Zhang, Xiaojuan Qi, and Hayden Kwok-Hay So (University of Hong Kong, Hong Kong) By eliminating compute operations intelligently based on the run time input, dynamic pruning (DP) promises to improve deep neural network inference speed substantially without incurring a major impact on their accuracy. Although many DP algorithms with good pruning performance have been proposed, it remains a challenge to translate these theoretical reductions in compute operations into satisfactory end-to-end speedups in practical real-world implementations. The overhead of identifying operations to be pruned during run time, the need to efficiently process the resulting dynamic dataflow, and the non-trivial memory I/O bottleneck that emerges as the number of compute operations reduces, have all contributed to the challenge of implementing practical DP systems. In this paper, the design and implementation of DPACS are presented to address these challenges. DPACS utilizes a hardware-aware dynamic spatial and channel pruning algorithm in conjunction with a dynamic dataflow engine in hardware to facilitate efficient processing of the pruned network. A channel mask precomputation scheme is designed to reduce memory I/O, and a dedicated inter-layer pipeline is used to achieve efficient indexing and dataflow of sparse activation. Extensive design space exploration has been performed using two architectural variations implemented on FPGA to accelerate multiple networks from the ResNet family on the ImageNet and CIFAR10 dataset across a wide range of pruning ratios. Across the spectrum of configurations, DPACS is able to achieve 1.1× to 3.9× end-to-end speedup over a baseline hardware implementation without pruning. Analysis of the tradeoff among accuracy, compute, and memory I/O performance highlights the importance of algorithm-architecture codesign in developing DP systems. @InProceedings{ASPLOS23p237, author = {Yizhao Gao and Baoheng Zhang and Xiaojuan Qi and Hayden Kwok-Hay So}, title = {DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {237--251}, doi = {10.1145/3575693.3575728}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Solihin, Yan	ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Song, Jaeyong	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	Song, Yaoxu	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Song, Yongju	ASPLOS '23: "Prism: Optimizing Key-Value ..." Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices Yongju Song, Wook-Hee Kim, Sumit Kumar Monga, Changwoo Min, and Young Ik Eom (Sungkyunkwan University, South Korea; Konkuk University, South Korea; Virginia Tech, USA) As data generation has been on an upward trend, storing vast volumes of data cost-effectively as well as efficiently accessing them is paramount. At the same time, today's storage landscape continues to diversify, from high-bandwidth storage devices such as NVMe SSDs to low-latency non-volatile memory (e.g., Intel Optane DCPMM). These heterogeneous storage devices have the potential to deliver high performance in terms of bandwidth and latency with cost efficiency, while achieving the performance and cost targets together still remains a challenging problem. We provide our solution, Prism, a novel key-value store that utilizes modern heterogeneous storage devices. Prism uses heterogeneous storage devices synergistically to harness the advantages of each storage device while suppressing their downsides. We devise new techniques to balance the latency-bandwidth tradeoff when reading from SSD. For ensuring multicore scalability and crash consistency of data across heterogeneous storage media, Prism proposes cross-storage concurrency control and cross-storage crash consistency protocols. Our evaluation shows that Prism outperforms state-of-the-art key-value stores by up to 13.1× with significantly lower tail latency. @InProceedings{ASPLOS23p588, author = {Yongju Song and Wook-Hee Kim and Sumit Kumar Monga and Changwoo Min and Young Ik Eom}, title = {Prism: Optimizing Key-Value Store for Modern Heterogeneous Storage Devices}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {588--602}, doi = {10.1145/3575693.3575722}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Sorensen, Tyler	ASPLOS '23: "MC Mutants: Evaluating and ..." MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications Reese Levine, Tianhao Guo, Mingun Cho, Alan Baker, Raph Levien, David Neto, Andrew Quinn, and Tyler Sorensen (University of California at Santa Cruz, USA; New York University, USA; University of California at Davis, USA; Google, Canada; Google, USA) Shared memory platforms provide a memory consistency specification (MCS) so that developers can reason about the behaviors of their parallel programs. Unfortunately, ensuring that a platform conforms to its MCS is difficult, as is exemplified by numerous bugs in well-used platforms. While existing MCS testing approaches find bugs, their efficacy depends on the testing environment (e.g. if synthetic memory pressure is applied). MCS testing environments are difficult to evaluate since legitimate MCS violations are too rare to use as an efficacy metric. As a result, prior approaches have missed critical MCS bugs. This work proposes a mutation testing approach for evaluating MCS testing environments: MC Mutants. This approach mutates MCS tests such that the mutants simulate bugs that might occur. A testing environment can then be evaluated using a mutation score. We utilize MC Mutants in two novel contributions: (1) a parallel testing environment, and (2) An MCS testing confidence strategy that is parameterized over a time budget and confidence threshold. We implement our contributions in WebGPU, a new web-based GPU programming specification, and evaluate our techniques across four GPUs. We improve testing speed by three orders of magnitude over prior work, empowering us to create a conformance test suite that reproduces many mutated tests with high confidence and requires only 64 seconds per test. We identified two bugs in WebGPU implementations, one of which led to a specification change. Moreover, the official WebGPU conformance test suite has adopted our approach due to its efficiency, effectiveness, and broad applicability. @InProceedings{ASPLOS23p473, author = {Reese Levine and Tianhao Guo and Mingun Cho and Alan Baker and Raph Levien and David Neto and Andrew Quinn and Tyler Sorensen}, title = {MC Mutants: Evaluating and Improving Testing for Memory Consistency Specifications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {473--488}, doi = {10.1145/3575693.3575750}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Souza, Abel	ASPLOS '23: "Ecovisor: A Virtual Energy ..." Ecovisor: A Virtual Energy System for Carbon-Efficient Applications Abel Souza, Noman Bashir, Jorge Murillo, Walid Hanafy, Qianlin Liang, David Irwin, and Prashant Shenoy (University of Massachusetts at Amherst, USA) Cloud platforms' rapid growth is raising significant concerns about their carbon emissions. To reduce carbon emissions, future cloud platforms will need to increase their reliance on renewable energy sources, such as solar and wind, which have zero emissions but are highly unreliable. Unfortunately, today's energy systems effectively mask this unreliability in hardware, which prevents applications from optimizing their carbon-efficiency, or work done per kilogram of carbon emitted. To address the problem, we design an "ecovisor", which virtualizes the energy system and exposes software-defined control of it to applications. An ecovisor enables each application to handle clean energy's unreliability in software based on its own specific requirements. We implement a small-scale ecovisor prototype that virtualizes a physical energy system to enable software-based application-level i) visibility into variable grid carbon-intensity and local renewable generation and ii) control of server power usage and battery charging and discharging. We evaluate the ecovisor approach by showing how multiple applications can concurrently exercise their virtual energy system in different ways to better optimize carbon-efficiency based on their specific requirements compared to general system-wide policies. @InProceedings{ASPLOS23p252, author = {Abel Souza and Noman Bashir and Jorge Murillo and Walid Hanafy and Qianlin Liang and David Irwin and Prashant Shenoy}, title = {Ecovisor: A Virtual Energy System for Carbon-Efficient Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {252--265}, doi = {10.1145/3575693.3575709}, year = {2023}, } Publisher's Version
	Stričević, Lazar	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Subramanian, Suvinay	ASPLOS '23: "FLAT: An Optimized Dataflow ..." FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks Sheng-Chun Kao, Suvinay Subramanian, Gaurav Agrawal, Amir Yazdanbakhsh, and Tushar Krishna (Georgia Institute of Technology, USA; Google, USA; Microsoft, USA) Attention mechanisms, primarily designed to capture pairwise correlations between words, have become the backbone of machine learning, expanding beyond natural language processing into other domains. This growth in adaptation comes at the cost of prohibitively large memory requirements and computational complexity, especially at higher number of input elements. This limitation is due to inherently limited data reuse opportunities and quadratic growth in memory footprints, leading to severe memory-boundedness and limited scalability of input elements. This work addresses these challenges by devising a tailored dataflow optimization, called FLAT, for attention mechanisms without altering their functionality. This dataflow processes costly attention operations through a unique fusion mechanism, transforming the memory footprint quadratic growth to merely a linear one. To realize the full potential of this bespoke mechanism, we propose a tiling approach to enhance the data reuse across attention operations. Our method both mitigates the off-chip bandwidth bottleneck as well as reduces the on-chip memory requirement. FLAT delivers 1.94x (1.76x) speedup and 49% and (42%) of energy savings compared to the state-of-the-art Edge (Cloud) accelerators with no customized dataflow optimization. When on-chip resources are scarce (20 KB-200 KB), FLAT yields, on average, 1.5x end-to-end latency reduction across a diverse range of conventional attention-based models with input sequence lengths ranging from 512-token to 64K-token. Our evaluations demonstrate that state-of-the-art DNN dataflow applied to attention operations reach the efficiency limit for inputs above 512 elements. In contrast, FLAT unblocks transformer models for inputs with up to 64K elements. @InProceedings{ASPLOS23p295, author = {Sheng-Chun Kao and Suvinay Subramanian and Gaurav Agrawal and Amir Yazdanbakhsh and Tushar Krishna}, title = {FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {295--310}, doi = {10.1145/3575693.3575747}, year = {2023}, } Publisher's Version
	Sun, Chao	ASPLOS '23: "LeaFTL: A Learning-Based Flash ..." LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives Jinghan Sun, Shaobo Li, Yunxin Sun, Chao Sun, Dejan Vucinic, and Jian Huang (University of Illinois at Urbana-Champaign, USA; ETH Zurich, Switzerland; Western Digital Research, USA) In modern solid-state drives (SSDs), the indexing of flash pages is a critical component in their storage controllers. It not only affects the data access performance, but also determines the efficiency of the precious in-device DRAM resource. A variety of address mapping schemes and optimizations have been proposed. However, most of them were developed with human-driven heuristics. In this paper, we present a learning-based flash translation layer (FTL), named LeaFTL, which learns the address mapping to tolerate dynamic data access patterns via linear regression at runtime. By grouping a large set of mapping entries into a learned segment, it significantly reduces the memory footprint of the address mapping table, which further benefits the data caching in SSD controllers. LeaFTL also employs various optimization techniques, including out-of-band metadata verification to tolerate mispredictions, optimized flash allocation, and dynamic compaction of learned index segments. We implement LeaFTL with both a validated SSD simulator and a real open-channel SSD board. Our evaluation with various storage workloads demonstrates that LeaFTL saves the memory consumption of the mapping table by 2.9× and improves the storage performance by 1.4× on average, in comparison with state-of-the-art FTL schemes. @InProceedings{ASPLOS23p442, author = {Jinghan Sun and Shaobo Li and Yunxin Sun and Chao Sun and Dejan Vucinic and Jian Huang}, title = {LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {442--456}, doi = {10.1145/3575693.3575744}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Sun, Chengnian	ASPLOS '23: "Compilation Consistency Modulo ..." Compilation Consistency Modulo Debug Information Theodore Luo Wang, Yongqiang Tian, Yiwen Dong, Zhenyang Xu, and Chengnian Sun (University of Waterloo, Canada) Compilation Consistency Modulo Debug Information (CCMD) is an essential compiler property that a production compiler should support: the compiler should emit the same machine code regardless of enabling debug information. CCMD is vital to developers’ experiences with debugging a production binary containing no debug information. To debug such a binary, developers need build another binary with the same compiler flags and enable debug information. Without CCMD, the machine code in the latter binary will be different, which can confuse the debugger, hide the bug, or even cause a miscompilation (as GCC once did with the Linux Kernel). This paper is the first to introduce to the research community the validation of CCMD, a new research problem that has been overlooked for decades despite its importance. More importantly, we propose the first testing technique Dfusor to automatically validate CCMD for C compilers. At the high level, given a compilable program P as a seed, Dfusor automatically generates compilable program variants via multiple effective program transformations. Such variants can cause a compiler to emit more debug information than it would when compiling P, thus exercising more code paths in the compiler and increasing the chance to find CCMD bugs. Our extensive evaluations of Dfusor demonstrate that Dfusor can produce variants that exhibit significant increases in the quantity and complexity of the emitted debug information, and thus has found new, real bugs in GCC and LLVM. With a sample of 100 variants derived from distinct seed programs, Dfusor introduces 214% more debug information entries and 36% more distinct debug information entries in the variants than the seeds, and improves the code coverage of GCC and Clang by up to 6.00% and 6.82%. More importantly, Dfusor has found CCMD bugs; within 10 months of development and intermittent testing, Dfusor has found 23 bugs (9 in GCC and 14 in Clang), with 3 confirmed and 18 fixed. @InProceedings{ASPLOS23p146, author = {Theodore Luo Wang and Yongqiang Tian and Yiwen Dong and Zhenyang Xu and Chengnian Sun}, title = {Compilation Consistency Modulo Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {146--158}, doi = {10.1145/3575693.3575740}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Sun, Jinghan	ASPLOS '23: "LeaFTL: A Learning-Based Flash ..." LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives Jinghan Sun, Shaobo Li, Yunxin Sun, Chao Sun, Dejan Vucinic, and Jian Huang (University of Illinois at Urbana-Champaign, USA; ETH Zurich, Switzerland; Western Digital Research, USA) In modern solid-state drives (SSDs), the indexing of flash pages is a critical component in their storage controllers. It not only affects the data access performance, but also determines the efficiency of the precious in-device DRAM resource. A variety of address mapping schemes and optimizations have been proposed. However, most of them were developed with human-driven heuristics. In this paper, we present a learning-based flash translation layer (FTL), named LeaFTL, which learns the address mapping to tolerate dynamic data access patterns via linear regression at runtime. By grouping a large set of mapping entries into a learned segment, it significantly reduces the memory footprint of the address mapping table, which further benefits the data caching in SSD controllers. LeaFTL also employs various optimization techniques, including out-of-band metadata verification to tolerate mispredictions, optimized flash allocation, and dynamic compaction of learned index segments. We implement LeaFTL with both a validated SSD simulator and a real open-channel SSD board. Our evaluation with various storage workloads demonstrates that LeaFTL saves the memory consumption of the mapping table by 2.9× and improves the storage performance by 1.4× on average, in comparison with state-of-the-art FTL schemes. @InProceedings{ASPLOS23p442, author = {Jinghan Sun and Shaobo Li and Yunxin Sun and Chao Sun and Dejan Vucinic and Jian Huang}, title = {LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {442--456}, doi = {10.1145/3575693.3575744}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Sun, Peng	ASPLOS '23: "Lucid: A Non-intrusive, Scalable ..." Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs Qinghao Hu, Meng Zhang, Peng Sun, Yonggang Wen, and Tianwei Zhang (Nanyang Technological University, Singapore; Shanghai AI Laboratory, China; SenseTime, China) While recent deep learning workload schedulers exhibit excellent performance, it is arduous to deploy them in practice due to some substantial defects, including inflexible intrusive manner, exorbitant integration and maintenance cost, limited scalability, as well as opaque decision processes. Motivated by these issues, we design and implement Lucid, a non-intrusive deep learning workload scheduler based on interpretable models. It consists of three innovative modules. First, a two-dimensional optimized profiler is introduced for efficient job metric collection and timely debugging job feedback. Second, Lucid utilizes an indolent packing strategy to circumvent interference. Third, Lucid orchestrates resources based on estimated job priority values and sharing scores to achieve efficient scheduling. Additionally, Lucid promotes model performance maintenance and system transparent adjustment via a well-designed system optimizer. Our evaluation shows that Lucid reduces the average job completion time by up to 1.3× compared with state-of-the-art preemptive scheduler Tiresias. Furthermore, it provides explicit system interpretations and excellent scalability for practical deployment. @InProceedings{ASPLOS23p457, author = {Qinghao Hu and Meng Zhang and Peng Sun and Yonggang Wen and Tianwei Zhang}, title = {Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {457--472}, doi = {10.1145/3575693.3575705}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Sun, Yudi	ASPLOS '23: "TiLT: A Time-Centric Approach ..." TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization Anand Jayarajan, Wei Zhao, Yudi Sun, and Gennady Pekhimenko (University of Toronto, Canada) Stream processing engines (SPEs) are widely used for large scale streaming analytics over unbounded time-ordered data streams. Modern day streaming analytics applications exhibit diverse compute characteristics and demand strict latency and throughput requirements. Over the years, there has been significant attention in building hardware-efficient stream processing engines (SPEs) that support several query optimization, parallelization, and execution strategies to meet the performance requirements of large scale streaming analytics applications. However, in this work, we observe that these strategies often fail to generalize well on many real-world streaming analytics applications due to several inherent design limitations of current SPEs. We further argue that these limitations stem from the shortcomings of the fundamental design choices and the query representation model followed in modern SPEs. To address these challenges, we first propose TiLT, a novel intermediate representation (IR) that offers a highly expressive temporal query language amenable to effective query optimization and parallelization strategies. We subsequently build a compiler backend for TiLT that applies such optimizations on streaming queries and generates hardware-efficient code to achieve high performance on multi-core stream query executions. We demonstrate that TiLT achieves up to 326× (20.49× on average) higher throughput compared to state-of-the-art SPEs (e.g., Trill) across eight real-world streaming analytics applications. TiLT source code is available at https://github.com/ampersand-projects/tilt.git. @InProceedings{ASPLOS23p818, author = {Anand Jayarajan and Wei Zhao and Yudi Sun and Gennady Pekhimenko}, title = {TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {818--832}, doi = {10.1145/3575693.3575704}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Sun, Yunxin	ASPLOS '23: "LeaFTL: A Learning-Based Flash ..." LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives Jinghan Sun, Shaobo Li, Yunxin Sun, Chao Sun, Dejan Vucinic, and Jian Huang (University of Illinois at Urbana-Champaign, USA; ETH Zurich, Switzerland; Western Digital Research, USA) In modern solid-state drives (SSDs), the indexing of flash pages is a critical component in their storage controllers. It not only affects the data access performance, but also determines the efficiency of the precious in-device DRAM resource. A variety of address mapping schemes and optimizations have been proposed. However, most of them were developed with human-driven heuristics. In this paper, we present a learning-based flash translation layer (FTL), named LeaFTL, which learns the address mapping to tolerate dynamic data access patterns via linear regression at runtime. By grouping a large set of mapping entries into a learned segment, it significantly reduces the memory footprint of the address mapping table, which further benefits the data caching in SSD controllers. LeaFTL also employs various optimization techniques, including out-of-band metadata verification to tolerate mispredictions, optimized flash allocation, and dynamic compaction of learned index segments. We implement LeaFTL with both a validated SSD simulator and a real open-channel SSD board. Our evaluation with various storage workloads demonstrates that LeaFTL saves the memory consumption of the mapping table by 2.9× and improves the storage performance by 1.4× on average, in comparison with state-of-the-art FTL schemes. @InProceedings{ASPLOS23p442, author = {Jinghan Sun and Shaobo Li and Yunxin Sun and Chao Sun and Dejan Vucinic and Jian Huang}, title = {LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {442--456}, doi = {10.1145/3575693.3575744}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Switzer, Jennifer	ASPLOS '23: "Junkyard Computing: Repurposing ..." Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon Jennifer Switzer, Gabriel Marcano, Ryan Kastner, and Pat Pannuto (University of California at San Diego, USA) 1.5 billion smartphones are sold annually, and most are decommissioned less than two years later. Most of these unwanted smartphones are neither discarded nor recycled but languish in junk drawers and storage units. This computational stockpile represents a substantial wasted potential: modern smartphones have increasingly high-performance and energy-efficient processors, extensive networking capabilities, and a reliable built-in power supply. This project studies the ability to reuse smartphones as "junkyard computers." Junkyard computers grow global computing capacity by extending device lifetimes, which supplants the manufacture of new devices. We show that the capabilities of even decade-old smartphones are within those demanded by modern cloud microservices and discuss how to combine phones to perform increasingly complex tasks. We describe how current operation-focused metrics do not capture the actual carbon costs of compute. We propose Computational Carbon Intensity---a performance metric that balances the continued service of older devices with the superlinear runtime improvements of newer machines. We use this metric to redefine device service lifetime in terms of carbon efficiency. We develop a cloudlet of reused Pixel 3A phones. We analyze the carbon benefits of deploying large, end-to-end microservice-based applications on these smartphones. Finally, we describe system architectures and associated challenges to scale to cloudlets with hundreds and thousands of smartphones. @InProceedings{ASPLOS23p400, author = {Jennifer Switzer and Gabriel Marcano and Ryan Kastner and Pat Pannuto}, title = {Junkyard Computing: Repurposing Discarded Smartphones to Minimize Carbon}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {400--412}, doi = {10.1145/3575693.3575710}, year = {2023}, } Publisher's Version
	Szekely, Ariel	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Tallam, Sriraman	ASPLOS '23: "Propeller: A Profile Guided, ..." Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications Han Shen, Krzysztof Pszeniczny, Rahman Lavaee, Snehasish Kumar, Sriraman Tallam, and Xinliang David Li (Google, USA; Google, Switzerland) While profile guided optimizations (PGO) and link time optimiza-tions (LTO) have been widely adopted, post link optimizations (PLO)have languished until recently when researchers demonstrated that late injection of profiles can yield significant performance improvements. However, the disassembly-driven, monolithic design of post link optimizers face scaling challenges with large binaries andis at odds with distributed build systems. To reconcile and enable post link optimizations within a distributed build environment, we propose Propeller, a relinking optimizer for warehouse scale work-loads. To enable flexible code layout optimizations, we introduce basic block sections, a novel linker abstraction. Propeller uses basic block sections to enable a new approach to PLO without disassembly. Propeller achieves scalability by relinking the binary using precise profiles instead of rewriting the binary. The overhead of relinking is lowered by caching and leveraging distributed compiler actions during code generation. Propeller has been deployed to production at Google with over tens of millions of cores executing Propeller optimized code at any time. An evaluation of internal warehouse-scale applications show Propeller improves performance by 1.1% to 8% beyond PGO and ThinLTO. Compiler tools such as Clang improve by 7% while MySQL improves by 1%. Compared to the state of the art binary optimizer, Propeller achieves comparable performance while lowering memory overheads by 30%-70% on large benchmarks. @InProceedings{ASPLOS23p617, author = {Han Shen and Krzysztof Pszeniczny and Rahman Lavaee and Snehasish Kumar and Sriraman Tallam and Xinliang David Li}, title = {Propeller: A Profile Guided, Relinking Optimizer for Warehouse-Scale Applications}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {617--631}, doi = {10.1145/3575693.3575727}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Tan, Cheng	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Tang, Jiawei	ASPLOS '23: "BeeHive: Sub-second Elasticity ..." BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution Ziming Zhao, Mingyu Wu, Jiawei Tang, Binyu Zang, Zhaoguo Wang, and Haibo Chen (Shanghai Jiao Tong University, China; Shanghai AI Laboratory, China; Engineering Research Center for Domain-specific Operating Systems of the Ministry of Education of China, China) Function-as-a-service (FaaS), an emerging cloud computing paradigm, is expected to provide strong elasticity due to its promise to auto-scale fine-grained functions rapidly. Although appealing for applications with good parallelism and dynamic workload, this paper shows that it is non-trivial to adapt existing monolithic applications (like web services) to FaaS due to their complexity. To bridge the gap between complicated web services and FaaS, this paper proposes a runtime-based Semi-FaaS execution model, which dynamically extracts time-consuming code snippets (closures) from applications and offloads them to FaaS platforms for execution. It further proposes BeeHive, an offloading framework for Semi-FaaS, which relies on the managed runtime to provide a fallback-based execution model and addresses the performance issues in traditional offloading mechanisms for FaaS. Meanwhile, the runtime system of BeeHive selects offloading candidates in a user-transparent way and supports efficient object sharing, memory management, and failure recovery in a distributed environment. The evaluation using various web applications suggests that the Semi-FaaS execution supported by BeeHive can reach sub-second resource provisioning on commercialized FaaS platforms like AWS Lambda, which is up to two orders of magnitude better than other alternative scaling approaches in cloud computing. @InProceedings{ASPLOS23p74, author = {Ziming Zhao and Mingyu Wu and Jiawei Tang and Binyu Zang and Zhaoguo Wang and Haibo Chen}, title = {BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {74--87}, doi = {10.1145/3575693.3575752}, year = {2023}, } Publisher's Version
	Tang, Yuhua	ASPLOS '23: "NUBA: Non-Uniform Bandwidth ..." NUBA: Non-Uniform Bandwidth GPUs Xia Zhao, Magnus Jahre, Yuhua Tang, Guangda Zhang, and Lieven Eeckhout (Academy of Military Sciences, China; NTNU, Norway; National University of Defense Technology, China; Ghent University, Belgium) The parallel execution model of GPUs enables scaling to hundreds of thousands of threads, which is a key capability that many modern high-performance applications exploit. GPU vendors are hence increasing the compute and memory resources with every GPU generation — resulting in the need to efficiently stitch together a plethora of Symmetric Multiprocessors (SMs), Last-Level Cache (LLC) slices and memory controllers while maximizing bandwidth and keeping power consumption and design complexity in check. Conventional GPUs are Uniform Bandwidth Architectures (UBAs) as they provide equal bandwidth between all SMs and all LLC slices. UBA GPUs require a uniform high-bandwidth Network-on-Chip (NoC), and our key observation is that provisioning a NoC to match the LLC slice bandwidth incurs a hefty power and complexity overhead. We propose the Non-Uniform Bandwidth Architecture (NUBA), a GPU system architecture aimed at fully utilizing LLC slice bandwidth. A NUBA GPU consists of partitions that each feature a few SMs and LLC slices as well as a memory controller — hence exposing the complete LLC bandwidth to the SMs within a partition since they can be connected with point-to-point links — and a NoC between partitions — to enable access to remote data.Exploiting the potential of NUBA GPUs however requires carefully co-designing system software, the compiler and architectural policies. The critical system software component is our Local-And-Balanced (LAB) page placement policy which enables the GPU driver to place data in local partitions while avoiding load imbalance. Moreover, we propose Model-Driven Replication (MDR) which identifies read-only shared data with data-flow analysis at compile time. At run time, MDR leverages an architectural mechanism that replicates read-only shared data across LLC slices when this can be done without pressuring cache capacity. With LAB and MDR, our NUBA GPU improves average performance by 23.1% and 22.2% (and up to 183.9% and 182.4%) compared to iso-resource memory-side and SM-side UBA GPUs, respectively. When the NUBA concept is leveraged to reduce overhead while maintaining similar performance, NUBA reduces NoC power consumption by 12.1× and 9.4×, respectively. @InProceedings{ASPLOS23p544, author = {Xia Zhao and Magnus Jahre and Yuhua Tang and Guangda Zhang and Lieven Eeckhout}, title = {NUBA: Non-Uniform Bandwidth GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {544--559}, doi = {10.1145/3575693.3575745}, year = {2023}, } Publisher's Version
	Tarte, Isha	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Tian, Yongqiang	ASPLOS '23: "Compilation Consistency Modulo ..." Compilation Consistency Modulo Debug Information Theodore Luo Wang, Yongqiang Tian, Yiwen Dong, Zhenyang Xu, and Chengnian Sun (University of Waterloo, Canada) Compilation Consistency Modulo Debug Information (CCMD) is an essential compiler property that a production compiler should support: the compiler should emit the same machine code regardless of enabling debug information. CCMD is vital to developers’ experiences with debugging a production binary containing no debug information. To debug such a binary, developers need build another binary with the same compiler flags and enable debug information. Without CCMD, the machine code in the latter binary will be different, which can confuse the debugger, hide the bug, or even cause a miscompilation (as GCC once did with the Linux Kernel). This paper is the first to introduce to the research community the validation of CCMD, a new research problem that has been overlooked for decades despite its importance. More importantly, we propose the first testing technique Dfusor to automatically validate CCMD for C compilers. At the high level, given a compilable program P as a seed, Dfusor automatically generates compilable program variants via multiple effective program transformations. Such variants can cause a compiler to emit more debug information than it would when compiling P, thus exercising more code paths in the compiler and increasing the chance to find CCMD bugs. Our extensive evaluations of Dfusor demonstrate that Dfusor can produce variants that exhibit significant increases in the quantity and complexity of the emitted debug information, and thus has found new, real bugs in GCC and LLVM. With a sample of 100 variants derived from distinct seed programs, Dfusor introduces 214% more debug information entries and 36% more distinct debug information entries in the variants than the seeds, and improves the code coverage of GCC and Clang by up to 6.00% and 6.82%. More importantly, Dfusor has found CCMD bugs; within 10 months of development and intermittent testing, Dfusor has found 23 bugs (9 in GCC and 14 in Clang), with 3 confirmed and 18 fixed. @InProceedings{ASPLOS23p146, author = {Theodore Luo Wang and Yongqiang Tian and Yiwen Dong and Zhenyang Xu and Chengnian Sun}, title = {Compilation Consistency Modulo Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {146--158}, doi = {10.1145/3575693.3575740}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Tian, Zijie	ASPLOS '23: "Mobius: Fine Tuning Large-Scale ..." Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers Yangyang Feng, Minhui Xie, Zijie Tian, Shuo Wang, Youyou Lu, and Jiwu Shu (Tsinghua University, China) Fine-tuning on cheap commodity GPU servers makes large-scale deep learning models benefit more people. However, the low inter-GPU communication bandwidth and pressing communication contention on the commodity GPU server obstruct training efficiency. In this paper, we present Mobius, a communication-efficient system for fine tuning large-scale models on commodity GPU servers. The key idea is a novel pipeline parallelism scheme enabling heterogeneous memory for large-scale model training, while bringing fewer communications than existing systems. Mobius partitions the model into stages and carefully schedules them between GPU memory and DRAM to overlap communication with computation. It formulates pipeline execution into a mixed-integer program problem to find the optimal pipeline partition. It also features a new stage-to-GPU mapping method termed cross mapping, to minimize communication contention. Experiments on various scale models and GPU topologies show that Mobius significantly reduces the training time by 3.8-5.1× compared with the prior art. @InProceedings{ASPLOS23p489, author = {Yangyang Feng and Minhui Xie and Zijie Tian and Shuo Wang and Youyou Lu and Jiwu Shu}, title = {Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {489--501}, doi = {10.1145/3575693.3575703}, year = {2023}, } Publisher's Version
	Torres, Carlos	ASPLOS '23: "Ditto: End-to-End Application ..." Ditto: End-to-End Application Cloning for Networked Cloud Services Mingyu Liang, Yu Gan, Yueying Li, Carlos Torres, Abhishek Dhanotia, Mahesh Ketkar, and Christina Delimitrou (Cornell University, USA; Meta, USA; Intel, USA; Massachusetts Institute of Technology, USA) The lack of representative, publicly-available cloud services has been a recurring problem in the architecture and systems communities. While open-source benchmarks exist, they do not capture the full complexity of cloud services. Application cloning is a promising way to address this, however, prior work is limited to CPU-/cache-centric, single-node services, operating at user level. We present Ditto, an automated framework for cloning end-to-end cloud applications, both monolithic and microservices, which captures I/O and network activity, as well as kernel operations, in addition to application logic. Ditto takes a hierarchical approach to application cloning, starting with capturing the dependency graph across distributed services, to recreating each tier's control/data flow, and finally generating system calls and assembly that mimics the individual applications. Ditto does not reveal the logic of the original application, facilitating publicly sharing clones of production services with hardware vendors, cloud providers, and the research community. We show that across a diverse set of single- and multi-tier applications, Ditto accurately captures their CPU and memory characteristics as well as their high-level performance metrics, is portable across platforms, and facilitates a wide range of system studies. @InProceedings{ASPLOS23p222, author = {Mingyu Liang and Yu Gan and Yueying Li and Carlos Torres and Abhishek Dhanotia and Mahesh Ketkar and Christina Delimitrou}, title = {Ditto: End-to-End Application Cloning for Networked Cloud Services}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {222--236}, doi = {10.1145/3575693.3575751}, year = {2023}, } Publisher's Version
	Vafeiadis, Viktor	ASPLOS '23: "AtoMig: Automatically Migrating ..." AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM Martin Beck, Koustubha Bhat, Lazar Stričević, Geng Chen, Diogo Behrens, Ming Fu, Viktor Vafeiadis, Haibo Chen, and Hermann Härtig (Huawei Dresden Research Center, Dresden, Germany; Huawei Fundamental Software Innovation Lab, Shenzhen, China; MPI-SWS, Kaiserslautern, Germany; Huawei Central Software Institute, Shenzhen, China; Shanghai Jiao Tong University, Shanghai, China; TU Dresden, Dresden, Germany) CPUs with weak memory-consistency models (WMMs), such as Arm and RISC-V, are rapidly increasing their market share. Porting legacy x86 applications to such CPUs requires introducing extra synchronization to prevent WMM-related concurrency bugs---a task often left to human experts. Given the rarity of such experts and the enormous size of legacy applications, we develop AtoMig, an effective, fully automated tool for porting large, real-world applications to WMM CPU architectures. AtoMig detects shared memory access patterns with novel static analysis strategies and performs program transformations to properly protect them from WMM effects. In the absence of sufficiently scalable verification methods, AtoMig shows practicality of focusing on code patterns more prone to WMM faults, trading off completeness for scalability. We validate the correctness of AtoMig's transformations on several small concurrent benchmarks via model checking. We demonstrate the scalability and performance of our approach by applying AtoMig to popular real-world large code bases with up to millions of lines of code, viz., MariaDB, Postgres, SQlite, LevelDB, and Memcached. As part of this work, we also found a WMM bug in MariaDB, which AtoMig fixes automatically. @InProceedings{ASPLOS23p61, author = {Martin Beck and Koustubha Bhat and Lazar Stričević and Geng Chen and Diogo Behrens and Ming Fu and Viktor Vafeiadis and Haibo Chen and Hermann Härtig}, title = {AtoMig: Automatically Migrating Millions Lines of Code from TSO to WMM}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {61--73}, doi = {10.1145/3575693.3579849}, year = {2023}, } Publisher's Version
	Vainbrand, Dmitri	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Vigfusson, Ymir	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version
	Vucinic, Dejan	ASPLOS '23: "LeaFTL: A Learning-Based Flash ..." LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives Jinghan Sun, Shaobo Li, Yunxin Sun, Chao Sun, Dejan Vucinic, and Jian Huang (University of Illinois at Urbana-Champaign, USA; ETH Zurich, Switzerland; Western Digital Research, USA) In modern solid-state drives (SSDs), the indexing of flash pages is a critical component in their storage controllers. It not only affects the data access performance, but also determines the efficiency of the precious in-device DRAM resource. A variety of address mapping schemes and optimizations have been proposed. However, most of them were developed with human-driven heuristics. In this paper, we present a learning-based flash translation layer (FTL), named LeaFTL, which learns the address mapping to tolerate dynamic data access patterns via linear regression at runtime. By grouping a large set of mapping entries into a learned segment, it significantly reduces the memory footprint of the address mapping table, which further benefits the data caching in SSD controllers. LeaFTL also employs various optimization techniques, including out-of-band metadata verification to tolerate mispredictions, optimized flash allocation, and dynamic compaction of learned index segments. We implement LeaFTL with both a validated SSD simulator and a real open-channel SSD board. Our evaluation with various storage workloads demonstrates that LeaFTL saves the memory consumption of the mapping table by 2.9× and improves the storage performance by 1.4× on average, in comparison with state-of-the-art FTL schemes. @InProceedings{ASPLOS23p442, author = {Jinghan Sun and Shaobo Li and Yunxin Sun and Chao Sun and Dejan Vucinic and Jian Huang}, title = {LeaFTL: A Learning-Based Flash Translation Layer for Solid-State Drives}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {442--456}, doi = {10.1145/3575693.3575744}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Wang, Mian	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Wang, Ruipu	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version
	Wang, Shuo	ASPLOS '23: "Mobius: Fine Tuning Large-Scale ..." Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers Yangyang Feng, Minhui Xie, Zijie Tian, Shuo Wang, Youyou Lu, and Jiwu Shu (Tsinghua University, China) Fine-tuning on cheap commodity GPU servers makes large-scale deep learning models benefit more people. However, the low inter-GPU communication bandwidth and pressing communication contention on the commodity GPU server obstruct training efficiency. In this paper, we present Mobius, a communication-efficient system for fine tuning large-scale models on commodity GPU servers. The key idea is a novel pipeline parallelism scheme enabling heterogeneous memory for large-scale model training, while bringing fewer communications than existing systems. Mobius partitions the model into stages and carefully schedules them between GPU memory and DRAM to overlap communication with computation. It formulates pipeline execution into a mixed-integer program problem to find the optimal pipeline partition. It also features a new stage-to-GPU mapping method termed cross mapping, to minimize communication contention. Experiments on various scale models and GPU topologies show that Mobius significantly reduces the training time by 3.8-5.1× compared with the prior art. @InProceedings{ASPLOS23p489, author = {Yangyang Feng and Minhui Xie and Zijie Tian and Shuo Wang and Youyou Lu and Jiwu Shu}, title = {Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {489--501}, doi = {10.1145/3575693.3575703}, year = {2023}, } Publisher's Version
	Wang, Theodore Luo	ASPLOS '23: "Compilation Consistency Modulo ..." Compilation Consistency Modulo Debug Information Theodore Luo Wang, Yongqiang Tian, Yiwen Dong, Zhenyang Xu, and Chengnian Sun (University of Waterloo, Canada) Compilation Consistency Modulo Debug Information (CCMD) is an essential compiler property that a production compiler should support: the compiler should emit the same machine code regardless of enabling debug information. CCMD is vital to developers’ experiences with debugging a production binary containing no debug information. To debug such a binary, developers need build another binary with the same compiler flags and enable debug information. Without CCMD, the machine code in the latter binary will be different, which can confuse the debugger, hide the bug, or even cause a miscompilation (as GCC once did with the Linux Kernel). This paper is the first to introduce to the research community the validation of CCMD, a new research problem that has been overlooked for decades despite its importance. More importantly, we propose the first testing technique Dfusor to automatically validate CCMD for C compilers. At the high level, given a compilable program P as a seed, Dfusor automatically generates compilable program variants via multiple effective program transformations. Such variants can cause a compiler to emit more debug information than it would when compiling P, thus exercising more code paths in the compiler and increasing the chance to find CCMD bugs. Our extensive evaluations of Dfusor demonstrate that Dfusor can produce variants that exhibit significant increases in the quantity and complexity of the emitted debug information, and thus has found new, real bugs in GCC and LLVM. With a sample of 100 variants derived from distinct seed programs, Dfusor introduces 214% more debug information entries and 36% more distinct debug information entries in the variants than the seeds, and improves the code coverage of GCC and Clang by up to 6.00% and 6.82%. More importantly, Dfusor has found CCMD bugs; within 10 months of development and intermittent testing, Dfusor has found 23 bugs (9 in GCC and 14 in Clang), with 3 confirmed and 18 fixed. @InProceedings{ASPLOS23p146, author = {Theodore Luo Wang and Yongqiang Tian and Yiwen Dong and Zhenyang Xu and Chengnian Sun}, title = {Compilation Consistency Modulo Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {146--158}, doi = {10.1145/3575693.3575740}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Wang, Yida	ASPLOS '23: "Hidet: Task-Mapping Programming ..." Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding, Cody Hao Yu, Bojian Zheng, Yizhi Liu, Yida Wang, and Gennady Pekhimenko (University of Toronto, Canada; Vector Institute, Canada; Amazon Web Services, USA) As deep learning models nowadays are widely adopted by both cloud services and edge devices, reducing the latency of deep learning model inferences becomes crucial to provide efficient model serving. However, it is challenging to develop efficient tensor programs for deep learning operators due to the high complexity of modern accelerators (e.g., NVIDIA GPUs and Google TPUs) and the rapidly growing number of operators. Deep learning compilers, such as Apache TVM, adopt declarative scheduling primitives to lower the bar of developing tensor programs. However, we show that this approach is insufficient to cover state-of-the-art tensor program optimizations (e.g., double buffering). In this paper, we propose to embed the scheduling process into tensor programs and use dedicated mappings, called task mappings, to define the computation assignment and ordering directly in the tensor programs. This new approach greatly enriches the expressible optimizations by allowing developers to manipulate tensor programs at a much finer granularity (e.g., allowing program-statement-level optimizations). We call the proposed method the task-mapping programming paradigm. In addition, we propose a new post-scheduling fusion optimization that allows developers to focus on scheduling every single operator and automates the fusion after scheduling. It greatly reduces the engineering efforts for operator fusion. Our proposed paradigm also constructs an efficient hardware-centric schedule space, which is agnostic to the program input size and greatly reduces the tuning time. With the proposed paradigm, we implement a deep learning compiler Hidet. Extensive experiments on modern convolution and transformer models show that Hidet outperforms state-of-the-art DNN inference framework, ONNX Runtime, and compiler, TVM equipped with scheduler AutoTVM and Ansor, by up to 1.48x (1.22x on average). It also reduces the tuning time by 20x and 11x compared with AutoTVM and Ansor, respectively. We open-sourced hidet at https://www.github.com/hidet-org/hidet. @InProceedings{ASPLOS23p370, author = {Yaoyao Ding and Cody Hao Yu and Bojian Zheng and Yizhi Liu and Yida Wang and Gennady Pekhimenko}, title = {Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {370--384}, doi = {10.1145/3575693.3575702}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Wang, Zhaoguo	ASPLOS '23: "BeeHive: Sub-second Elasticity ..." BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution Ziming Zhao, Mingyu Wu, Jiawei Tang, Binyu Zang, Zhaoguo Wang, and Haibo Chen (Shanghai Jiao Tong University, China; Shanghai AI Laboratory, China; Engineering Research Center for Domain-specific Operating Systems of the Ministry of Education of China, China) Function-as-a-service (FaaS), an emerging cloud computing paradigm, is expected to provide strong elasticity due to its promise to auto-scale fine-grained functions rapidly. Although appealing for applications with good parallelism and dynamic workload, this paper shows that it is non-trivial to adapt existing monolithic applications (like web services) to FaaS due to their complexity. To bridge the gap between complicated web services and FaaS, this paper proposes a runtime-based Semi-FaaS execution model, which dynamically extracts time-consuming code snippets (closures) from applications and offloads them to FaaS platforms for execution. It further proposes BeeHive, an offloading framework for Semi-FaaS, which relies on the managed runtime to provide a fallback-based execution model and addresses the performance issues in traditional offloading mechanisms for FaaS. Meanwhile, the runtime system of BeeHive selects offloading candidates in a user-transparent way and supports efficient object sharing, memory management, and failure recovery in a distributed environment. The evaluation using various web applications suggests that the Semi-FaaS execution supported by BeeHive can reach sub-second resource provisioning on commercialized FaaS platforms like AWS Lambda, which is up to two orders of magnitude better than other alternative scaling approaches in cloud computing. @InProceedings{ASPLOS23p74, author = {Ziming Zhao and Mingyu Wu and Jiawei Tang and Binyu Zang and Zhaoguo Wang and Haibo Chen}, title = {BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {74--87}, doi = {10.1145/3575693.3575752}, year = {2023}, } Publisher's Version
	Wen, Yonggang	ASPLOS '23: "Lucid: A Non-intrusive, Scalable ..." Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs Qinghao Hu, Meng Zhang, Peng Sun, Yonggang Wen, and Tianwei Zhang (Nanyang Technological University, Singapore; Shanghai AI Laboratory, China; SenseTime, China) While recent deep learning workload schedulers exhibit excellent performance, it is arduous to deploy them in practice due to some substantial defects, including inflexible intrusive manner, exorbitant integration and maintenance cost, limited scalability, as well as opaque decision processes. Motivated by these issues, we design and implement Lucid, a non-intrusive deep learning workload scheduler based on interpretable models. It consists of three innovative modules. First, a two-dimensional optimized profiler is introduced for efficient job metric collection and timely debugging job feedback. Second, Lucid utilizes an indolent packing strategy to circumvent interference. Third, Lucid orchestrates resources based on estimated job priority values and sharing scores to achieve efficient scheduling. Additionally, Lucid promotes model performance maintenance and system transparent adjustment via a well-designed system optimizer. Our evaluation shows that Lucid reduces the average job completion time by up to 1.3× compared with state-of-the-art preemptive scheduler Tiresias. Furthermore, it provides explicit system interpretations and excellent scalability for practical deployment. @InProceedings{ASPLOS23p457, author = {Qinghao Hu and Meng Zhang and Peng Sun and Yonggang Wen and Tianwei Zhang}, title = {Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {457--472}, doi = {10.1145/3575693.3575705}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Wentzlaff, David	ASPLOS '23: "SMAPPIC: Scalable Multi-FPGA ..." SMAPPIC: Scalable Multi-FPGA Architecture Prototype Platform in the Cloud Grigory Chirkov and David Wentzlaff (Princeton University, USA) Traditionally, architecture prototypes are built on top of FPGA infrastructure, with two associated problems. First, very large FPGAs are prohibitively expensive for most people and institutions. Second, the burden of FPGA development adds to an already uneasy life of researchers, especially those who focus on software. Large designs that do not fit into a single FPGA exacerbate these issues even more. This work presents SMAPPIC — the first open-source prototype platform for shared memory multi-die architectures on cloud FPGAs. SMAPPIC leverages the OpenPiton/BYOC infrastructure and AWS F1 instances to make FPGA-based prototypes of System-on-Chips, processor cores, accelerators, cache subsystems, etc., cheap, scalable, and straightforward. SMAPPIC enables many use cases that are not possible or significantly more complicated in existing software and FPGA tools. This work has the potential to accelerate the rate of innovation in computer engineering fields in the nearest future. @InProceedings{ASPLOS23p733, author = {Grigory Chirkov and David Wentzlaff}, title = {SMAPPIC: Scalable Multi-FPGA Architecture Prototype Platform in the Cloud}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {733--746}, doi = {10.1145/3575693.3575753}, year = {2023}, } Publisher's Version
	Wiranata, Fandi A.	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version
	Won, Jaeyeon	ASPLOS '23: "WACO: Learning Workload-Aware ..." WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program Jaeyeon Won, Charith Mendis, Joel S. Emer, and Saman Amarasinghe (Massachusetts Institute of Technology, USA; University of Illinois at Urbana-Champaign, USA; NVIDIA, USA) In this paper, we present WACO, a novel method of co-optimizing the format and the schedule of a given sparsity pattern in a sparse tensor program. A core challenge in this paper is the design of a lightweight cost model that accurately predicts the runtime of a sparse tensor program by considering the sparsity pattern, the format, and the schedule. The key idea in addressing this is exploiting a sparse convolutional network to learn meaningful features of the sparsity pattern and embedding a coupled behavior between the format and the schedule using a specially designed schedule template. In addition, within the enormous search space of co-optimization, our novel search strategy, an approximate nearest neighbor search, efficiently and accurately retrieves the best format and schedule for a given sparsity pattern. We evaluated WACO for four different algorithms (SpMV, SpMM, SDDMM, and MTTKRP) on a CPU using 726 different sparsity patterns. Our experimental results showed that WACO outperformed four state-of-the-art baselines, Intel MKL, BestFormat, TACO with a default schedule, and ASpT. Compared to the best of four baselines, WACO achieved 1.43×, 1.18×, 1.14×, and 1.27× average speedups on SpMV, SpMM, SDDMM, and MTTKRP, respectively. @InProceedings{ASPLOS23p920, author = {Jaeyeon Won and Charith Mendis and Joel S. Emer and Saman Amarasinghe}, title = {WACO: Learning Workload-Aware Co-optimization of the Format and Schedule of a Sparse Tensor Program}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {920--934}, doi = {10.1145/3575693.3575742}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Wu, Carole-Jean	ASPLOS '23: "Carbon Explorer: A Holistic ..." Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters Bilge Acun, Benjamin Lee, Fiodar Kazhamiaka, Kiwan Maeng, Udit Gupta, Manoj Chakkaravarthy, David Brooks, and Carole-Jean Wu (Meta, USA; University of Pennsylvania, USA; Stanford University, USA; Harvard University, USA) Technology companies reduce their datacenters’ carbon footprint by investing in renewable energy generation and receiving credits from power purchase agreements. Annually, datacenters offset their energy consumption with generation credits (Net Zero). But hourly, datacenters often consume carbon-intensive energy from the grid when carbon-free energy is scarce. Relying on intermittent renewable energy in every hour (24/7) requires a mix of renewable energy from complementary sources, energy storage, and workload scheduling. In this paper, we present the Carbon Explorer framework to analyze the solution space. We use Carbon Explorer to balance trade-offs between operational and embodied carbon, optimizing the mix of solutions for 24/7 carbon-free datacenter operation based on geographic location and workload. Carbon Explorer has been open-sourced at https://github.com/facebookresearch/CarbonExplorer. @InProceedings{ASPLOS23p118, author = {Bilge Acun and Benjamin Lee and Fiodar Kazhamiaka and Kiwan Maeng and Udit Gupta and Manoj Chakkaravarthy and David Brooks and Carole-Jean Wu}, title = {Carbon Explorer: A Holistic Framework for Designing Carbon Aware Datacenters}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {118--132}, doi = {10.1145/3575693.3575754}, year = {2023}, } Publisher's Version
	Wu, Hanqing	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Wu, Mingyu	ASPLOS '23: "BeeHive: Sub-second Elasticity ..." BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution Ziming Zhao, Mingyu Wu, Jiawei Tang, Binyu Zang, Zhaoguo Wang, and Haibo Chen (Shanghai Jiao Tong University, China; Shanghai AI Laboratory, China; Engineering Research Center for Domain-specific Operating Systems of the Ministry of Education of China, China) Function-as-a-service (FaaS), an emerging cloud computing paradigm, is expected to provide strong elasticity due to its promise to auto-scale fine-grained functions rapidly. Although appealing for applications with good parallelism and dynamic workload, this paper shows that it is non-trivial to adapt existing monolithic applications (like web services) to FaaS due to their complexity. To bridge the gap between complicated web services and FaaS, this paper proposes a runtime-based Semi-FaaS execution model, which dynamically extracts time-consuming code snippets (closures) from applications and offloads them to FaaS platforms for execution. It further proposes BeeHive, an offloading framework for Semi-FaaS, which relies on the managed runtime to provide a fallback-based execution model and addresses the performance issues in traditional offloading mechanisms for FaaS. Meanwhile, the runtime system of BeeHive selects offloading candidates in a user-transparent way and supports efficient object sharing, memory management, and failure recovery in a distributed environment. The evaluation using various web applications suggests that the Semi-FaaS execution supported by BeeHive can reach sub-second resource provisioning on commercialized FaaS platforms like AWS Lambda, which is up to two orders of magnitude better than other alternative scaling approaches in cloud computing. @InProceedings{ASPLOS23p74, author = {Ziming Zhao and Mingyu Wu and Jiawei Tang and Binyu Zang and Zhaoguo Wang and Haibo Chen}, title = {BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {74--87}, doi = {10.1145/3575693.3575752}, year = {2023}, } Publisher's Version
	Wu, Wenfei	ASPLOS '23: "A Generic Service to Provide ..." A Generic Service to Provide In-Network Aggregation for Key-Value Streams Yongchao He, Wenfei Wu, Yanfang Le, Ming Liu, and ChonLam Lao (Tsinghua University, China; Peking University, China; Intel, USA; University of Wisconsin-Madison, USA; Harvard University, USA) Key-value stream aggregation is a common operation in distributed systems, which requires intensive computation and network resources. We propose a generic in-network aggregation service for key-value streams, ASK, to accelerate the aggregation operations in diverse distributed applications. ASK is a switch-host co-designed system, where the programmable switch provides a best-effort aggregation service, and the host runs a daemon to interact with applications. ASK makes in-depth optimization tailored to traffic characteristics, hardware restrictions, and network unreliable natures: it vectorizes multiple key-value tuples’ aggregation of one packet in one switch pipeline pass, which improves the per-host’s goodput; it develops a lightweight reliability mechanism for key-value stream’s asynchronous aggregation, which guarantees computation correctness; it designs a hot-key agnostic prioritization for key-skewed workloads, which improves the switch memory utilization. We prototype ASK and use it to support Spark and BytePS. The evaluation shows that ASK could accelerate pure key-value aggregation tasks by up to 155 times and big data jobs by 3-5 times, and be backward compatible with existing INA-empowered distributed training solutions with the same speedup. @InProceedings{ASPLOS23p33, author = {Yongchao He and Wenfei Wu and Yanfang Le and Ming Liu and ChonLam Lao}, title = {A Generic Service to Provide In-Network Aggregation for Key-Value Streams}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {33--47}, doi = {10.1145/3575693.3575708}, year = {2023}, } Publisher's Version
	Wu, Yongwei	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version
	Xiao, Haocheng	ASPLOS '23: "Hacky Racers: Exploiting Instruction-Level ..." Hacky Racers: Exploiting Instruction-Level Parallelism to Generate Stealthy Fine-Grained Timers Haocheng Xiao and Sam Ainsworth (University of Edinburgh, UK) Side-channel attacks pose serious threats to many security models, especially sandbox-based browsers. While transient-execution side channels in out-of-order processors have previously been blamed for vulnerabilities such as Spectre and Meltdown, we show that in fact, the capability of out-of-order execution itself to cause mayhem is far more general. We develop Hacky Racers, a new type of timing gadget that uses instruction-level parallelism, another key feature of out-of-order execution, to measure arbitrary fine-grained timing differences, even in the presence of highly restricted JavaScript sandbox environments. While such environments try to mitigate timing side channels by reducing timer precision and removing language features such as SharedArrayBuffer that can be used to indirectly generate timers via thread-level parallelism, no such restrictions can be designed to limit Hacky Racers. We also design versions of Hacky Racers that require no misspeculation whatsoever, demonstrating that transient execution is not the only threat to security from modern microarchitectural performance optimization. We use Hacky Racers to construct novel backwards-in-time Spectre gadgets, which break many hardware countermeasures in the literature by leaking secrets before misspeculation is discovered. We also use them to generate the first known last-level cache eviction set generator in JavaScript that does not require SharedArrayBuffer support. @InProceedings{ASPLOS23p354, author = {Haocheng Xiao and Sam Ainsworth}, title = {Hacky Racers: Exploiting Instruction-Level Parallelism to Generate Stealthy Fine-Grained Timers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {354--369}, doi = {10.1145/3575693.3575700}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Xie, Minhui	ASPLOS '23: "Mobius: Fine Tuning Large-Scale ..." Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers Yangyang Feng, Minhui Xie, Zijie Tian, Shuo Wang, Youyou Lu, and Jiwu Shu (Tsinghua University, China) Fine-tuning on cheap commodity GPU servers makes large-scale deep learning models benefit more people. However, the low inter-GPU communication bandwidth and pressing communication contention on the commodity GPU server obstruct training efficiency. In this paper, we present Mobius, a communication-efficient system for fine tuning large-scale models on commodity GPU servers. The key idea is a novel pipeline parallelism scheme enabling heterogeneous memory for large-scale model training, while bringing fewer communications than existing systems. Mobius partitions the model into stages and carefully schedules them between GPU memory and DRAM to overlap communication with computation. It formulates pipeline execution into a mixed-integer program problem to find the optimal pipeline partition. It also features a new stage-to-GPU mapping method termed cross mapping, to minimize communication contention. Experiments on various scale models and GPU topologies show that Mobius significantly reduces the training time by 3.8-5.1× compared with the prior art. @InProceedings{ASPLOS23p489, author = {Yangyang Feng and Minhui Xie and Zijie Tian and Shuo Wang and Youyou Lu and Jiwu Shu}, title = {Mobius: Fine Tuning Large-Scale Models on Commodity GPU Servers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {489--501}, doi = {10.1145/3575693.3575703}, year = {2023}, } Publisher's Version
	Xie, Yuan	ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Xiong, Jinjun	ASPLOS '23: "GPU-Initiated On-Demand High-Throughput ..." GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture Zaid Qureshi, Vikram Sharma Mailthody, Isaac Gelado, Seungwon Min, Amna Masood, Jeongmin Park, Jinjun Xiong, C. J. Newburn, Dmitri Vainbrand, I-Hsin Chung, Michael Garland, William Dally, and Wen-mei Hwu (University of Illinois at Urbana-Champaign, USA; NVIDIA, USA; AMD, USA; University at Buffalo, USA; IBM Research, USA; Stanford University, USA) Graphics Processing Units (GPUs) have traditionally relied on the {host} CPU to initiate access to the data storage. This approach is well-suited for GPU applications with known data access patterns that enable partitioning of their dataset to be processed in a pipelined fashion in the GPU. However, emerging applications such as graph and data analytics, recommender systems, or graph neural networks, require fine-grained, data-dependent access to storage. CPU initiation of storage access is unsuitable for these applications due to high CPU-GPU synchronization overheads, I/O traffic amplification, and long CPU processing latencies. GPU-initiated storage removes these overheads from the storage control path and, thus, can potentially support these applications at much higher speed. However, there is a lack of systems architecture and software stack that enable efficient GPU-initiated storage access. This work presents a novel system architecture, BaM, that fills this gap. BaM features a fine-grained software cache to coalesce data storage requests while minimizing I/O traffic amplification. This software cache communicates with the storage system via high-throughput queues that enable the massive number of concurrent threads in modern GPUs to make I/O requests at a high rate to fully utilize the storage devices and the system interconnect. Experimental results show that BaM delivers 1.0x and 1.49x end-to-end speed up for BFS and CC graph analytics benchmarks while reducing hardware costs by up to 21.7x over accessing the graph data from the host memory. Furthermore, BaM speeds up data-analytics workloads by 5.3x over CPU-initiated storage access on the same hardware. @InProceedings{ASPLOS23p325, author = {Zaid Qureshi and Vikram Sharma Mailthody and Isaac Gelado and Seungwon Min and Amna Masood and Jeongmin Park and Jinjun Xiong and C. J. Newburn and Dmitri Vainbrand and I-Hsin Chung and Michael Garland and William Dally and Wen-mei Hwu}, title = {GPU-Initiated On-Demand High-Throughput Storage Access in the BaM System Architecture}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {325--339}, doi = {10.1145/3575693.3575748}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Xiong, Qian	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Xiong, Yifan	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "MSCCLang: Microsoft Collective ..." MSCCLang: Microsoft Collective Communication Language Meghan Cowan, Saeed Maleki, Madanlal Musuvathi, Olli Saarikivi, and Yifan Xiong (Microsoft Research, USA; Microsoft Research, China) Machine learning models with millions or billions of parameters are increasingly trained and served on large multi-GPU systems. As models grow in size and execute on more GPUs, collective communication becomes a bottleneck. Custom collective algorithms optimized for both particular network topologies and application-specific communication patterns can alleviate this bottleneck and help these applications scale. However, implementing correct and efficient custom algorithms is challenging. This paper introduces MSCCLang, a system for programmable GPU communication. MSCCLang provides a domain specific language for writing collective communication algorithms and an optimizing compiler for lowering them to an executable form, which can be executed efficiently and flexibly in an interpreter-based runtime. We used MSCCLang to write novel collective algorithms for AllReduce and AllToAll that are up to 1.9× and 1.3× faster than hand-optimized implementations, respectively. @InProceedings{ASPLOS23p502, author = {Meghan Cowan and Saeed Maleki and Madanlal Musuvathi and Olli Saarikivi and Yifan Xiong}, title = {MSCCLang: Microsoft Collective Communication Language}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {502--514}, doi = {10.1145/3575693.3575724}, year = {2023}, } Publisher's Version
	Xu, Yuanchao	ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Xu, Zhenyang	ASPLOS '23: "Compilation Consistency Modulo ..." Compilation Consistency Modulo Debug Information Theodore Luo Wang, Yongqiang Tian, Yiwen Dong, Zhenyang Xu, and Chengnian Sun (University of Waterloo, Canada) Compilation Consistency Modulo Debug Information (CCMD) is an essential compiler property that a production compiler should support: the compiler should emit the same machine code regardless of enabling debug information. CCMD is vital to developers’ experiences with debugging a production binary containing no debug information. To debug such a binary, developers need build another binary with the same compiler flags and enable debug information. Without CCMD, the machine code in the latter binary will be different, which can confuse the debugger, hide the bug, or even cause a miscompilation (as GCC once did with the Linux Kernel). This paper is the first to introduce to the research community the validation of CCMD, a new research problem that has been overlooked for decades despite its importance. More importantly, we propose the first testing technique Dfusor to automatically validate CCMD for C compilers. At the high level, given a compilable program P as a seed, Dfusor automatically generates compilable program variants via multiple effective program transformations. Such variants can cause a compiler to emit more debug information than it would when compiling P, thus exercising more code paths in the compiler and increasing the chance to find CCMD bugs. Our extensive evaluations of Dfusor demonstrate that Dfusor can produce variants that exhibit significant increases in the quantity and complexity of the emitted debug information, and thus has found new, real bugs in GCC and LLVM. With a sample of 100 variants derived from distinct seed programs, Dfusor introduces 214% more debug information entries and 36% more distinct debug information entries in the variants than the seeds, and improves the code coverage of GCC and Clang by up to 6.00% and 6.82%. More importantly, Dfusor has found CCMD bugs; within 10 months of development and intermittent testing, Dfusor has found 23 bugs (9 in GCC and 14 in Clang), with 3 confirmed and 18 fixed. @InProceedings{ASPLOS23p146, author = {Theodore Luo Wang and Yongqiang Tian and Yiwen Dong and Zhenyang Xu and Chengnian Sun}, title = {Compilation Consistency Modulo Debug Information}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {146--158}, doi = {10.1145/3575693.3575740}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional
	Xygkis, Athanasios	ASPLOS '23: "uBFT: Microsecond-Scale BFT ..." uBFT: Microsecond-Scale BFT using Disaggregated Memory Marcos K. Aguilera, Naama Ben-David, Rachid Guerraoui, Antoine Murat, Athanasios Xygkis, and Igor Zablotchi (VMware Research, USA; EPFL, Switzerland; Massachusetts Institute of Technology, USA) We propose uBFT, the first State Machine Replication (SMR) system to achieve microsecond-scale latency in data centers, while using only 2f+1 replicas to tolerate f Byzantine failures. The Byzantine Fault Tolerance (BFT) provided by uBFT is essential as pure crashes appear to be a mere illusion with real-life systems reportedly failing in many unexpected ways. uBFT relies on a small non-tailored trusted computing base—disaggregated memory—and consumes a practically bounded amount of memory. uBFT is based on a novel abstraction called Consistent Tail Broadcast, which we use to prevent equivocation while bounding memory. We implement uBFT using RDMA-based disaggregated memory and obtain an end-to-end latency of as little as 10 us. This is at least 50× faster than MinBFT, a state-of-the-art 2f+1 BFT SMR based on Intel’s SGX. We use uBFT to replicate two KV-stores (Memcached and Redis), as well as a financial order matching engine (Liquibook). These applications have low latency (up to 20 us) and become Byzantine tolerant with as little as 10 us more. The price for uBFT is a small amount of reliable disaggregated memory (less than 1 MiB), which in our prototype consists of a small number of memory servers connected through RDMA and replicated for fault tolerance. @InProceedings{ASPLOS23p862, author = {Marcos K. Aguilera and Naama Ben-David and Rachid Guerraoui and Antoine Murat and Athanasios Xygkis and Igor Zablotchi}, title = {uBFT: Microsecond-Scale BFT using Disaggregated Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {862--877}, doi = {10.1145/3575693.3575732}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yang, Dingqing	ASPLOS '23: "HuffDuff: Stealing Pruned ..." HuffDuff: Stealing Pruned DNNs from Sparse Accelerators Dingqing Yang, Prashant J. Nair, and Mieszko Lis (University of British Columbia, Canada) Deep learning models are a valuable “secret sauce” that confers a significant competitive advantage. Many models are never visible to the user and even publicly known state-of-the-art models are either completely proprietary or only accessible via access-controlled APIs. Increasingly, these models run directly on the edge, often using a low-power DNN accelerator. This makes models particularly vulnerable, as an attacker with physical access can exploit side channels like off-chip memory access volumes. Indeed, prior work has shown that this channel can be used to steal dense DNNs from edge devices by correlating data transfer volumes with layer geometry. Unfortunately, prior techniques become intractable when the model is sparse in either weights or activations because off-chip transfers no longer correspond exactly to layer dimensions. Could it be that the many mobile-class sparse accelerators are inherently safe from this style of attack? In this paper, we show that it is feasible to steal a pruned DNN model architecture from a mobile-class sparse accelerator using the DRAM access volume channel. We describe HuffDuff, an attack scheme with two novel techniques that leverage (i) the boundary effect present in CONV layers, and (ii) the timing side channel of on-the-fly activation compression. Together, these techniques dramatically reduce the space of possible model architectures up to 94 orders of magnitude, resulting in fewer than 100 candidate models — a number that can be feasibly tested. Finally, we sample network instances from our solution space and show that (i) our solutions reach the victim accuracy under the iso-footprint constraint, and (ii) significantly improve black-box targeted attack success rates. @InProceedings{ASPLOS23p385, author = {Dingqing Yang and Prashant J. Nair and Mieszko Lis}, title = {HuffDuff: Stealing Pruned DNNs from Sparse Accelerators}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {385--399}, doi = {10.1145/3575693.3575738}, year = {2023}, } Publisher's Version
	Yang, Fan	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yang, Hailong	ASPLOS '23: "VClinic: A Portable and Efficient ..." VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers Xin You, Hailong Yang, Kelun Lei, Zhongzhi Luan, and Depei Qian (Beihang University, China) Fine-grained value profilers reveal a promising way to accurately detect value-related software inefficiencies with binary instrumentation. Due to the architecture-dependent implementation details of binary instrumentation, existing value profilers suffer from poor portability as well as high engineering efforts to achieve efficiency across platforms. In this paper, we propose VClinic, a portable and efficient fine-grained value profiling framework for analyzing highly optimized binaries on both X86 and ARM platforms. VClinic exploits operand-centric two-level designs in its implementation to provide the common building blocks required for value profilers. By constructing four representative value profilers with VClinic, we demonstrate that VClinic can ease the development of value profilers with portability and efficiency across platforms. Guided by the value profilers built upon VClinic, we can achieve up to 89.94% and 74.66% speedup for real-world programs on X86 and ARM platforms, respectively. @InProceedings{ASPLOS23p892, author = {Xin You and Hailong Yang and Kelun Lei and Zhongzhi Luan and Depei Qian}, title = {VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {892--904}, doi = {10.1145/3575693.3576934}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yang, Ke	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version
	Yang, Shuangyan	ASPLOS '23: "Betty: Enabling Large-Scale ..." Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning Shuangyan Yang, Minjia Zhang, Wenqian Dong, and Dong Li (University of California at Merced, USA; Microsoft Research, USA; Florida International University, USA) The Graph Neural Network (GNN) is showing outstanding results in improving the performance of graph-based applications. Recent studies demonstrate that GNN performance can be boosted via using more advanced aggregators, deeper aggregation depth, larger sampling rate, etc. While leading to promising results, the improvements come at a cost of significantly increased memory footprint, easily exceeding GPU memory capacity. In this paper, we introduce a method, Betty, to make GNN training more scalable and accessible via batch-level partitioning. Different from DNN training, a mini-batch in GNN has complex dependencies between input features and output labels, making batch-level partitioning difficult. Betty introduces two noveltechniques, redundancy-embedded graph (REG) partitioning and memory-aware partitioning, to effectively mitigate the redundancy and load imbalances issues across the partitions. Our evaluation of large-scale real-world datasets shows that Betty can significantly mitigate the memory bottleneck, enabling scalable GNN training with much deeper aggregation depths, larger sampling rate, larger training batch sizes, together with more advanced aggregators, with a few as a single GPU. @InProceedings{ASPLOS23p103, author = {Shuangyan Yang and Minjia Zhang and Wenqian Dong and Dong Li}, title = {Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {103--117}, doi = {10.1145/3575693.3575725}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yao, Jiabo	ASPLOS '23: "Revisiting Log-Structured ..." Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems Zhuohui Duan, Jiabo Yao, Haikun Liu, Xiaofei Liao, Hai Jin, and Yu Zhang (Huazhong University of Science and Technology, China) We present MioDB, a novel LSM-tree based key-value (KV) store system designed to fully exploit the advantages of byte-addressable non-volatile memories (NVMs). Our experimental studies reveal that the performance bottleneck of LSM-tree based KV stores using NVMs mainly stems from (1) costly data serialization/deserialization across memory and storage, and (2) unbalanced speed between memory-to-disk data flushing and on-disk data compaction. They may cause unpredictable performance degradation due to write stalls and write amplification. To address these problems, we advocate byte-addressable and persistent skip lists to replace the on-disk data structure of LSM-tree, and design four novel techniques to make the best use of fast NVMs. First, we propose one-piece flushing to minimize the cost of data serialization from DRAM to NVM. Second, we exploit an elastic NVM buffer with multiple levels and zero-copy compaction to eliminate write stalls and reduce write amplification. Third, we propose parallel compaction to orchestrate data flushing and compactions across all levels of LSM-trees. Finally, MioDB increases the depth of LSM-tree and exploits bloom filters to improve the read performance. Our extensive experimental studies demonstrate that MioDB achieves 17.1× and 21.7× lower 99.9th percentile latency, 8.3× and 2.5× higher random write throughput, and up to 5× and 4.9× lower write amplification compared with the state-of-the-art NoveLSM and MatrixKV, respectively. @InProceedings{ASPLOS23p674, author = {Zhuohui Duan and Jiabo Yao and Haikun Liu and Xiaofei Liao and Hai Jin and Yu Zhang}, title = {Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {674--687}, doi = {10.1145/3575693.3575715}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yazdanbakhsh, Amir	ASPLOS '23: "FLAT: An Optimized Dataflow ..." FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks Sheng-Chun Kao, Suvinay Subramanian, Gaurav Agrawal, Amir Yazdanbakhsh, and Tushar Krishna (Georgia Institute of Technology, USA; Google, USA; Microsoft, USA) Attention mechanisms, primarily designed to capture pairwise correlations between words, have become the backbone of machine learning, expanding beyond natural language processing into other domains. This growth in adaptation comes at the cost of prohibitively large memory requirements and computational complexity, especially at higher number of input elements. This limitation is due to inherently limited data reuse opportunities and quadratic growth in memory footprints, leading to severe memory-boundedness and limited scalability of input elements. This work addresses these challenges by devising a tailored dataflow optimization, called FLAT, for attention mechanisms without altering their functionality. This dataflow processes costly attention operations through a unique fusion mechanism, transforming the memory footprint quadratic growth to merely a linear one. To realize the full potential of this bespoke mechanism, we propose a tiling approach to enhance the data reuse across attention operations. Our method both mitigates the off-chip bandwidth bottleneck as well as reduces the on-chip memory requirement. FLAT delivers 1.94x (1.76x) speedup and 49% and (42%) of energy savings compared to the state-of-the-art Edge (Cloud) accelerators with no customized dataflow optimization. When on-chip resources are scarce (20 KB-200 KB), FLAT yields, on average, 1.5x end-to-end latency reduction across a diverse range of conventional attention-based models with input sequence lengths ranging from 512-token to 64K-token. Our evaluations demonstrate that state-of-the-art DNN dataflow applied to attention operations reach the efficiency limit for inputs above 512 elements. In contrast, FLAT unblocks transformer models for inputs with up to 64K elements. @InProceedings{ASPLOS23p295, author = {Sheng-Chun Kao and Suvinay Subramanian and Gaurav Agrawal and Amir Yazdanbakhsh and Tushar Krishna}, title = {FLAT: An Optimized Dataflow for Mitigating Attention Bottlenecks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {295--310}, doi = {10.1145/3575693.3575747}, year = {2023}, } Publisher's Version
	Ye, Chencheng	ASPLOS '23: "SpecPMT: Speculative Logging ..." SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory Chencheng Ye, Yuanchao Xu, Xipeng Shen, Yan Sha, Xiaofei Liao, Hai Jin, and Yan Solihin (Huazhong University of Science and Technology, China; North Carolina State University, USA; University of Central Florida, USA) Crash consistency overhead is a long-standing barrier to the adoption of byte-addressable persistent memory in practice. Despite continuous progress, persistent transactions for crash consistency still incur a 5.6X slowdown, making persistent memory prohibitively costly in practical settings. This paper introduces speculative logging, a new method that forgoes most memory fences and reduces data persistence overhead by logging data values early. This technique enables a novel persistent transaction model, speculatively persistent memory transactions (SpecPMT). Our evaluation shows that SpecPMT reduces the execution time overheads of persistent transactions substantially to just 10%. @InProceedings{ASPLOS23p762, author = {Chencheng Ye and Yuanchao Xu and Xipeng Shen and Yan Sha and Xiaofei Liao and Hai Jin and Yan Solihin}, title = {SpecPMT: Speculative Logging for Resolving Crash Consistency Overhead of Persistent Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {762--777}, doi = {10.1145/3575693.3575696}, year = {2023}, } Publisher's Version
	Ye, Zihao	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Yim, Jinkyu	ASPLOS '23: "Optimus-CC: Efficient Large ..." Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression Jaeyong Song, Jinkyu Yim, Jaewon Jung, Hongsun Jang, Hyung-Jin Kim, Youngsok Kim, and Jinho Lee (Yonsei University, South Korea; Seoul National University, South Korea; Samsung Electronics, South Korea) In training of modern large natural language processing (NLP) models, it has become a common practice to split models using 3D parallelism to multiple GPUs. Such technique, however, suffers from a high overhead of inter-node communication. Compressing the communication is one way to mitigate the overhead by reducing the inter-node traffic volume; however, the existing compression techniques have critical limitations to be applied for NLP models with 3D parallelism in that 1) only the data parallelism traffic is targeted, and 2) the existing compression schemes already harm the model quality too much. In this paper, we present Optimus-CC, a fast and scalable distributed training framework for large NLP models with aggressive communication compression. Optimus-CC differs from existing communication compression frameworks in the following ways: First, we compress pipeline parallel (inter-stage) traffic. In specific, we compress the inter-stage backpropagation and the embedding synchronization in addition to the existing data-parallel traffic compression methods. Second, we propose techniques to avoid the model quality drop that comes from the compression. We further provide mathematical and empirical analyses to show that our techniques can successfully suppress the compression error. Lastly, we analyze the pipeline and opt to selectively compress those traffic lying on the critical path. This further helps reduce the compression error. We demonstrate our solution on a GPU cluster, and achieve superior speedup from the baseline state-of-the-art solutions for distributed training without sacrificing the model quality. @InProceedings{ASPLOS23p560, author = {Jaeyong Song and Jinkyu Yim and Jaewon Jung and Hongsun Jang and Hyung-Jin Kim and Youngsok Kim and Jinho Lee}, title = {Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {560--573}, doi = {10.1145/3575693.3575712}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available
	You, Xin	ASPLOS '23: "VClinic: A Portable and Efficient ..." VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers Xin You, Hailong Yang, Kelun Lei, Zhongzhi Luan, and Depei Qian (Beihang University, China) Fine-grained value profilers reveal a promising way to accurately detect value-related software inefficiencies with binary instrumentation. Due to the architecture-dependent implementation details of binary instrumentation, existing value profilers suffer from poor portability as well as high engineering efforts to achieve efficiency across platforms. In this paper, we propose VClinic, a portable and efficient fine-grained value profiling framework for analyzing highly optimized binaries on both X86 and ARM platforms. VClinic exploits operand-centric two-level designs in its implementation to provide the common building blocks required for value profilers. By constructing four representative value profilers with VClinic, we demonstrate that VClinic can ease the development of value profilers with portability and efficiency across platforms. Guided by the value profilers built upon VClinic, we can achieve up to 89.94% and 74.66% speedup for real-world programs on X86 and ARM platforms, respectively. @InProceedings{ASPLOS23p892, author = {Xin You and Hailong Yang and Kelun Lei and Zhongzhi Luan and Depei Qian}, title = {VClinic: A Portable and Efficient Framework for Fine-Grained Value Profilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {892--904}, doi = {10.1145/3575693.3576934}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yu, Cody Hao	ASPLOS '23: "Hidet: Task-Mapping Programming ..." Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding, Cody Hao Yu, Bojian Zheng, Yizhi Liu, Yida Wang, and Gennady Pekhimenko (University of Toronto, Canada; Vector Institute, Canada; Amazon Web Services, USA) As deep learning models nowadays are widely adopted by both cloud services and edge devices, reducing the latency of deep learning model inferences becomes crucial to provide efficient model serving. However, it is challenging to develop efficient tensor programs for deep learning operators due to the high complexity of modern accelerators (e.g., NVIDIA GPUs and Google TPUs) and the rapidly growing number of operators. Deep learning compilers, such as Apache TVM, adopt declarative scheduling primitives to lower the bar of developing tensor programs. However, we show that this approach is insufficient to cover state-of-the-art tensor program optimizations (e.g., double buffering). In this paper, we propose to embed the scheduling process into tensor programs and use dedicated mappings, called task mappings, to define the computation assignment and ordering directly in the tensor programs. This new approach greatly enriches the expressible optimizations by allowing developers to manipulate tensor programs at a much finer granularity (e.g., allowing program-statement-level optimizations). We call the proposed method the task-mapping programming paradigm. In addition, we propose a new post-scheduling fusion optimization that allows developers to focus on scheduling every single operator and automates the fusion after scheduling. It greatly reduces the engineering efforts for operator fusion. Our proposed paradigm also constructs an efficient hardware-centric schedule space, which is agnostic to the program input size and greatly reduces the tuning time. With the proposed paradigm, we implement a deep learning compiler Hidet. Extensive experiments on modern convolution and transformer models show that Hidet outperforms state-of-the-art DNN inference framework, ONNX Runtime, and compiler, TVM equipped with scheduler AutoTVM and Ansor, by up to 1.48x (1.22x on average). It also reduces the tuning time by 20x and 11x compared with AutoTVM and Ansor, respectively. We open-sourced hidet at https://www.github.com/hidet-org/hidet. @InProceedings{ASPLOS23p370, author = {Yaoyao Ding and Cody Hao Yu and Bojian Zheng and Yizhi Liu and Yida Wang and Gennady Pekhimenko}, title = {Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {370--384}, doi = {10.1145/3575693.3575702}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Yu, Hangchen	ASPLOS '23: "Towards a Machine Learning-Assisted ..." Towards a Machine Learning-Assisted Kernel with LAKE Henrique Fingler, Isha Tarte, Hangchen Yu, Ariel Szekely, Bodun Hu, Aditya Akella, and Christopher J. Rossbach (University of Texas at Austin, USA; Meta, USA; Massachusetts Institute of Technology, USA; Katana Graph, USA) The complexity of modern operating systems (OSes), rapid diversification of hardware, and steady evolution of machine learning (ML) motivate us to explore the potential of ML to improve decision-making in OS kernels. We conjecture that ML can better manage tradeoff spaces for subsystems such as memory management and process and I/O scheduling that currently rely on hand-tuned heuristics to provide reasonable average-case performance. We explore the replacement of heuristics with ML-driven decision-making in five kernel subsystems, consider the implications for kernel design, shared OS-level components, and access to hardware acceleration. We identify obstacles, address challenges and characterize tradeoffs for the benefits ML can provide that arise in kernel-space. We find that use of specialized hardware such as GPUs is critical to absorbing the additional computational load required by ML decisioning, but that poor accessibility of accelerators in kernel space is a barrier to adoption. We also find that the benefits of ML and acceleration for OSes is subsystem-, workload- and hardware-dependent, suggesting that using ML in kernels will require frameworks to help kernel developers navigate new tradeoff spaces. We address these challenge by building a system called LAKE for supporting ML and exposing accelerators in kernel space. LAKE includes APIs for feature collection and management across abstraction layers and module boundaries. LAKE provides mechanisms for managing the variable profitability of acceleration, and interfaces for mitigating contention for resources between user and kernel space. We show that an ML-backed I/O latency predictor can have its inference time reduced by up to 96% with acceleration. @InProceedings{ASPLOS23p846, author = {Henrique Fingler and Isha Tarte and Hangchen Yu and Ariel Szekely and Bodun Hu and Aditya Akella and Christopher J. Rossbach}, title = {Towards a Machine Learning-Assisted Kernel with LAKE}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {846--861}, doi = {10.1145/3575693.3575697}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Yu, Yong	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Yuan, Ziqi	ASPLOS '23: "VDom: Fast and Unlimited Virtual ..." VDom: Fast and Unlimited Virtual Domains on Multiple Architectures Ziqi Yuan, Siyu Hong, Rui Chang, Yajin Zhou, Wenbo Shen, and Kui Ren (Zhejiang University, China) Hardware memory domain primitives, such as Intel MPK and ARM Memory Domain, have been used for efficient in-process memory isolation. However, they can only provide a limited number of memory domains (16 domains), which cannot satisfy the compelling need for more isolated domains inside the address space of a process. Existing solutions to virtualize memory domains are either intrusive (need the modification to existing hardware), or incur a large performance overhead. In this paper, we propose VDom, a fast and scalable memory domain virtualization system that supports unlimited memory domains. VDom leverages separate address spaces to provide an unlimited number of virtual domains, and optimizes related memory management operations. To map virtual domains to hardware domains, we design a domain virtualization algorithm, which manages address spaces and domain maps for threads to efficiently access other domains that are unmapped in the current address space. According to our evaluation on real Intel and ARM platforms, on real-world server applications (httpd and MySQL), VDom incurs less than 2.65% performance overhead, which is lower than the overheads of the state-of-the-art software approaches (libmpk and EPK). In random domain access tests, VDom is comparable to EPK and has significantly higher efficiency than libmpk. @InProceedings{ASPLOS23p905, author = {Ziqi Yuan and Siyu Hong and Rui Chang and Yajin Zhou and Wenbo Shen and Kui Ren}, title = {VDom: Fast and Unlimited Virtual Domains on Multiple Architectures}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {905--919}, doi = {10.1145/3575693.3575735}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zablotchi, Igor	ASPLOS '23: "uBFT: Microsecond-Scale BFT ..." uBFT: Microsecond-Scale BFT using Disaggregated Memory Marcos K. Aguilera, Naama Ben-David, Rachid Guerraoui, Antoine Murat, Athanasios Xygkis, and Igor Zablotchi (VMware Research, USA; EPFL, Switzerland; Massachusetts Institute of Technology, USA) We propose uBFT, the first State Machine Replication (SMR) system to achieve microsecond-scale latency in data centers, while using only 2f+1 replicas to tolerate f Byzantine failures. The Byzantine Fault Tolerance (BFT) provided by uBFT is essential as pure crashes appear to be a mere illusion with real-life systems reportedly failing in many unexpected ways. uBFT relies on a small non-tailored trusted computing base—disaggregated memory—and consumes a practically bounded amount of memory. uBFT is based on a novel abstraction called Consistent Tail Broadcast, which we use to prevent equivocation while bounding memory. We implement uBFT using RDMA-based disaggregated memory and obtain an end-to-end latency of as little as 10 us. This is at least 50× faster than MinBFT, a state-of-the-art 2f+1 BFT SMR based on Intel’s SGX. We use uBFT to replicate two KV-stores (Memcached and Redis), as well as a financial order matching engine (Liquibook). These applications have low latency (up to 20 us) and become Byzantine tolerant with as little as 10 us more. The price for uBFT is a small amount of reliable disaggregated memory (less than 1 MiB), which in our prototype consists of a small number of memory servers connected through RDMA and replicated for fault tolerance. @InProceedings{ASPLOS23p862, author = {Marcos K. Aguilera and Naama Ben-David and Rachid Guerraoui and Antoine Murat and Athanasios Xygkis and Igor Zablotchi}, title = {uBFT: Microsecond-Scale BFT using Disaggregated Memory}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {862--877}, doi = {10.1145/3575693.3575732}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zang, Binyu	ASPLOS '23: "BeeHive: Sub-second Elasticity ..." BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution Ziming Zhao, Mingyu Wu, Jiawei Tang, Binyu Zang, Zhaoguo Wang, and Haibo Chen (Shanghai Jiao Tong University, China; Shanghai AI Laboratory, China; Engineering Research Center for Domain-specific Operating Systems of the Ministry of Education of China, China) Function-as-a-service (FaaS), an emerging cloud computing paradigm, is expected to provide strong elasticity due to its promise to auto-scale fine-grained functions rapidly. Although appealing for applications with good parallelism and dynamic workload, this paper shows that it is non-trivial to adapt existing monolithic applications (like web services) to FaaS due to their complexity. To bridge the gap between complicated web services and FaaS, this paper proposes a runtime-based Semi-FaaS execution model, which dynamically extracts time-consuming code snippets (closures) from applications and offloads them to FaaS platforms for execution. It further proposes BeeHive, an offloading framework for Semi-FaaS, which relies on the managed runtime to provide a fallback-based execution model and addresses the performance issues in traditional offloading mechanisms for FaaS. Meanwhile, the runtime system of BeeHive selects offloading candidates in a user-transparent way and supports efficient object sharing, memory management, and failure recovery in a distributed environment. The evaluation using various web applications suggests that the Semi-FaaS execution supported by BeeHive can reach sub-second resource provisioning on commercialized FaaS platforms like AWS Lambda, which is up to two orders of magnitude better than other alternative scaling approaches in cloud computing. @InProceedings{ASPLOS23p74, author = {Ziming Zhao and Mingyu Wu and Jiawei Tang and Binyu Zang and Zhaoguo Wang and Haibo Chen}, title = {BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {74--87}, doi = {10.1145/3575693.3575752}, year = {2023}, } Publisher's Version
	Zardoshti, Pantea	ASPLOS '23: "Pond: CXL-Based Memory Pooling ..." Pond: CXL-Based Memory Pooling Systems for Cloud Platforms Huaicheng Li, Daniel S. Berger, Lisa Hsu, Daniel Ernst, Pantea Zardoshti, Stanko Novakovic, Monish Shah, Samir Rajadnya, Scott Lee, Ishwar Agarwal, Mark D. Hill, Marcus Fontoura, and Ricardo Bianchini (Virginia Tech, USA; Carnegie Mellon University, USA; Microsoft Azure, USA; University of Washington, USA; Unaffiliated, USA; Google, USA; Microsoft, USA; Intel, USA; University of Wisconsin-Madison, USA; Stone, USA) Public cloud providers seek to meet stringent performance requirements and low hardware cost. A key driver of performance and cost is main memory. Memory pooling promises to improve DRAM utilization and thereby reduce costs. However, pooling is challenging under cloud performance requirements. This paper proposes Pond, the first memory pooling system that both meets cloud performance goals and significantly reduces DRAM cost. Pond builds on the Compute Express Link (CXL) standard for load/store access to pool memory and two key insights. First, our analysis of cloud production traces shows that pooling across 8-16 sockets is enough to achieve most of the benefits. This enables a small-pool design with low access latency. Second, it is possible to create machine learning models that can accurately predict how much local and pool memory to allocate to a virtual machine (VM) to resemble same-NUMA-node memory performance. Our evaluation with 158 workloads shows that Pond reduces DRAM costs by 7% with performance within 1-5% of same-NUMA-node VM allocations. @InProceedings{ASPLOS23p574, author = {Huaicheng Li and Daniel S. Berger and Lisa Hsu and Daniel Ernst and Pantea Zardoshti and Stanko Novakovic and Monish Shah and Samir Rajadnya and Scott Lee and Ishwar Agarwal and Mark D. Hill and Marcus Fontoura and Ricardo Bianchini}, title = {Pond: CXL-Based Memory Pooling Systems for Cloud Platforms}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {574--587}, doi = {10.1145/3575693.3578835}, year = {2023}, } Publisher's Version
	Zhai, Yi	ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Zhang, Baoheng	ASPLOS '23: "DPACS: Hardware Accelerated ..." DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design Yizhao Gao, Baoheng Zhang, Xiaojuan Qi, and Hayden Kwok-Hay So (University of Hong Kong, Hong Kong) By eliminating compute operations intelligently based on the run time input, dynamic pruning (DP) promises to improve deep neural network inference speed substantially without incurring a major impact on their accuracy. Although many DP algorithms with good pruning performance have been proposed, it remains a challenge to translate these theoretical reductions in compute operations into satisfactory end-to-end speedups in practical real-world implementations. The overhead of identifying operations to be pruned during run time, the need to efficiently process the resulting dynamic dataflow, and the non-trivial memory I/O bottleneck that emerges as the number of compute operations reduces, have all contributed to the challenge of implementing practical DP systems. In this paper, the design and implementation of DPACS are presented to address these challenges. DPACS utilizes a hardware-aware dynamic spatial and channel pruning algorithm in conjunction with a dynamic dataflow engine in hardware to facilitate efficient processing of the pruned network. A channel mask precomputation scheme is designed to reduce memory I/O, and a dedicated inter-layer pipeline is used to achieve efficient indexing and dataflow of sparse activation. Extensive design space exploration has been performed using two architectural variations implemented on FPGA to accelerate multiple networks from the ResNet family on the ImageNet and CIFAR10 dataset across a wide range of pruning ratios. Across the spectrum of configurations, DPACS is able to achieve 1.1× to 3.9× end-to-end speedup over a baseline hardware implementation without pruning. Analysis of the tradeoff among accuracy, compute, and memory I/O performance highlights the importance of algorithm-architecture codesign in developing DP systems. @InProceedings{ASPLOS23p237, author = {Yizhao Gao and Baoheng Zhang and Xiaojuan Qi and Hayden Kwok-Hay So}, title = {DPACS: Hardware Accelerated Dynamic Neural Network Pruning through Algorithm-Architecture Co-design}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {237--251}, doi = {10.1145/3575693.3575728}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhang, Guangda	ASPLOS '23: "NUBA: Non-Uniform Bandwidth ..." NUBA: Non-Uniform Bandwidth GPUs Xia Zhao, Magnus Jahre, Yuhua Tang, Guangda Zhang, and Lieven Eeckhout (Academy of Military Sciences, China; NTNU, Norway; National University of Defense Technology, China; Ghent University, Belgium) The parallel execution model of GPUs enables scaling to hundreds of thousands of threads, which is a key capability that many modern high-performance applications exploit. GPU vendors are hence increasing the compute and memory resources with every GPU generation — resulting in the need to efficiently stitch together a plethora of Symmetric Multiprocessors (SMs), Last-Level Cache (LLC) slices and memory controllers while maximizing bandwidth and keeping power consumption and design complexity in check. Conventional GPUs are Uniform Bandwidth Architectures (UBAs) as they provide equal bandwidth between all SMs and all LLC slices. UBA GPUs require a uniform high-bandwidth Network-on-Chip (NoC), and our key observation is that provisioning a NoC to match the LLC slice bandwidth incurs a hefty power and complexity overhead. We propose the Non-Uniform Bandwidth Architecture (NUBA), a GPU system architecture aimed at fully utilizing LLC slice bandwidth. A NUBA GPU consists of partitions that each feature a few SMs and LLC slices as well as a memory controller — hence exposing the complete LLC bandwidth to the SMs within a partition since they can be connected with point-to-point links — and a NoC between partitions — to enable access to remote data.Exploiting the potential of NUBA GPUs however requires carefully co-designing system software, the compiler and architectural policies. The critical system software component is our Local-And-Balanced (LAB) page placement policy which enables the GPU driver to place data in local partitions while avoiding load imbalance. Moreover, we propose Model-Driven Replication (MDR) which identifies read-only shared data with data-flow analysis at compile time. At run time, MDR leverages an architectural mechanism that replicates read-only shared data across LLC slices when this can be done without pressuring cache capacity. With LAB and MDR, our NUBA GPU improves average performance by 23.1% and 22.2% (and up to 183.9% and 182.4%) compared to iso-resource memory-side and SM-side UBA GPUs, respectively. When the NUBA concept is leveraged to reduce overhead while maintaining similar performance, NUBA reduces NoC power consumption by 12.1× and 9.4×, respectively. @InProceedings{ASPLOS23p544, author = {Xia Zhao and Magnus Jahre and Yuhua Tang and Guangda Zhang and Lieven Eeckhout}, title = {NUBA: Non-Uniform Bandwidth GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {544--559}, doi = {10.1145/3575693.3575745}, year = {2023}, } Publisher's Version
	Zhang, Lingming	ASPLOS '23: "NNSmith: Generating Diverse ..." NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers Jiawei Liu, Jinkun Lin, Fabian Ruffy, Cheng Tan, Jinyang Li, Aurojit Panda, and Lingming Zhang (University of Illinois at Urbana-Champaign, USA; New York University, USA; Northeastern University, USA) Deep-learning (DL) compilers such as TVM and TensorRT are increasingly being used to optimize deep neural network (DNN) models to meet performance, resource utilization and other requirements. Bugs in these compilers can result in models whose semantics differ from the original ones, producing incorrect results that corrupt the correctness of downstream applications. However, finding bugs in these compilers is challenging due to their complexity. In this work, we propose a new fuzz testing approach for finding bugs in deep-learning compilers. Our core approach consists of (i) generating diverse yet valid DNN test models that can exercise a large part of the compiler's transformation logic using light-weight operator specifications; (ii) performing gradient-based search to find model inputs that avoid any floating-point exceptional values during model execution, reducing the chance of missed bugs or false alarms; and (iii) using differential testing to identify bugs. We implemented this approach in NNSmith which has found 72 new bugs for TVM, TensorRT, ONNXRuntime, and PyTorch to date. Of these 58 have been confirmed and 51 have been fixed by their respective project maintainers. @InProceedings{ASPLOS23p530, author = {Jiawei Liu and Jinkun Lin and Fabian Ruffy and Cheng Tan and Jinyang Li and Aurojit Panda and Lingming Zhang}, title = {NNSmith: Generating Diverse and Valid Test Cases for Deep Learning Compilers}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {530--543}, doi = {10.1145/3575693.3575707}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhang, Meng	ASPLOS '23: "Lucid: A Non-intrusive, Scalable ..." Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs Qinghao Hu, Meng Zhang, Peng Sun, Yonggang Wen, and Tianwei Zhang (Nanyang Technological University, Singapore; Shanghai AI Laboratory, China; SenseTime, China) While recent deep learning workload schedulers exhibit excellent performance, it is arduous to deploy them in practice due to some substantial defects, including inflexible intrusive manner, exorbitant integration and maintenance cost, limited scalability, as well as opaque decision processes. Motivated by these issues, we design and implement Lucid, a non-intrusive deep learning workload scheduler based on interpretable models. It consists of three innovative modules. First, a two-dimensional optimized profiler is introduced for efficient job metric collection and timely debugging job feedback. Second, Lucid utilizes an indolent packing strategy to circumvent interference. Third, Lucid orchestrates resources based on estimated job priority values and sharing scores to achieve efficient scheduling. Additionally, Lucid promotes model performance maintenance and system transparent adjustment via a well-designed system optimizer. Our evaluation shows that Lucid reduces the average job completion time by up to 1.3× compared with state-of-the-art preemptive scheduler Tiresias. Furthermore, it provides explicit system interpretations and excellent scalability for practical deployment. @InProceedings{ASPLOS23p457, author = {Qinghao Hu and Meng Zhang and Peng Sun and Yonggang Wen and Tianwei Zhang}, title = {Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {457--472}, doi = {10.1145/3575693.3575705}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhang, Mingxing	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version
	Zhang, Minjia	ASPLOS '23: "Betty: Enabling Large-Scale ..." Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning Shuangyan Yang, Minjia Zhang, Wenqian Dong, and Dong Li (University of California at Merced, USA; Microsoft Research, USA; Florida International University, USA) The Graph Neural Network (GNN) is showing outstanding results in improving the performance of graph-based applications. Recent studies demonstrate that GNN performance can be boosted via using more advanced aggregators, deeper aggregation depth, larger sampling rate, etc. While leading to promising results, the improvements come at a cost of significantly increased memory footprint, easily exceeding GPU memory capacity. In this paper, we introduce a method, Betty, to make GNN training more scalable and accessible via batch-level partitioning. Different from DNN training, a mini-batch in GNN has complex dependencies between input features and output labels, making batch-level partitioning difficult. Betty introduces two noveltechniques, redundancy-embedded graph (REG) partitioning and memory-aware partitioning, to effectively mitigate the redundancy and load imbalances issues across the partitions. Our evaluation of large-scale real-world datasets shows that Betty can significantly mitigate the memory bottleneck, enabling scalable GNN training with much deeper aggregation depths, larger sampling rate, larger training batch sizes, together with more advanced aggregators, with a few as a single GPU. @InProceedings{ASPLOS23p103, author = {Shuangyan Yang and Minjia Zhang and Wenqian Dong and Dong Li}, title = {Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {103--117}, doi = {10.1145/3575693.3575725}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhang, Tianwei	ASPLOS '23: "Lucid: A Non-intrusive, Scalable ..." Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs Qinghao Hu, Meng Zhang, Peng Sun, Yonggang Wen, and Tianwei Zhang (Nanyang Technological University, Singapore; Shanghai AI Laboratory, China; SenseTime, China) While recent deep learning workload schedulers exhibit excellent performance, it is arduous to deploy them in practice due to some substantial defects, including inflexible intrusive manner, exorbitant integration and maintenance cost, limited scalability, as well as opaque decision processes. Motivated by these issues, we design and implement Lucid, a non-intrusive deep learning workload scheduler based on interpretable models. It consists of three innovative modules. First, a two-dimensional optimized profiler is introduced for efficient job metric collection and timely debugging job feedback. Second, Lucid utilizes an indolent packing strategy to circumvent interference. Third, Lucid orchestrates resources based on estimated job priority values and sharing scores to achieve efficient scheduling. Additionally, Lucid promotes model performance maintenance and system transparent adjustment via a well-designed system optimizer. Our evaluation shows that Lucid reduces the average job completion time by up to 1.3× compared with state-of-the-art preemptive scheduler Tiresias. Furthermore, it provides explicit system interpretations and excellent scalability for practical deployment. @InProceedings{ASPLOS23p457, author = {Qinghao Hu and Meng Zhang and Peng Sun and Yonggang Wen and Tianwei Zhang}, title = {Lucid: A Non-intrusive, Scalable and Interpretable Scheduler for Deep Learning Training Jobs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {457--472}, doi = {10.1145/3575693.3575705}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhang, Yanyong	ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Zhang, Ye	ASPLOS '23: "GZKP: A GPU Accelerated Zero-Knowledge ..." GZKP: A GPU Accelerated Zero-Knowledge Proof System Weiliang Ma, Qian Xiong, Xuanhua Shi, Xiaosong Ma, Hai Jin, Haozhao Kuang, Mingyu Gao, Ye Zhang, Haichen Shen, and Weifang Hu (Huazhong University of Science and Technology, China; Hamad Bin Khalifa University, Qatar; Tsinghua University, China; Scroll Foundation, Seychelles) Zero-knowledge proof (ZKP) is a cryptographic protocol that allows one party to prove the correctness of a statement to another party without revealing any information beyond the correctness of the statement itself. It guarantees computation integrity and confidentiality, and is therefore increasingly adopted in industry for a variety of privacy-preserving applications, such as verifiable outsource computing and digital currency. A significant obstacle in using ZKP for online applications is the performance overhead of its proof generation. We develop GZKP, a GPU accelerated zero-knowledge proof system that supports different levels of security requirements and brings significant speedup toward making ZKP truly usable. For polynomial computation over a large finite field, GZKP promotes a cache-friendly memory access pattern while eliminating the costly external shuffle in existing solutions. For multi-scalar multiplication, GZKP adopts a new parallelization strategy, which aggressively combines integer elliptic curve point operations and exploits fine-grained task parallelism with load balancing for sparse integer distribution. GZKP outperforms the state-of-the-art ZKP systems by an order of magnitude, achieving up to 48.1× and 17.6× speedup with standard cryptographic benchmarks and a real-world application workload, respectively. @InProceedings{ASPLOS23p340, author = {Weiliang Ma and Qian Xiong and Xuanhua Shi and Xiaosong Ma and Hai Jin and Haozhao Kuang and Mingyu Gao and Ye Zhang and Haichen Shen and Weifang Hu}, title = {GZKP: A GPU Accelerated Zero-Knowledge Proof System}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {340--353}, doi = {10.1145/3575693.3575711}, year = {2023}, } Publisher's Version
	Zhang, Yu	ASPLOS '23: "Revisiting Log-Structured ..." Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems Zhuohui Duan, Jiabo Yao, Haikun Liu, Xiaofei Liao, Hai Jin, and Yu Zhang (Huazhong University of Science and Technology, China) We present MioDB, a novel LSM-tree based key-value (KV) store system designed to fully exploit the advantages of byte-addressable non-volatile memories (NVMs). Our experimental studies reveal that the performance bottleneck of LSM-tree based KV stores using NVMs mainly stems from (1) costly data serialization/deserialization across memory and storage, and (2) unbalanced speed between memory-to-disk data flushing and on-disk data compaction. They may cause unpredictable performance degradation due to write stalls and write amplification. To address these problems, we advocate byte-addressable and persistent skip lists to replace the on-disk data structure of LSM-tree, and design four novel techniques to make the best use of fast NVMs. First, we propose one-piece flushing to minimize the cost of data serialization from DRAM to NVM. Second, we exploit an elastic NVM buffer with multiple levels and zero-copy compaction to eliminate write stalls and reduce write amplification. Third, we propose parallel compaction to orchestrate data flushing and compactions across all levels of LSM-trees. Finally, MioDB increases the depth of LSM-tree and exploits bloom filters to improve the read performance. Our extensive experimental studies demonstrate that MioDB achieves 17.1× and 21.7× lower 99.9th percentile latency, 8.3× and 2.5× higher random write throughput, and up to 5× and 4.9× lower write amplification compared with the state-of-the-art NoveLSM and MatrixKV, respectively. @InProceedings{ASPLOS23p674, author = {Zhuohui Duan and Jiabo Yao and Haikun Liu and Xiaofei Liao and Hai Jin and Yu Zhang}, title = {Revisiting Log-Structured Merging for KV Stores in Hybrid Memory Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {674--687}, doi = {10.1145/3575693.3575715}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced ASPLOS '23: "TLP: A Deep Learning-Based ..." TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang (University of Science and Technology of China, China) Tensor program tuning is a non-convex objective optimization problem, to which search-based approaches have proven to be effective. At the core of the search-based approaches lies the design of the cost model. Though deep learning-based cost models perform significantly better than other methods, they still fall short and suffer from the following problems. First, their feature extraction heavily relies on expert-level domain knowledge in hardware architectures. Even so, the extracted features are often unsatisfactory and require separate considerations for CPUs and GPUs. Second, a cost model trained on one hardware platform usually performs poorly on another, a problem we call cross-hardware unavailability. In order to address these problems, we propose TLP and MTL-TLP. TLP is a deep learning-based cost model that facilitates tensor program tuning. Instead of extracting features from the tensor program itself, TLP extracts features from the schedule primitives. We treat schedule primitives as tensor languages. TLP is thus a Tensor Language Processing task. In this way, the task of predicting the tensor program latency through the cost model is transformed into a natural language processing (NLP) regression task. MTL-TLP combines Multi-Task Learning and TLP to cope with the cross-hardware unavailability problem. We incorporate these techniques into the Ansor framework and conduct detailed experiments. Results show that TLP can speed up the average search time by 9.1× and 3.0× on CPU and GPU workloads, respectively, compared to the state-of-the-art implementation. MTL-TLP can achieve a speed-up of 4.7× and 2.9× on CPU and GPU workloads, respectively, using only 7% of the target hardware data. To the best of our knowledge, TLP is the first tensor program cost model to extract features directly from schedule primitives, and MTL-TLP is the first open-sourced work that effectively addresses the cross-platform unavailability problem. The code is available at https://github.com/zhaiyi000/tlp. @InProceedings{ASPLOS23p833, author = {Yi Zhai and Yu Zhang and Shuo Liu and Xiaomeng Chu and Jie Peng and Jianmin Ji and Yanyong Zhang}, title = {TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {833--845}, doi = {10.1145/3575693.3575737}, year = {2023}, } Publisher's Version
	Zhang, Zhizhou	ASPLOS '23: "A Prediction System Service ..." A Prediction System Service Zhizhou Zhang, Alvin Oliver Glova, Timothy Sherwood, and Jonathan Balkind (University of California at Santa Barbara, USA) To better facilitate application performance programming we propose a software optimization strategy enabled by a novel low-latency Prediction System Service (PSS). Rather than relying on nuanced domain-specific knowledge or slapdash heuristics, a system service for prediction encourages programmers to spend their time uncovering new levers for optimization rather than worrying about the details of their control. The core idea is to write optimizations that improve performance in specific cases, or under specific tunings, and leave the decision of how and when exactly to apply those optimizations to the system to learn through feedback-directed learning. Such a prediction service can be implemented in any number of ways, including as a shared library that can be easily reused by software written in different programming languages, and opens the door to both new software optimization patterns and hardware design possibilities. As a demonstration of the utility of this approach, we show that three very different application-targeted optimization scenarios can each benefit from even a very straightforward perceptron-based implementation of the PSS as long as the service latency can be held low. First, we show that PSS can be used to more intelligently guide hardware lock elision with resulting speedups over a baseline implementation by 34% on average. Second, we show that a PSS can find good configuration parameters for PyPy’s Just-In-Time (JIT) compiler resulting in 15% speedup on average. Last, we show PSS can guide the page reclamation task within a kernel memory management subsystem to reduce the average memory latency by 33% on average. In all three cases, this new optimization pattern with service support is able to meet or beat the best-known hand-crafted methods with a fraction of the complexity. @InProceedings{ASPLOS23p48, author = {Zhizhou Zhang and Alvin Oliver Glova and Timothy Sherwood and Jonathan Balkind}, title = {A Prediction System Service}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {48--60}, doi = {10.1145/3575693.3575714}, year = {2023}, } Publisher's Version
	Zhao, Tian	ASPLOS '23: "Sigma: Compiling Einstein ..." Sigma: Compiling Einstein Summations to Locality-Aware Dataflow Tian Zhao, Alexander Rucker, and Kunle Olukotun (Stanford University, USA) Most dataflow accelerator compilers achieve high performance by mapping each node in a dataflow program to a dedicated hardware element on a dataflow accelerator. However, this approach misses critical data reuse optimizations required to exploit the data bandwidth from fine-grained memory elements, e.g., FIFOs and pipeline registers. Moreover, writing performant dataflow programs requires users to have domain expertise in the underlying dataflow accelerators. To address these issues, we designed Sigma, a novel compiler that supports high-level programming constructs such as Einstein summations, index notations, and tensors, finds opportunities for data reuse from high-level dataflow graphs, and exploits on-chip data bandwidth from fine-grained memory elements. Sigma targeting a research dataflow accelerator demonstrates a 5.4x speedup and 44.6x area-normalized speedup over Nvidia's V100 accelerator, and a 7.1x speedup over hand-written dataflow programs. @InProceedings{ASPLOS23p718, author = {Tian Zhao and Alexander Rucker and Kunle Olukotun}, title = {Sigma: Compiling Einstein Summations to Locality-Aware Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {718--732}, doi = {10.1145/3575693.3575694}, year = {2023}, } Publisher's Version
	Zhao, Wei	ASPLOS '23: "TiLT: A Time-Centric Approach ..." TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization Anand Jayarajan, Wei Zhao, Yudi Sun, and Gennady Pekhimenko (University of Toronto, Canada) Stream processing engines (SPEs) are widely used for large scale streaming analytics over unbounded time-ordered data streams. Modern day streaming analytics applications exhibit diverse compute characteristics and demand strict latency and throughput requirements. Over the years, there has been significant attention in building hardware-efficient stream processing engines (SPEs) that support several query optimization, parallelization, and execution strategies to meet the performance requirements of large scale streaming analytics applications. However, in this work, we observe that these strategies often fail to generalize well on many real-world streaming analytics applications due to several inherent design limitations of current SPEs. We further argue that these limitations stem from the shortcomings of the fundamental design choices and the query representation model followed in modern SPEs. To address these challenges, we first propose TiLT, a novel intermediate representation (IR) that offers a highly expressive temporal query language amenable to effective query optimization and parallelization strategies. We subsequently build a compiler backend for TiLT that applies such optimizations on streaming queries and generates hardware-efficient code to achieve high performance on multi-core stream query executions. We demonstrate that TiLT achieves up to 326× (20.49× on average) higher throughput compared to state-of-the-art SPEs (e.g., Trill) across eight real-world streaming analytics applications. TiLT source code is available at https://github.com/ampersand-projects/tilt.git. @InProceedings{ASPLOS23p818, author = {Anand Jayarajan and Wei Zhao and Yudi Sun and Gennady Pekhimenko}, title = {TiLT: A Time-Centric Approach for Stream Query Optimization and Parallelization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {818--832}, doi = {10.1145/3575693.3575704}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhao, Xia	ASPLOS '23: "NUBA: Non-Uniform Bandwidth ..." NUBA: Non-Uniform Bandwidth GPUs Xia Zhao, Magnus Jahre, Yuhua Tang, Guangda Zhang, and Lieven Eeckhout (Academy of Military Sciences, China; NTNU, Norway; National University of Defense Technology, China; Ghent University, Belgium) The parallel execution model of GPUs enables scaling to hundreds of thousands of threads, which is a key capability that many modern high-performance applications exploit. GPU vendors are hence increasing the compute and memory resources with every GPU generation — resulting in the need to efficiently stitch together a plethora of Symmetric Multiprocessors (SMs), Last-Level Cache (LLC) slices and memory controllers while maximizing bandwidth and keeping power consumption and design complexity in check. Conventional GPUs are Uniform Bandwidth Architectures (UBAs) as they provide equal bandwidth between all SMs and all LLC slices. UBA GPUs require a uniform high-bandwidth Network-on-Chip (NoC), and our key observation is that provisioning a NoC to match the LLC slice bandwidth incurs a hefty power and complexity overhead. We propose the Non-Uniform Bandwidth Architecture (NUBA), a GPU system architecture aimed at fully utilizing LLC slice bandwidth. A NUBA GPU consists of partitions that each feature a few SMs and LLC slices as well as a memory controller — hence exposing the complete LLC bandwidth to the SMs within a partition since they can be connected with point-to-point links — and a NoC between partitions — to enable access to remote data.Exploiting the potential of NUBA GPUs however requires carefully co-designing system software, the compiler and architectural policies. The critical system software component is our Local-And-Balanced (LAB) page placement policy which enables the GPU driver to place data in local partitions while avoiding load imbalance. Moreover, we propose Model-Driven Replication (MDR) which identifies read-only shared data with data-flow analysis at compile time. At run time, MDR leverages an architectural mechanism that replicates read-only shared data across LLC slices when this can be done without pressuring cache capacity. With LAB and MDR, our NUBA GPU improves average performance by 23.1% and 22.2% (and up to 183.9% and 182.4%) compared to iso-resource memory-side and SM-side UBA GPUs, respectively. When the NUBA concept is leveraged to reduce overhead while maintaining similar performance, NUBA reduces NoC power consumption by 12.1× and 9.4×, respectively. @InProceedings{ASPLOS23p544, author = {Xia Zhao and Magnus Jahre and Yuhua Tang and Guangda Zhang and Lieven Eeckhout}, title = {NUBA: Non-Uniform Bandwidth GPUs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {544--559}, doi = {10.1145/3575693.3575745}, year = {2023}, } Publisher's Version
	Zhao, Yihao	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhao, Ziming	ASPLOS '23: "BeeHive: Sub-second Elasticity ..." BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution Ziming Zhao, Mingyu Wu, Jiawei Tang, Binyu Zang, Zhaoguo Wang, and Haibo Chen (Shanghai Jiao Tong University, China; Shanghai AI Laboratory, China; Engineering Research Center for Domain-specific Operating Systems of the Ministry of Education of China, China) Function-as-a-service (FaaS), an emerging cloud computing paradigm, is expected to provide strong elasticity due to its promise to auto-scale fine-grained functions rapidly. Although appealing for applications with good parallelism and dynamic workload, this paper shows that it is non-trivial to adapt existing monolithic applications (like web services) to FaaS due to their complexity. To bridge the gap between complicated web services and FaaS, this paper proposes a runtime-based Semi-FaaS execution model, which dynamically extracts time-consuming code snippets (closures) from applications and offloads them to FaaS platforms for execution. It further proposes BeeHive, an offloading framework for Semi-FaaS, which relies on the managed runtime to provide a fallback-based execution model and addresses the performance issues in traditional offloading mechanisms for FaaS. Meanwhile, the runtime system of BeeHive selects offloading candidates in a user-transparent way and supports efficient object sharing, memory management, and failure recovery in a distributed environment. The evaluation using various web applications suggests that the Semi-FaaS execution supported by BeeHive can reach sub-second resource provisioning on commercialized FaaS platforms like AWS Lambda, which is up to two orders of magnitude better than other alternative scaling approaches in cloud computing. @InProceedings{ASPLOS23p74, author = {Ziming Zhao and Mingyu Wu and Jiawei Tang and Binyu Zang and Zhaoguo Wang and Haibo Chen}, title = {BeeHive: Sub-second Elasticity for Web Services with Semi-FaaS Execution}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {74--87}, doi = {10.1145/3575693.3575752}, year = {2023}, } Publisher's Version
	Zheng, Bojian	ASPLOS '23: "Hidet: Task-Mapping Programming ..." Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs Yaoyao Ding, Cody Hao Yu, Bojian Zheng, Yizhi Liu, Yida Wang, and Gennady Pekhimenko (University of Toronto, Canada; Vector Institute, Canada; Amazon Web Services, USA) As deep learning models nowadays are widely adopted by both cloud services and edge devices, reducing the latency of deep learning model inferences becomes crucial to provide efficient model serving. However, it is challenging to develop efficient tensor programs for deep learning operators due to the high complexity of modern accelerators (e.g., NVIDIA GPUs and Google TPUs) and the rapidly growing number of operators. Deep learning compilers, such as Apache TVM, adopt declarative scheduling primitives to lower the bar of developing tensor programs. However, we show that this approach is insufficient to cover state-of-the-art tensor program optimizations (e.g., double buffering). In this paper, we propose to embed the scheduling process into tensor programs and use dedicated mappings, called task mappings, to define the computation assignment and ordering directly in the tensor programs. This new approach greatly enriches the expressible optimizations by allowing developers to manipulate tensor programs at a much finer granularity (e.g., allowing program-statement-level optimizations). We call the proposed method the task-mapping programming paradigm. In addition, we propose a new post-scheduling fusion optimization that allows developers to focus on scheduling every single operator and automates the fusion after scheduling. It greatly reduces the engineering efforts for operator fusion. Our proposed paradigm also constructs an efficient hardware-centric schedule space, which is agnostic to the program input size and greatly reduces the tuning time. With the proposed paradigm, we implement a deep learning compiler Hidet. Extensive experiments on modern convolution and transformer models show that Hidet outperforms state-of-the-art DNN inference framework, ONNX Runtime, and compiler, TVM equipped with scheduler AutoTVM and Ansor, by up to 1.48x (1.22x on average). It also reduces the tuning time by 20x and 11x compared with AutoTVM and Ansor, respectively. We open-sourced hidet at https://www.github.com/hidet-org/hidet. @InProceedings{ASPLOS23p370, author = {Yaoyao Ding and Cody Hao Yu and Bojian Zheng and Yizhi Liu and Yida Wang and Gennady Pekhimenko}, title = {Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {370--384}, doi = {10.1145/3575693.3575702}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zheng, Hongzhong	ASPLOS '23: "Spada: Accelerating Sparse ..." Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow Zhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu, Hongzhong Zheng, Yuan Xie, and Mingyu Gao (Tsinghua University, China; Northwestern University, USA; Alibaba DAMO Academy, China; Shanghai Qi Zhi Institute, China) Sparse matrix-matrix multiplication (SpGEMM) is widely used in many scientific and deep learning applications. The highly irregular structures of SpGEMM limit its performance and efficiency on conventional computation platforms, and thus motivate a large body of specialized hardware designs. Existing SpGEMM accelerators only support specific types of rigid execution dataflow such as inner/output-product or row-based schemes. Each dataflow is only optimized for certain sparse patterns and fails to generalize with robust performance to the widely diverse SpGEMM workloads across various domains. We propose Spada, a combination of three novel techniques for SpGEMM accelerators to efficiently adapt to various sparse patterns. First, we describe a window-based adaptive dataflow that can be flexibly adapted to different modes to best match the data distributions and realize different reuse benefits. Then, our hardware architecture efficiently supports this dataflow template, with flexible, fast, and low-cost reconfigurability and effective load balancing features. Finally, we use a profiling-guided approach to detect the sparse pattern and determine the optimized dataflow mode to use, based on the key observations of sparse pattern similarity in nearby matrix regions. Our evaluation results demonstrate that Spada is able to match or exceed the best among three state-of-the-art SpGEMM accelerators, and avoid the performance degradation of the others if data distribution and dataflow mismatch. It achieves an average 1.44× speedup across a wide range of sparse matrices and compressed neural network models. @InProceedings{ASPLOS23p747, author = {Zhiyao Li and Jiaxiang Li and Taijie Chen and Dimin Niu and Hongzhong Zheng and Yuan Xie and Mingyu Gao}, title = {Spada: Accelerating Sparse Matrix Multiplication with Adaptive Dataflow}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {747--761}, doi = {10.1145/3575693.3575706}, year = {2023}, } Publisher's Version
	Zheng, Lianmin	ASPLOS '23: "TensorIR: An Abstraction for ..." TensorIR: An Abstraction for Automatic Tensorized Program Optimization Siyuan Feng, Bohan Hou, Hongyi Jin, Wuwei Lin, Junru Shao, Ruihang Lai, Zihao Ye, Lianmin Zheng, Cody Hao Yu, Yong Yu, and Tianqi Chen (Shanghai Jiao Tong University, China; Carnegie Mellon University, USA; OctoML, USA; University of Washington, USA; University of California at Berkeley, USA; Amazon Web Services, USA) Deploying deep learning models on various devices has become an important topic. The wave of hardware specialization brings a diverse set of acceleration primitives for multi-dimensional ten- sor computations. These new acceleration primitives, along with the emerging machine learning models, bring tremendous engineering challenges. In this paper, we present TensorIR, a compiler abstraction for optimizing programs with these tensor computation primitives. TensorIR generalizes the loop nest representation used in existing machine learning compilers to bring tensor computation as the first-class citizen. Finally, we build an end-to-end framework on top of our abstraction to automatically optimize deep learning models for given tensor computation primitives. Experimental results show that TensorIR compilation automatically uses the tensor computation primitives for given hardware backends and delivers performance that is competitive to state-of-art hand-optimized systems across platforms. @InProceedings{ASPLOS23p804, author = {Siyuan Feng and Bohan Hou and Hongyi Jin and Wuwei Lin and Junru Shao and Ruihang Lai and Zihao Ye and Lianmin Zheng and Cody Hao Yu and Yong Yu and Tianqi Chen}, title = {TensorIR: An Abstraction for Automatic Tensorized Program Optimization}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {804--817}, doi = {10.1145/3575693.3576933}, year = {2023}, } Publisher's Version
	Zhong, Li	ASPLOS '23: "Protecting Data Integrity ..." Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code Haochen Huang, Bingyu Shen, Li Zhong, and Yuanyuan Zhou (University of California at San Diego, USA) Database-backed web applications persist a large amount of production data and have high requirements for integrity. To protect data integrity against application code bugs and operator mistakes, most RDBMSes allow application developers to specify various types of integrity constraints. Unfortunately, applications (e.g., e-commerce web apps) often do not take full advantage of this capability and miss specifying many database constraints, resulting in many severe consequences, such as crashing the order placement page and corrupting the store inventory data. In this paper, we focus on the problem of missing database constraints in web applications. We first study several widely used open-source e-commerce and communication applications, and observe that all these applications have missed integrity constraints and many were added later as afterthoughts after issues occurred. Motivated by our observations, we build a tool called CFinder to automatically infer missing database constraints from application source code by cleverly leveraging the observation that many source code patterns usually imply certain data integrity constraints. By analyzing application source code automatically, CFinder can extract such constraints and check against their database schemas to detect missing ones. We evaluate CFinder with eight widelydeployed web applications, including one commercial company with millions of users. Overall, our tool identifies 210 previously unknown missing constraints. We have reported 92 of them to the developers of these applications, so far 75 are confirmed. Our tool achieves a precision of 78% and a recall of 79%. @InProceedings{ASPLOS23p632, author = {Haochen Huang and Bingyu Shen and Li Zhong and Yuanyuan Zhou}, title = {Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {632--645}, doi = {10.1145/3575693.3575699}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhong, Yinmin	ASPLOS '23: "ElasticFlow: An Elastic Serverless ..." ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning Diandian Gu, Yihao Zhao, Yinmin Zhong, Yifan Xiong, Zhenhua Han, Peng Cheng, Fan Yang, Gang Huang, Xin Jin, and Xuanzhe Liu (Peking University, China; Microsoft Research, China) This paper proposes ElasticFlow, an elastic serverless training platform for distributed deep learning. ElasticFlow provides a serverless interface with two distinct features: (i) users specify only the deep neural network (DNN) model and hyperparameters for a job, but not the number of GPUs; (ii) users specify the deadline for a job, but not the amount of time to occupy GPUs. In contrast to existing server-centric platforms, ElasticFlow provides performance guarantees in terms of meeting deadlines while alleviating tedious, low-level, and manual resource management for deep learning developers. The characteristics of distributed training introduce two challenges. First, the training throughput scales non-linearly with the number of GPUs. Second, the scaling efficiency is affected by worker placement. To address these challenges, we propose Minimum Satisfactory Share to capture the resource usage of training jobs to meet deadlines, and ElasticFlow performs admission control based on it. We develop a greedy algorithm that dynamically allocates resources to admitted jobs based on diminishing returns. We apply buddy allocation to worker placement to eliminate the effect of topology. Evaluation results on a cluster of 128 GPUs show that ElasticFlow increases the number of jobs that can meet their deadlines by 1.46–7.65× compared to existing solutions. @InProceedings{ASPLOS23p266, author = {Diandian Gu and Yihao Zhao and Yinmin Zhong and Yifan Xiong and Zhenhua Han and Peng Cheng and Fan Yang and Gang Huang and Xin Jin and Xuanzhe Liu}, title = {ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {266--280}, doi = {10.1145/3575693.3575721}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhou, Yajin	ASPLOS '23: "VDom: Fast and Unlimited Virtual ..." VDom: Fast and Unlimited Virtual Domains on Multiple Architectures Ziqi Yuan, Siyu Hong, Rui Chang, Yajin Zhou, Wenbo Shen, and Kui Ren (Zhejiang University, China) Hardware memory domain primitives, such as Intel MPK and ARM Memory Domain, have been used for efficient in-process memory isolation. However, they can only provide a limited number of memory domains (16 domains), which cannot satisfy the compelling need for more isolated domains inside the address space of a process. Existing solutions to virtualize memory domains are either intrusive (need the modification to existing hardware), or incur a large performance overhead. In this paper, we propose VDom, a fast and scalable memory domain virtualization system that supports unlimited memory domains. VDom leverages separate address spaces to provide an unlimited number of virtual domains, and optimizes related memory management operations. To map virtual domains to hardware domains, we design a domain virtualization algorithm, which manages address spaces and domain maps for threads to efficiently access other domains that are unmapped in the current address space. According to our evaluation on real Intel and ARM platforms, on real-world server applications (httpd and MySQL), VDom incurs less than 2.65% performance overhead, which is lower than the overheads of the state-of-the-art software approaches (libmpk and EPK). In random domain access tests, VDom is comparable to EPK and has significantly higher efficiency than libmpk. @InProceedings{ASPLOS23p905, author = {Ziqi Yuan and Siyu Hong and Rui Chang and Yajin Zhou and Wenbo Shen and Kui Ren}, title = {VDom: Fast and Unlimited Virtual Domains on Multiple Architectures}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {905--919}, doi = {10.1145/3575693.3575735}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zhou, Yangjie	ASPLOS '23: "uGrapher: High-Performance ..." uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks Yangjie Zhou, Jingwen Leng, Yaoxu Song, Shuwen Lu, Mian Wang, Chao Li, Minyi Guo, Wenting Shen, Yong Li, Wei Lin, Xiangwen Liu, and Hanqing Wu (Shanghai Jiao Tong University, China; Alibaba Group, China) As graph neural networks (GNNs) have achieved great success in many graph learning problems, it is of paramount importance to support their efficient execution. Different graphs and different operators present different patterns during execution. However, there is still a gap in the existing GNN acceleration research to explore adaptive parallelism. We show that existing GNN frameworks rely on handwritten static kernels, which fail to achieve the best performance across different graph operators and input graph structures. In this work, we propose uGrapher, a unified interface that achieves general high performance for different graph operators and datasets. The existing GNN frameworks can easily integrate our design for its simple and unified API. We take a principled approach that decouples a graph operator’s computation and schedule to achieve that. We first build a GNN-specific operator abstraction that incorporates the semantics of graph tensors and graph loops. We explore various schedule strategies based on the abstraction that can balance the well-established trade-off relationship between parallelism, locality, and efficiency. Our evaluation shows that uGrapher can bring up to 29.1× (3.5× on average) performance improvement over the state-of-the-art baselines on two studied NVIDIA GPUs. @InProceedings{ASPLOS23p878, author = {Yangjie Zhou and Jingwen Leng and Yaoxu Song and Shuwen Lu and Mian Wang and Chao Li and Minyi Guo and Wenting Shen and Yong Li and Wei Lin and Xiangwen Liu and Hanqing Wu}, title = {uGrapher: High-Performance Graph Operator Computation via Unified Abstraction for Graph Neural Networks}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {878--891}, doi = {10.1145/3575693.3575723}, year = {2023}, } Publisher's Version
	Zhou, Yuanyuan	ASPLOS '23: "Protecting Data Integrity ..." Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code Haochen Huang, Bingyu Shen, Li Zhong, and Yuanyuan Zhou (University of California at San Diego, USA) Database-backed web applications persist a large amount of production data and have high requirements for integrity. To protect data integrity against application code bugs and operator mistakes, most RDBMSes allow application developers to specify various types of integrity constraints. Unfortunately, applications (e.g., e-commerce web apps) often do not take full advantage of this capability and miss specifying many database constraints, resulting in many severe consequences, such as crashing the order placement page and corrupting the store inventory data. In this paper, we focus on the problem of missing database constraints in web applications. We first study several widely used open-source e-commerce and communication applications, and observe that all these applications have missed integrity constraints and many were added later as afterthoughts after issues occurred. Motivated by our observations, we build a tool called CFinder to automatically infer missing database constraints from application source code by cleverly leveraging the observation that many source code patterns usually imply certain data integrity constraints. By analyzing application source code automatically, CFinder can extract such constraints and check against their database schemas to detect missing ones. We evaluate CFinder with eight widelydeployed web applications, including one commercial company with millions of users. Overall, our tool identifies 210 previously unknown missing constraints. We have reported 92 of them to the developers of these applications, so far 75 are confirmed. Our tool achieves a precision of 78% and a recall of 79%. @InProceedings{ASPLOS23p632, author = {Haochen Huang and Bingyu Shen and Li Zhong and Yuanyuan Zhou}, title = {Protecting Data Integrity of Web Applications with Database Constraints Inferred from Application Code}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {632--645}, doi = {10.1145/3575693.3575699}, year = {2023}, } Publisher's Version Published Artifact Artifacts Available Artifacts Functional Results Reproduced
	Zomaya, Albert	ASPLOS '23: "Achieving Sub-second Pairwise ..." Achieving Sub-second Pairwise Query over Evolving Graphs Hongtao Chen, Mingxing Zhang, Ke Yang, Kang Chen, Albert Zomaya, Yongwei Wu, and Xuehai Qian (Tsinghua University, China; Beijing HaiZhi XingTu Technology, China; University of Sydney, Australia; Purdue University, USA) Many real-time OLAP systems have been proposed to query evolving data with sub-second latency. Although this feature is highly attractive, it is very hard to be achieved on analytic graph queries that can only be answered after accessing every connected vertex. Fortunately, researchers recently observed that answering pairwise queries is enough for many real-world scenarios. These pairwise queries avoid the exhaustive nature and hence may only need to access a small portion of the graph. Obviously, the crux of achieving low latency is to what extent the system can eliminate unnecessary computations. This pruning process, according to our investigation, is usually achieved by estimating certain upper bounds of the query result in existing systems. However, our evaluation results demonstrate that these existing upper-bound-only pruning techniques can only prune about half of the vertex activations, which is still far away from achieving the sub-second latency goal on large graphs. In contrast, we found that it is possible to substantially accelerate the processing if we are able to not only estimate the upper bounds, but also foresee a tighter lower bound for certain pairs of vertices in the graph. Our experiments show that only less than 1% of the vertices are activated via using this novel lower bound based pruning technique. Based on this observation, we build SGraph, a system that is able to answer dynamic pairwise queries over evolving graphs with sub-second latency. It can ingest millions of updates per second and simultaneously answer pairwise queries with a latency that is several orders of magnitude smaller than state-of-the-art systems. @InProceedings{ASPLOS23p1, author = {Hongtao Chen and Mingxing Zhang and Ke Yang and Kang Chen and Albert Zomaya and Yongwei Wu and Xuehai Qian}, title = {Achieving Sub-second Pairwise Query over Evolving Graphs}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {1--15}, doi = {10.1145/3575693.3576173}, year = {2023}, } Publisher's Version
	Zulkifli, Kahfi S.	ASPLOS '23: "EVStore: Storage and Caching ..." EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems Daniar H. Kurniawan, Ruipu Wang, Kahfi S. Zulkifli, Fandi A. Wiranata, John Bent, Ymir Vigfusson, and Haryadi S. Gunawi (University of Chicago, USA; Beijing University of Technology, China; Bandung Institute of Technology, Indonesia; Seagate Technology, USA; Emory University, USA) Modern recommendation systems, primarily driven by deep-learning models, depend on fast model inferences to be useful. To tackle the sparsity in the input space, particularly for categorical variables, such inferences are made by storing increasingly large embedding vector (EV) tables in memory. A core challenge is that the inference operation has an all-or-nothing property: each inference requires multiple EV table lookups, but if any memory access is slow, the whole inference request is slow. In our paper, we design, implement and evaluate EVStore, a 3-layer EV table lookup system that harnesses both structural regularity in inference operations and domain-specific approximations to provide optimized caching, yielding up to 23% and 27% reduction on the average and p90 latency while quadrupling throughput at 0.2% loss in accuracy. Finally, we show that at a minor cost of accuracy, EVStore can reduce the Deep Recommendation System (DRS) memory usage by up to 94%, yielding potentially enormous savings for these costly, pervasive systems. @InProceedings{ASPLOS23p281, author = {Daniar H. Kurniawan and Ruipu Wang and Kahfi S. Zulkifli and Fandi A. Wiranata and John Bent and Ymir Vigfusson and Haryadi S. Gunawi}, title = {EVStore: Storage and Caching Capabilities for Scaling Embedding Tables in Deep Recommendation Systems}, booktitle = {Proc.\ ASPLOS}, publisher = {ACM}, pages = {281--294}, doi = {10.1145/3575693.3575718}, year = {2023}, } Publisher's Version