Demo Data from SIGIR 2026 / reviewers in the wild / expert
Jidong Zhai
dblp:72/7413
· DBLP profile ↗
151ranked-venue papers
11as first author
93since 2021 · last 2026
0000-0002-7656-6428ORCID · verified
Domains — the database's venue-derived domains; a paper can count in several
Systems, architecture and hardware · 115 · 9 first-author · 66 since 2021Databases, data management, data science and information retrieval · 15 · 12 since 2021Software engineering, systems software and programming languages · 14 · 11 since 2021Applied, interdisciplinary, general and emerging computing · 7 · 2 first-author · 3 since 2021Artificial intelligence and machine learning · 6 · 5 since 2021Graphics, computer vision, multimedia, augmented reality and games · 3 · 2 since 2021Computer networks · 1
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Computer architecture, parallel and distributed computing, and storage systems
87 papers |
Parallel and multicore computing · 20% GPUs and heterogeneous computing · 13% Performance modeling and evaluation · 12% | |
| Artificial intelligence
32 papers |
Efficient and distributed learning · 54% Deep learning architectures and training · 13% Language models and text generation · 10% | |
| Software engineering, system software, and programming languages
24 papers |
Compilers and program optimization · 74% Program analysis · 12% Operating systems · 9% | |
| Databases, data mining, and information retrieval
16 papers |
Query processing and optimization · 32% Data mining · 10% Indexing and storage engines · 10% |
Topics — the 30 heaviest of 248, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Hardware accelerators and domain-specific architectures
machine learning accelerator |
4.1 | 7 | 2026 | FlashFuser: Expanding the Scale of Kernel Fusion for Compute-Intensive Operators via Inter-Core Connection · HPCA 2026 QFactory: Accelerating Quantized Large Language Model Serving with Qtile Graphs · USENIX ATC 2025 PluS: Highly Efficient and Expandable ML Compiler with Pluggable Graph Schedules · USENIX ATC 2025 |
Machine learning › Efficient and distributed learning
distributed training |
4.0 | 6 | 2026 | UniEP: Unified Expert-Parallel MegaKernel MoE for LLM Training · HPDC 2026 mTuner: Accelerating Parameter-Efficient Fine-Tuning on Multi-GPU Servers with Elastic Tensor · USENIX ATC 2025 POSTER: Pattern-Aware Sparse Communication for Scalable Recommendation Model Training · PPoPP 2024 |
Compilers and program optimization › deep learning compiler
tensor program optimization |
3.8 | 6 | 2025 | FlashTensor: Optimizing Tensor Programs by Leveraging Fine-grained Tensor Property · PPoPP 2025 Optimizing DNNs With Partially Equivalent Transformations and Automated Corrections · IEEE Trans. Computers 2023 EINNET: Optimizing Tensor Programs with Derivation-Based Transformations · OSDI 2023 |
Query processing and optimization
compressed data processing |
3.4 | 5 | 2026 | Enabling Tile-Based Direct Query on Adaptively Compressed Data With GPU Acceleration · IEEE Trans. Parallel Distributed Syst. 2026 ComStar: Compression-Aware Stream Query for Heterogeneous Hybrid Architecture · IEEE Trans. Parallel Distributed Syst. 2026 POCLib: A High-Performance Framework for Enabling Near Orthogonal Processing on Compression · IEEE Trans. Parallel Distributed Syst. 2022 |
Machine learning › Efficient and distributed learning
model compression |
2.9 | 4 | 2026 | RoMeo: Mitigating Dual-dimensional Outliers with Rotated Mixed Precision Quantization · PPoPP 2026 mTuner: Accelerating Parameter-Efficient Fine-Tuning on Multi-GPU Servers with Elastic Tensor · USENIX ATC 2025 Efficient Inference for Pruned CNN Models on Mobile Devices With Holistic Sparsity Alignment · IEEE Trans. Parallel Distributed Syst. 2024 |
Compilers and program optimization
machine learning compiler |
2.1 | 3 | 2025 | PluS: Highly Efficient and Expandable ML Compiler with Pluggable Graph Schedules · USENIX ATC 2025 BladeDISC: Optimizing Dynamic Shape Machine Learning Workloads via Compiler Approach · Proc. ACM Manag. Data 2023 AStitch: enabling a new multi-dimensional optimization space for memory-intensive ML training and inference on modern SIMT architectures · ASPLOS 2022 |
Parallel and multicore computing
parallel programming models |
2.0 | 4 | 2026 | WiseGraph: Optimizing GNN with Joint Workload Partition of Graph and Operations · EuroSys 2024 UniQ: A Unified Programming Model for Efficient Quantum Circuit Simulation · SC 2022 HiWayLib: A Software Framework for Enabling High Performance Communications for Heterogeneous Pipeline Computations · ASPLOS 2019 |
GPUs and heterogeneous computing
GPU computing |
1.9 | 3 | 2024 | G-Learned Index: Enabling Efficient Learned Index on GPU · IEEE Trans. Parallel Distributed Syst. 2024 Enabling Efficient Random Access to Hierarchically Compressed Text Data on Diverse GPU Platforms · IEEE Trans. Parallel Distributed Syst. 2023 G-TADOC: Enabling Efficient GPU-Based Text Analytics without Decompression · ICDE 2021 |
Machine learning › Efficient and distributed learning
KV cache management |
1.9 | 2 | 2026 | Lethe: Layer- and Time-Adaptive KV Cache Pruning for Reasoning-Intensive LLM Serving · AAAI 2026 Jenga: Effective Memory Management for Serving LLM with Heterogeneity · SOSP 2025 |
Cloud and datacenter computing
inference serving |
1.9 | 2 | 2026 | ChituDiffusion: A Data-Characteristic-Aware Serving System for Diffusion Models · PPoPP 2026 QFactory: Accelerating Quantized Large Language Model Serving with Qtile Graphs · USENIX ATC 2025 |
Emerging computing paradigms
quantum computer architecture |
1.6 | 3 | 2022 | UniQ: A Unified Programming Model for Efficient Quantum Circuit Simulation · SC 2022 Suppressing ZZ crosstalk of Quantum computers through pulse and scheduling co-optimization · ASPLOS 2022 Mitigating Crosstalk in Quantum Computers through Commutativity-Based Instruction Reordering · DAC 2021 |
Performance modeling and evaluation
workload characterization |
1.5 | 5 | 2022 | Detecting Performance Variance for Parallel Applications Without Source Code · IEEE Trans. Parallel Distributed Syst. 2022 Leveraging Code Snippets to Detect Variations in the Performance of HPC Systems · IEEE Trans. Parallel Distributed Syst. 2022 Understanding Co-Running Behaviors on Integrated CPU/GPU Architectures · IEEE Trans. Parallel Distributed Syst. 2017 |
Performance modeling and evaluation
performance variability |
1.5 | 3 | 2022 | Detecting Performance Variance for Parallel Applications Without Source Code · IEEE Trans. Parallel Distributed Syst. 2022 Leveraging Code Snippets to Detect Variations in the Performance of HPC Systems · IEEE Trans. Parallel Distributed Syst. 2022 vSensor: leveraging fixed-workload snippets of programs for performance variance detection · PPoPP 2018 |
Performance modeling and evaluation › performance variability
performance variance detection |
1.5 | 3 | 2022 | Detecting Performance Variance for Parallel Applications Without Source Code · IEEE Trans. Parallel Distributed Syst. 2022 Leveraging Code Snippets to Detect Variations in the Performance of HPC Systems · IEEE Trans. Parallel Distributed Syst. 2022 vSensor: leveraging fixed-workload snippets of programs for performance variance detection · PPoPP 2018 |
Compilers and program optimization › deep learning compiler
operator fusion |
1.3 | 2 | 2024 | Optimal Kernel Orchestration for Tensor Programs with Korch · ASPLOS (3) 2024 AStitch: enabling a new multi-dimensional optimization space for memory-intensive ML training and inference on modern SIMT architectures · ASPLOS 2022 |
Distributed systems
root cause analysis |
1.3 | 2 | 2025 | Leveraging Graph Analysis to Pinpoint Root Causes of Scalability Issues for Parallel Applications · IEEE Trans. Parallel Distributed Syst. 2025 ScalAna: automating scaling loss detection with graph analysis · SC 2020 |
Graph algorithms and graph theory › graph algorithms
graph pattern matching |
1.3 | 2 | 2025 | X-SET: An Efficient Graph Pattern Matching Accelerator With Order-Aware Parallel Intersection Units · MICRO 2025 GraphPi: high performance graph pattern matching through effective redundancy elimination · SC 2020 |
Machine learning › Generative modeling
diffusion model |
1.3 | 2 | 2026 | DiTango: Cost-Effective Parallel Diffusion Generation with Selective Attention State Reuse · HPDC 2026 ChituDiffusion: A Data-Characteristic-Aware Serving System for Diffusion Models · PPoPP 2026 |
Storage systems
data compression |
1.3 | 5 | 2026 | Exploring Data Analytics Without Decompression on Embedded GPU Systems · IEEE Trans. Parallel Distributed Syst. 2022 Enabling Tile-Based Direct Query on Adaptively Compressed Data With GPU Acceleration · IEEE Trans. Parallel Distributed Syst. 2026 POCLib: A High-Performance Framework for Enabling Near Orthogonal Processing on Compression · IEEE Trans. Parallel Distributed Syst. 2022 |
Machine learning › Graph learning
graph neural network |
1.3 | 2 | 2024 | WiseGraph: Optimizing GNN with Joint Workload Partition of Graph and Operations · EuroSys 2024 Understanding and bridging the gaps in current GNN performance optimizations · PPoPP 2021 |
Parallel and multicore computing
task partitioning |
1.3 | 2 | 2024 | WiseGraph: Optimizing GNN with Joint Workload Partition of Graph and Operations · EuroSys 2024 Automatic Irregularity-Aware Fine-Grained Workload Partitioning on Integrated Architectures · IEEE Trans. Knowl. Data Eng. 2021 |
Natural language and speech › Information extraction and text analysis
semantic parsing |
1.2 | 2 | 2023 | Unveiling the Black Box of PLMs with Semantic Anchors: Towards Interpretable Neural Semantic Parsing · AAAI 2023 GraphQ IR: Unifying the Semantic Parsing of Graph Query Languages with One Intermediate Representation · EMNLP 2022 |
Natural language and speech › Language models and text generation
large language model inference |
1.2 | 2 | 2026 | RoMeo: Mitigating Dual-dimensional Outliers with Rotated Mixed Precision Quantization · PPoPP 2026 MixQ: Taming Dynamic Outliers in Mixed-Precision Quantization by Online Prediction · SC 2024 |
Performance modeling and evaluation
performance prediction |
1.2 | 4 | 2024 | A Survey on Performance Modeling and Prediction for Distributed DNN Training · IEEE Trans. Parallel Distributed Syst. 2024 Performance Prediction for Large-Scale Parallel Applications Using Representative Replay · IEEE Trans. Computers 2016 PHANTOM: predicting performance of parallel applications on large-scale parallel machines using a single node · PPoPP 2010 |
Machine learning › Transfer learning and domain adaptation › pre-training and adaptation
large-scale pre-trained models |
1.1 | 2 | 2022 | BaGuaLu: targeting brain scale pretrained models with over 37 million cores · PPoPP 2022 FasterMoE: modeling and optimizing training of large-scale dynamic pre-trained models · PPoPP 2022 |
Program analysis
static analysis |
1.0 | 6 | 2025 | Leveraging Graph Analysis to Pinpoint Root Causes of Scalability Issues for Parallel Applications · IEEE Trans. Parallel Distributed Syst. 2025 Graph-Centric Performance Analysis for Large-Scale Parallel Applications · IEEE Trans. Parallel Distributed Syst. 2024 CYPRESS: Combining Static and Dynamic Analysis for Top-Down Communication Trace Compression · SC 2014 |
Machine learning › Generative modeling › diffusion model
diffusion transformer |
1.0 | 1 | 2026 | DiTango: Cost-Effective Parallel Diffusion Generation with Selective Attention State Reuse · HPDC 2026 |
Machine learning › Efficient and distributed learning
inference efficiency |
1.0 | 1 | 2026 | Lethe: Layer- and Time-Adaptive KV Cache Pruning for Reasoning-Intensive LLM Serving · AAAI 2026 |
Machine learning › Efficient and distributed learning
inference serving |
1.0 | 1 | 2026 | ChituDiffusion: A Data-Characteristic-Aware Serving System for Diffusion Models · PPoPP 2026 |
Machine learning › Efficient and distributed learning › model compression › transformer compression › attention compression
KV cache pruning |
1.0 | 1 | 2026 | Lethe: Layer- and Time-Adaptive KV Cache Pruning for Reasoning-Intensive LLM Serving · AAAI 2026 |
Methods — techniques the papers use, named apart from their topics
profiling · 5.0graph analysis · 3.7compression · 3.6automated correction · 2.3tracing · 2.2LSTM · 2.0selective attention state reuse · 2.0anchor-guided state selection · 2.0ahead-of-time scheduling · 2.0thread-level workload scheduling · 1.1visibility metadata · 1.0tile-based execution · 1.0tape restructuring · 1.0recency-aware retention · 1.0mixed-precision quantization · 1.0layerwise sparsity allocation · 1.0fine-grained batching · 1.0direct indexing · 1.0
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Lethe: Layer- and Time-Adaptive KV Cache Pruning for Reasoning-Intensive LLM ServingabstractGenerative reasoning with large language models (LLMs) often involves long decoding sequences, leading to substantial memory and latency overheads from accumulating key-value (KV) caches. While existing KV compression methods primarily focus on reducing prefill memory from long input sequences, they fall short in addressing the dynamic and layer-sensitive nature of long-form generation, which is central to reasoning tasks. We propose Lethe, a dynamic KV cache management framework that introduces adaptivity along both the spatial and temporal dimensions of decoding. Along the spatial dimension, Lethe performs layerwise sparsity-aware allocation, assigning token pruning budgets to each transformer layer based on estimated attention redundancy. Along the temporal dimension, Lethe conducts multi-round token pruning during generation, driven by a Recency-Aware Selective Retention (RASR) mechanism. RASR extends traditional recency-based heuristics by also considering token relevance derived from evolving attention patterns, enabling informed decisions about which tokens to retain or evict. Empirical results demonstrate that Lethe achieves a favorable balance between efficiency and generation quality across diverse models and tasks, increases throughput by up to 2.56×. Daming Zhao, WenXuan Hou, Tianyang Zheng, Weiye Ji, Jidong Zhai |
AAAI | 8 |
| 2026 | FlashFuser: Expanding the Scale of Kernel Fusion for Compute-Intensive Operators via Inter-Core ConnectionabstractThe scaling of computation throughput continues to outpace improvements in memory bandwidth, making many deep learning workloads memory-bound. Kernel fusion is a key technique to alleviate this problem, but the fusion strategies of existing compilers and frameworks are limited to using local scratchpad memory. When the intermediate results exceed the limited capacity (such as FFN), the fusion fails. Although modern GPUs (like the NVIDIA H100) now incorporate an inter-core connection mechanism known as Distributed Shared Memory (DSM)—providing a larger, high-bandwidth, and low-latency on-chip memory pool—this hardware potential has yet to be exploited by software frameworks. To bridge this gap, we present FlashFuser, the first compiler framework to utilize inter-core connection for kernel fusion on modern GPUs. FlashFuser extends established fusion techniques to the DSM domain through three core contributions. First, we propose a powerful DSM-based communication abstraction that formalizes complex cluster-based data exchange patterns, such as reduce, shuffle and multiply. Second, we introduce a dataflow analyzer that generalizes loop scheduling, resource mapping, and tile selection to the distributed memory hierarchy; it determines the optimal execution order and tile sizes by quantifying data movement across memory levels. Finally, FlashFuser integrates these components into a unified search engine that employs analytical cost modeling and DSM-aware pruning strategies to efficiently discover the optimal execution plan. Our evaluation on an NVIDIA H100 GPU shows that FlashFuser reduces memory access by 58 % and delivers kernel speedups of$3.3 x$against highly-tuned libraries and 4.1x against state-of-the-art compilers, resulting in a$1.24 \times$end-to-end speedup. Yangjie Zhou 0001, Zihan Liu 0002, Xinhao Luo, Yijia Diao, Minyi Guo, Jidong Zhai, Yu Feng 0007, Chen Zhang 0001, Anbang Wu, Jingwen Leng |
HPCA | 7 |
| 2026 | DiTango: Cost-Effective Parallel Diffusion Generation with Selective Attention State ReuseabstractRecent advances in AI-generated content have driven widespread adoption of Diffusion Transformers (DiTs) for high-resolution, long-duration content generation. While parallelization techniques accelerate diffusion inference, they face significant scalability challenges due to excessive communication overhead in multi-node environments. We observe that sequence partitions in Context Parallelism (CP) exhibit distinct heterogeneity: spatially proximate partitions contribute more significantly to attention computation results. By mapping this heterogeneous pattern to hierarchical communication topology, we can access high-contribution partitions with reduced communication cost. This insight motivates our novel selective attention state mechanism that strategically balances partial attention computation and historical result reuse across denoising steps. We present DiTango, an efficient parallel framework for DiT generation. DiTango features an anchor-guided state selection planner that optimizes computation-reuse decisions for each partition, complemented by a runtime that orchestrates efficient state-centric operations. This design achieves superior system efficiency while preserving generation quality. Experimental evaluation on popular diffusion models demonstrates that DiTango achieves up to 1.9x end-to-end and 3.2x attention speedup with near-linear scaling in multi-node settings, while maintaining generation quality comparable to state-of-the-art approaches. Runxin Zhong, Zan Zong, Hengjie Li, Yuyang Jin 0001, Jidong Zhai |
HPDC | 6 |
| 2026 | Odin: Harnessing Rendering Visibility to Break Global Barriers in Distributed Point-Based Neural RenderingabstractWe present Odin, a training system for distributed point-based neural rendering (PBNR) that accelerates training by removing unnecessary global barriers. Odin observes that two training steps need synchronization only when they update overlapping visible parameters. It uses lightweight visibility metadata and a two-phase design with ahead-of-time scheduling plus runtime refinement to handle irregular and dynamic behavior. Odin hides 82% of exposed communication and achieves up to 1.89 × throughput speedup on 64 GPUs while preserving baseline accuracy. Zhenxiang Ma, Yuanzhen Zhou, Yuchang Zhang, Miao Tao, Jidong Zhai, Hengjie Li |
HPDC | 8 |
| 2026 | SYCL++: A Unified Programming Framework for Heterogeneous Supercomputers at Scale
Zitao Shen, Yuyang Jin 0001, Kinman Lei, Zixuan Ma, Zhenchuan Chen, Di Wei, Fei Wang 0096, Ying Liu 0055, Lin Gan 0001, Jidong Zhai |
HPDC | 14 |
| 2026 | UniEP: Unified Expert-Parallel MegaKernel MoE for LLM TrainingabstractAs LLM training grows increasingly resource-intensive and expert parallelism (EP) becomes essential for scaling MoE models, EP optimizations are widely adopted in production frameworks like Megatron-LM. Existing solutions often rely on ad-hoc, complex kernels that lack adaptability across diverse optimization configurations and frequently neglect numerical stability, failing to meet the strict precision requirements of large-scale training. Size Zheng 0001, Xuegui Zheng, Li-Wen Chang, Jidong Zhai |
HPDC | 4 |
| 2026 | Taming Dynamic Diffusion LLM Inference through Virtual Static Execution
Jianian Zhu, Haojie Wang 0004, Ruixuan Li 0001, Jidong Zhai |
ICS | 6 |
| 2026 | ParDiff: Efficiently Parallelizing Reverse-Mode Automatic Differentiation with Direct IndexingabstractAutomatic Differentiation (AD) is a technique that computes the derivatives of numerical programs by systematically applying the chain rule, playing a critical role in domains such as machine learning, simulation, and control systems. However, parallelizing differentiated programs remains a significant challenge due to the conflict between tapes (a data structure for intermediate variable storage) and summations: the differentiation process inherently introduces inter-thread summation patterns, which require prohibitively expensive atomic operations; and traditional tape designs tightly couple data retrieval with the program’s control flow, preventing code restructuring needed to eliminate these costly dependencies. Shuhong Huang, Shizhi Tang, Yuan Wen, Huanqi Cao, Ruibai Tang, Yidong Chen 0003, Jiping Yu, Jidong Zhai |
PPoPP | 11 |
| 2026 | ChituDiffusion: A Data-Characteristic-Aware Serving System for Diffusion ModelsabstractDiffusion models have become the dominant approach for generative tasks in images, videos, and other domains. However, diverse data properties in generation requests, which are critical for efficient serving, remain underexploited. To address this issue, we propose a diffusion model serving system ChituDiffusion. ChituDiffusion leverages the locality of data properties to recompose a diffusion pipeline into dGraphs with shared optimization opportunities, enabling thorough compile-time and runtime co-optimizations. During compilation, ChituDiffusion compiles each dGraph into multiple execution engines optimized for specific data properties. At runtime, heterogeneous requests are elaborately reorganized into fine-grained batching tasks with similar properties and then efficiently executed by matched engines. Evaluation on five diffusion applications shows that ChituDiffusion improves the throughput by up to 2.13× (1.58× on average) on A100 and 2.19× (1.51× on average) on H100 compared with existing frameworks. The code for ChituDiffusion and the production traces have been made open-source at https://github.com/thu-pacman/chitu/tree/Diffusion. Chengzhang Wu, Liyan Zheng 0001, Haojie Wang 0004, Kezhao Huang, Zixuan Ma, Dong Dong 0001, Jidong Zhai |
PPoPP | 7 |
| 2026 | RoMeo: Mitigating Dual-dimensional Outliers with Rotated Mixed Precision QuantizationabstractMixed precision quantization has been adopted to accelerate large language models (LLMs) serving by leveraging high-throughput low-precision compute units in GPUs while preserving outliers in higher precision to maintain model accuracy. However, existing methods focus on mitigating single-dimensional channel-wise outliers, leading to model accuracy degradation when scaled to 4-bit precision. Qihao Zhang, Mingliang Tang, Mingshu Zhai, Kinman Lei, Jidong Zhai |
PPoPP | 5 |
| 2026 | A comprehensive taxonomy of prompt engineering techniques for large language modelsabstractAbstract Large Language Models (LLMs) have demonstrated remarkable performance across various downstream tasks, as evidenced by numerous studies. Since 2022, generative AI has shown significant potential in diverse application domains, including gaming, film and television, media, and finance. By 2023, the global AI-generated content (AIGC) industry had attracted over $26 billion in investment. As LLMs become increasingly prevalent, prompt engineering has emerged as a key research area to enhance user-AI interactions and improve LLM performance. The prompt, which serves as the input instruction for the LLM, is closely linked to the model’s responses. Prompt engineering refines the content and structure of prompts, thereby enhancing the performance of LLMs without changing the underlying model parameters. Despite significant advancements in prompt engineering, a comprehensive and systematic summary of existing techniques and their practical applications remains absent. To fill this gap, we investigate existing techniques and applications of prompt engineering. We conduct a thorough review and propose a novel taxonomy that provides a foundational framework for prompt construction. This taxonomy categorizes prompt engineering into four distinct aspects: profile and instruction, knowledge, reasoning and planning, and reliability. By providing a structured framework for understanding its various dimensions, we aim to facilitate the systematic design of prompts. Furthermore, we summarize existing prompt engineering techniques and explore the applications of LLMs across various domains, highlighting their interrelation with prompt engineering strategies. This survey underscores the progress of prompt engineering and its critical role in advancing AI applications, ultimately aiming to provide a systematic reference for future research and applications. Yao-Yang Liu, Zhen Zheng, Feng Zhang 0007, Jin-Cheng Feng, Yi-Yang Fu, Jidong Zhai, Bingsheng He, Xiao Zhang 0001, Xiaoyong Du 0001 |
Frontiers Comput. Sci. | 6 |
| 2026 | Concord: A GPU cluster scheduler with enhanced interference profiling and asymmetric job packing
Weiwei Lin 0001, Haijie Wu, Jidong Zhai, Keqin Li 0001 |
Future Gener. Comput. Syst. | 4 |
| 2026 | ComStar: Compression-Aware Stream Query for Heterogeneous Hybrid Architecture
Yani Liu, Feng Zhang 0007, Yu Zhang 0027, Shuhao Zhang 0001, Bingsheng He, Jidong Zhai, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 7 |
| 2026 | Enabling Tile-Based Direct Query on Adaptively Compressed Data With GPU AccelerationabstractThe explosive growth of data poses significant challenges for GPU-based databases, which must balance limited memory capacity with the need for high-speed query execution. Compression has become an essential technique for optimizing memory utilization and reducing data movement. However, its benefits have been limited to the necessary data decompression. Querying compressed data conventionally requires decompression, which causes the query process to be significantly slower than a direct query on uncompressed data. To address this problem, this paper presents a novel GPU-accelerated tile-based direct query framework that successfully eliminates the limitation, significantly enhancing query performance. By employing direct query strategies, the framework minimizes data movement and maximizes memory bandwidth utilization. It incorporates tilebased hardware-conscious execution strategies for direct query, including memory management and control flow coordination, to improve execution efficiency. Additionally, adaptive data-driven compression formats are paired with tailored SQL operators to enable efficient support for diverse queries. Our experiments, conducted using the Star Schema Benchmark, show an average improvement of 3.5× compared to the state-of-the-art tilebased decompression scheme, while maintaining the space-saving advantages of compression. Notably, our solution consistently outperforms existing direct execution schemes for compressed data across all query types. Yu Zhang 0027, Feng Zhang 0007, Yani Liu, Huanchen Zhang, Jidong Zhai, Wenchao Zhou, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2025 | IntelliGen: Instruction-Level Auto-tuning for Tensor Program with Monotonic Memory OptimizationabstractTensor compilers play a critical role in optimizing deep neural networks (DNNs), with memory performance emerging as a key bottleneck in code generation for DNN models. Existing tensor compilers are constrained by inefficient auto-tuning algorithms. They either must deploy coarse-grained descriptions, thus miss potential optimization, or struggle with vast search spaces, rendering auto-tuning inapplicable. Tensor compilers require a more holistic optimization of memory performance to overcome these constraints. To address this issue, we focus our optimization objective on memory performance, which allows us to design monotonic optimization methods, significantly enhancing the efficiency of auto-tuning and thus enabling auto-tuning on a fine-granularity description. Based on these observations, we propose IntelliGen, a tensor compiler with instruction-level auto-tuning and monotonic memory optimization. We design an instruction-level graph description, and a monotonic optimization method for optimization on . Benefiting from auto-tuning techniques with fine-grained description, IntelliGen demonstrates significant speedup of up to 3.13×, 3.55×, and 16.9× (averaging 1.46×, 1.85×, and 2.30×, respectively) on NVIDIA GPUs, AMD GPUs, and Cambricon MLUs over the most efficient existing frameworks. Zixuan Ma, Haojie Wang 0004, Jingze Xing, Shuhong Huang, Liyan Zheng 0001, Chen Zhang 0001, Huanqi Cao, Kezhao Huang, Mingshu Zhai, Shizhi Tang, Penghan Wang, Jidong Zhai |
CGO | 12 |
| 2025 | X-SET: An Efficient Graph Pattern Matching Accelerator With Order-Aware Parallel Intersection UnitsabstractGraph Pattern Matching (GPM) is a critical task in a wide range of graph analytics applications, such as social network analysis and cybersecurity.Despite its importance, GPM remains challenging to accelerate due to its inherently irregular control flow and heavy reliance on set operations, which dominate execution time and introduce data dependencies that limit parallelism.While recent GPM accelerators attempt to improve performance, they often overlook the ordered nature of input data, resulting in redundant computations and inefficient hardware utilization.This paper presents X-SET, a GPM accelerator that overcomes these limitations by introducing two key innovations.First, we propose an Order-Aware Set Intersection Unit (SIU), which exploits input ordering to reduce the hardware complexity of parallel set intersection from O (𝑁 2 ) to O (𝑁 log 𝑁 ), achieving high throughput and significant area savings by avoiding unnecessary comparisons.Second, we develop a barrier-free task scheduler that breaks traditional DFS scheduling constraints by enabling asynchronous, outof-order task execution across different levels of the GPM search tree.X-SET is integrated into a RISC-V SoC, supporting end-to-end acceleration.Extensive experimental results show that X-SET outperforms state-of-the-art GPM accelerators, achieving 4.6×-142.9×improvements in compute density, with a geometric mean of 13.7×, and delivering 6.4× geometric mean and 42.9× maximum speedup in end-to-end performance.X-SET is open-sourced at github 1 . Tianhui Shi, Shixuan Sun, Jidong Zhai, Xinyu Chen 0001 |
MICRO | 4 |
| 2025 | HSampler : Optimizing Multi-GPU GNN Sampling with Collision-Avoid Selection
Yuyang Jin 0001, Jidong Zhai, Kezhao Huang |
NPC (1) | 2 |
| 2025 | FlashTensor: Optimizing Tensor Programs by Leveraging Fine-grained Tensor PropertyabstractDeep neural networks (DNNs) have shown significant effectiveness in natural language processing and video applications. However, DNN models, especially for long-context tasks, introduce extremely large intermediate tensors, producing substantial memory overhead. Although considerable efforts have been made to optimize DNNs, insufficient awareness of tensor properties has hindered effective memory optimization and can lead to inefficient computations in a long-context scenario. Runxin Zhong, Yuyang Jin 0001, Chen Zhang 0001, Kinman Lei, Shuangyu Li, Jidong Zhai |
PPoPP | 6 |
| 2025 | TraceFlow: Efficient Trace Analysis for Large-Scale Parallel Applications via Interaction Pattern-Aware Trace DistributionabstractTrace analysis of large-scale parallel applications is crucial for understanding and optimizing performance. It primarily focuses on the interaction behaviors between different parallel processes, such as synchronization waits and asynchronous overlaps. The trace size explodes as the parallel scale of applications, thus current methods analyze traces in parallel to ensure analysis speed. However, due to the interaction pattern-agnostic trace distribution, they often introduce inter-process communications to fetch non-local event data during interaction analysis, leading to excessively long trace analysis time. Yuyang Jin 0001, Xirui Shui, Mingshu Zhai, Zan Zong, Feng Zhang 0007, Felix Wolf 0001, Jidong Zhai |
SC | 7 |
| 2025 | UltraAttn: Efficiently Parallelizing Attention through Hierarchical Context-TilingabstractLong-context comprehension is critical for large language models. Context parallelism and irregular block-sparse attention are keyss to accelerating long-context training and inference. Existing context parallelism suffers from poor scalability due to the striped-like partition pattern, which causes high communication traffic, and the ring-based communication pattern, which limits kernel granularity, reduces device utilization, and incurs redundant communication. Zan Zong, Yuyang Jin 0001, Kinman Lei, Jiaao He, Qigang Yang, Jidong Zhai |
SC | 7 |
| 2025 | Jenga: Effective Memory Management for Serving LLM with Heterogeneity
Chen Zhang 0001, Kuntai Du, Woosuk Kwon, Xiangxi Mo, Kaichao You, Zhuohan Li 0001, Mingsheng Long, Jidong Zhai, Joseph Gonzalez 0001, Ion Stoica |
SOSP | 11 |
| 2025 | HypeReca: Distributed Heterogeneous In-Memory Embedding Database for Training Recommender Models
Jiaao He, Shengqi Chen 0001, Kezhao Huang, Jidong Zhai |
USENIX ATC | 4 |
| 2025 | mTuner: Accelerating Parameter-Efficient Fine-Tuning on Multi-GPU Servers with Elastic Tensor
Kezhao Huang, Siqi Zhu, Mingshu Zhai, Liyan Zheng 0001, Kinman Lei, Jiaao He, Yuyang Jin 0001, Jidong Zhai |
USENIX ATC | 8 |
| 2025 | PluS: Highly Efficient and Expandable ML Compiler with Pluggable Graph Schedules
Zhen Zheng, Feng Zhang 0007, Chuanjie Liu, Zaifeng Pan, Jidong Zhai, Xiaoyong Du 0001 |
USENIX ATC | 6 |
| 2025 | QFactory: Accelerating Quantized Large Language Model Serving with Qtile Graphs
Qihao Zhang, Mingshu Zhai, Jidong Zhai |
USENIX ATC | 4 |
| 2025 | HARMONY: A Scalable Distributed Vector Database for High-Throughput Approximate Nearest Neighbor Search
Qian Xu 0021, Feng Zhang 0007, Chengxi Li 0022, Lei Cao 0004, Zheng Chen 0023, Jidong Zhai, Xiaoyong Du 0001 |
Proc. ACM Manag. Data | 6 |
| 2025 | Breaking the Edge: Enabling Efficient Neural Network Inference on Integrated Edge Devices
Feng Zhang 0007, Chenyang Zhang 0005, Jiawei Guan, Qiangjun Zhou, Kuangyu Chen, Xiao Zhang 0001, Bingsheng He, Jidong Zhai, Xiaoyong Du 0001 |
IEEE Trans. Cloud Comput. | 8 |
| 2025 | Leveraging Graph Analysis to Pinpoint Root Causes of Scalability Issues for Parallel ApplicationsabstractIt is challenging to scale parallel applications to modern supercomputers because of load imbalance, resource contention, and communications between processes. Profiling and tracing are two main performance analysis approaches for detecting these scalability bottlenecks. Profiling is low-cost but lacks detailed dependence for identifying root causes. Tracing records plentiful information but incurs significant overheads. To address these issues, we presentScalAna, which employs static analysis techniques to combine the benefits of profiling and tracing - it enables tracing's analyzability with overhead similar to profiling.ScalAnauses static analysis to capture program structures and data dependence of parallel applications, and leverages lightweight profiling approaches to record performance data during runtime. Then a parallel performance graph is generated with both static and dynamic data. Based on this graph, we design a backtracking detection approach to automatically pinpoint the root causes of scaling issues. We evaluate the efficacy and efficiency ofScalAnausing several real applications with up to 704K lines of code and demonstrate that our approach can effectively pinpoint the root causes of scaling loss with an average overhead of 5.65% for up to 16,384 processes. By fixing the root causes detected by our tool, it achieves up to 33.01% performance improvement. Yuyang Jin 0001, Haojie Wang 0004, Xiongchao Tang, Zhenhua Guo 0003, Yaqian Zhao, Torsten Hoefler, Tao Liu 0029, Xu Liu 0001, Jidong Zhai |
IEEE Trans. Parallel Distributed Syst. | 9 |
| 2025 | A Reinforcement Learning Based Framework for Holistic Energy Optimization of Sustainable Cloud Data CentersabstractThe widespread adoption of cloud data centers has led to a rise in energy consumption, with the associated carbon emissions posing a further threat to the environment. Cloud providers are increasingly moving towards sustainable data centers powered by renewable energy sources (RES). The existing approaches fail to efficiently coordinate IT and cooling resources in such data centers due to the intermittent nature of RES and the complexity of state and action spaces among different devices, resulting in poor holistic energy efficiency. In this paper, a reinforcement learning (RL) based framework is proposed to optimize the holistic energy consumption of sustainable cloud data centers. First, a joint prediction method MTL-LSTM is developed to accurately evaluate both energy consumption and thermal status of each physical machine (PM) under different optimization scenarios to improve the state space information of the RL algorithm. Then, this framework designs a novel energy-aware approach named BayesDDQN, which leverages Bayesian optimization to synchronize the adjustments of VM migration and cooling parameter within the hybrid action space of the Double Deep Q-Network (DDQN) for achieving the holistic energy optimization. Moverover, the pre-cooling technology is integrated to further alleviate hotspot by making full use of RES. Experimental results demonstrate that the proposed RL-based framework achieves an average reduction of 2.83% in holistic energy consumption and 4.74% in brown energy, which also reduces cooling energy consumption by 13.48% with minimal occurrences of hotspots. Furthermore, the proposed MTL-LSTM method reduces the root mean square error (RMSE) of energy consumption and inlet temperature predictions by nearly half compared to LSTM and XGBoost. Da-Ming Zhao, Jiantao Zhou 0002, Jidong Zhai, Keqin Li 0001 |
IEEE Trans. Serv. Comput. | 3 |
| 2024 | Optimal Kernel Orchestration for Tensor Programs with KorchabstractKernel orchestration is the task of mapping the computation defined in different operators of a deep neural network (DNN) to the execution of GPU kernels on modern hardware platforms. Prior approaches optimize kernel orchestration by greedily applying operator fusion, which fuses the computation of multiple operators into a single kernel, and miss a variety of optimization opportunities in kernel orchestration. Muyan Hu, Ashwin Venkatram, Shreyashri Biswas, Balamurugan Marimuthu, Bohan Hou, Gabriele Oliaro, Haojie Wang 0004, Liyan Zheng 0001, Xupeng Miao, Jidong Zhai |
ASPLOS (3) | 10 |
| 2024 | WiseGraph: Optimizing GNN with Joint Workload Partition of Graph and OperationsabstractGraph Neural Network (GNN) has emerged as an important workload for learning on graphs. With the size of graph data and the complexity of GNN model architectures increasing, developing an efficient GNN system grows more important. As GNN has heavy neural computation workloads on a large graph, it is crucial to partition the entire workload into smaller parts for parallel execution and optimization. However, existing approaches separately partition graph data and GNN operations, resulting in inefficiency and large data movement overhead. Kezhao Huang, Jidong Zhai, Liyan Zheng 0001, Haojie Wang 0004, Yuyang Jin 0001, Qihao Zhang, Runqing Zhang, Zhen Zheng, Youngmin Yi, Xipeng Shen |
EuroSys | 2 |
| 2024 | F-TADOC: FPGA-Based Text Analytics Directly on Compression with HLSabstractWith the development of loT and edge computing, data analytics on edge has become popular, and text analytics directly on compression (TADOC) has been proven to be a promising technology for edge data analytics. At the same time, Field Programmable Gate Array (FPGA) also has broad application prospects in data analytics systems. Unfortunately, there is no work to date showing how to support TADOC using FPGAs. We propose FPGA-based text analytics directly on compression with HLS, namely F - TADOC, which is the first framework using HLS to provide FPGA-based text analytics directly on compressed data. It effectively supports efficient text analytics on FPGA without decompressing input data. F-TADOC addresses three major challenges. First, TADOC involves a large number of dependencies with unbalanced workload of rules, which causes extremely low pipeline efficiency on FPG As. To solve it, we use layer-wise approach to traverse the DAG composed of rules and allocate different pipeline processing strategies for rules of different sizes. Second, the data volume required can be large that beyond the on-chip memory capacity of FPGAs. We develop a memory pool supporting hash structure and on-chip caches on FPGA to deal with this challenge. Third, when traversing the DAG, there are massive indirect addressing with a large number of random accesses. This leads to redundant time overhead caused by the latency in accessing the High Bandwidth Memory (HBM) during the pipeline. We optimize the F - TADOC algorithm by using dataflow to expand the nested loop, thus eliminate indirect addressing. With four widely used datasets, experiments show that F - TADOC achieves 4.63 x and 1.49 x performance speedup over TADOC and G- TADOC. Yanliang Zhou, Feng Zhang 0007, Tuo Lin, Yuanjie Huang, Saiqin Long, Jidong Zhai, Xiaoyong Du 0001 |
ICDE | 6 |
| 2024 | POSTER: Pattern-Aware Sparse Communication for Scalable Recommendation Model TrainingabstractRecommendation models are an important category of deep learning models whose size is growing enormous. They consist of a sparse part with TBs of memory footprint and a dense part that demands PFLOPs of computing capability to train. Unfortunately, the high sparse communication cost to re-organize data for different parallel strategies of the two parts impedes the scalability in training. Jiaao He, Shengqi Chen 0001, Jidong Zhai |
PPoPP | 3 |
| 2024 | MixQ: Taming Dynamic Outliers in Mixed-Precision Quantization by Online PredictionabstractMixed-precision quantization has shown to be a promising method for enhancing the efficiency of LLMs. This technique boosts computational efficiency by processing most values with low-precision, high-throughput compute units and maintains accuracy by processing outliers in high-precision. However, due to the dynamic, irregular, and sparse nature of outliers, this approach is far from using hardware efficiently. In this work, we propose MixQ, an efficient mixed-precision quantization system. Through our in-depth analysis of outlier distribution, we introduce a locality-based outlier prediction algorithm that can predict all outliers of 95.8% of tokens. Based on this accurate prediction, we propose a quantization ahead of detection (QAD) technique that can verify the correctness of prediction. A new data structure is proposed for efficient outlier processing. Evaluation shows that MixQ achieves $1.52 \times$ and $1.78 \times$ speedup over FP16 and Bitsandbytes on 8-bit quantization; plus $1.48 \times 1.93 \times$ and $6 \times$ speedup over QUIK, FP16, and AWQ on 4-bit quantization.11Our code is available on:https://github.com/Qcompiler/MIXQ Yidong Chen 0003, Chen Zhang 0001, Rongchao Dong, Zhonghua Lu, Jidong Zhai |
SC | 7 |
| 2024 | PUZZLE: Efficiently Aligning Large Language Models through Light-Weight Context Switch
Kinman Lei, Yuyang Jin 0001, Mingshu Zhai, Kezhao Huang, Haoxing Ye, Jidong Zhai |
USENIX ATC | 6 |
| 2024 | MAGPY: Compiling Eager Mode DNN Programs by Monitoring Execution States
Chen Zhang 0001, Rongchao Dong, Haojie Wang 0004, Runxin Zhong, Jike Chen, Jidong Zhai |
USENIX ATC | 6 |
| 2024 | Editorial for the special issue on programming models and system software for High-Performance Computing (HPC) environments
Jianbin Fang, Jidong Zhai |
CCF Trans. High Perform. Comput. | 2 |
| 2024 | FreshGNN: Reducing Memory Access via Stable Historical Embeddings for Graph Neural Network TrainingabstractA key performance bottleneck when training graph neural network (GNN) models on large, real-world graphs is loading node features onto a GPU. Due to limited GPU memory, expensive data movement is necessary to facilitate the storage of these features on alternative devices with slower access (e.g. CPU memory). Moreover, the irregularity of graph structures contributes to poor data locality which further exacerbates the problem. Consequently, existing frameworks capable of efficiently training large GNN models usually incur a significant accuracy degradation because of the currently-available shortcuts involved. To address these limitations, we instead propose FreshGNN, a general-purpose GNN mini-batch training framework that leverages a historical cache for storing and reusing GNN node embeddings instead of re-computing them through fetching raw features at every iteration. Critical to its success, the corresponding cache policy is designed, using a combination of gradient-based and staleness criteria, to selectively screen those embeddings which are relatively stable and can be cached, from those that need to be re-computed to reduce estimation errors and subsequent downstream accuracy loss. When paired with complementary system enhancements to support this selective historical cache, FreshGNN is able to accelerate the training speed on large graph datasets such as ogbn-papers100M and MAG240M by 3.4× up to 20.5× and reduce the memory access by 59%, with less than 1% influence on test accuracy. Kezhao Huang, Haitian Jiang, Guangxuan Xiao, David P. Wipf, Xiang Song 0003, Zengfeng Huang, Jidong Zhai, Zheng Zhang 0001 |
Proc. VLDB Endow. | 9 |
| 2024 | Compressed Data Direct Computing for DatabasesabstractDirectly performing operations on compressed data has been proven to be a big success facing Big Data problems in modern data management systems. These systems have demonstrated significant compression benefits and performance improvement for data analytics applications. However, current systems only focus on data queries, while a complete Big Data system must support both data query and data manipulation. To solve this problem, we develop CompressDB, which is a new storage engine that can support data processing for databases without decompression. CompressDB has the following advantages. First, CompressDB utilizes context-free grammar to compress data, and supports both data query and data manipulation. Second, for adaptability, we integrate CompressDB to file systems so that a wide range of databases can directly use CompressDB without any change. Third, we enable operation pushdown to storage so that we can perform data query and manipulation in storage systems without bringing large data to memory for high efficiency. We validate the efficacy of CompressDB supporting various kinds of database systems, including SQLite, MySQL, LevelDB, MongoDB, ClickHouse, and Neo4j. We evaluate our method using seven real-world datasets with various lengths, structures, and content in both single node and cluster environments. Experiments show that CompressDB achieves 40% throughput improvement and 44% latency reduction, along with 1.75 compression ratio on average. Weitao Wan, Feng Zhang 0007, Chenyang Zhang 0005, Mingde Zhang, Jidong Zhai, Yunpeng Chai, Huanchen Zhang, Wei Lu 0015, Yuxing Chen 0003, Haixiang Li, Anqun Pan, Xiaoyong Du 0001 |
IEEE Trans. Knowl. Data Eng. | 5 |
| 2024 | A Survey on Performance Modeling and Prediction for Distributed DNN TrainingabstractThe recent breakthroughs in large-scale DNN attract significant attention from both academia and industry toward distributed DNN training techniques. Due to the time-consuming and expensive execution process of large-scale distributed DNN training, it is crucial to model and predict the performance of distributed DNN training before its actual deployment, in order to optimize the design of distributed DNN training at low cost. This paper analyzes and emphasizes the importance of modeling and predicting the performance of distributed DNN training, categorizes and analyses the related state-of-the-art works, and discusses future challenges and opportunities for this research field. The objectives of this paper are twofold: first, to assist researchers in understanding and choosing suitable modeling and prediction tools for large-scale distributed DNN training, and second, to encourage researchers to propose more valuable research about performance modeling and prediction for distributed DNN training in the future. Zhenhua Guo 0003, Yinan Tang, Jidong Zhai, Tongtong Yuan, Li Wang 0040, Yaqian Zhao, RenGang Li |
IEEE Trans. Parallel Distributed Syst. | 3 |
| 2024 | Graph-Centric Performance Analysis for Large-Scale Parallel ApplicationsabstractPerformance analysis is essential for understanding the performance behaviors of parallel programs and detecting performance bottlenecks. Whereas, complex interconnections across several types of performance bugs, as well as inter-process communications and data dependence, make efficient performance analysis even more difficult. Despite the fact that many performance tools have been developed, accurately identifying underlying performance bottlenecks for such complex scenarios requires specific in-depth analysis. Significant human efforts and analysis knowledge are often required to implement each specific analytic task. To alleviate the complexity of developing specific performance analytic tasks, we present a programmable performance analysis tool, calledPerFlow. InPerFlow, a step-by-step performance analysis process is represented as an Analysis Flow Diagram, which is constructed with several performance analysis sub-tasks, namely passes, that can be defined by developers or provided byPerFlow's built-in analysis pass library. Furthermore, we define a Performance Abstraction Graph to describe the performance behavior of a parallel program, where the edges indicate the interactions between parallel units, therefore the analytic sub-tasks are converted to graph analysis tasks.PerFlowprovides plentiful Python APIs for developing analytic tasks. Several case studies of real-world applications with up to 700 K lines of code are used to demonstrate the effectiveness ofPerFlow. The results indicate thatPerFlowmakes it much easier to implement specific performance analytic tasks, and these tasks are performed automatically and efficiently to detect underlying performance bottlenecks. Yuyang Jin 0001, Haojie Wang 0004, Runxin Zhong, Chen Zhang 0001, Xia Liao, Feng Zhang 0007, Jidong Zhai |
IEEE Trans. Parallel Distributed Syst. | 7 |
| 2024 | Efficient Inference for Pruned CNN Models on Mobile Devices With Holistic Sparsity AlignmentabstractMany artificial intelligence applications based on convolutional neural networks are directly deployed on mobile devices to avoid network unavailability and user privacy leakage. However, the significant increase in model parameter volumes makes it difficult to achieve high-performance convolutional neural network inference on these mobile devices with limited computing power. Weight pruning is one of the main approaches to compress models by reducing model parameters and computational operations, which also introduces irregular sparsity of neural networks, leading to inefficient computation and memory access during inference. This work proposes an end-to-end framework, namely MCPruner, for efficient inference of pruned convolutional neural networks on mobile devices by aligning the sparse patterns with hardware execution features in computation, memory access, and parallelism. It first co-designs pruning methods and code generation optimizations for the alignment of non-zero weight count and vector width, to improve computational efficiency while ensuring accuracy. During the code generation, it applies a sparse pattern-aware format to reduce inefficient memory accesses. Besides, convolution computations are reordered for alignment, and then mapped to parallel threads on accelerated units to achieve high parallelism. Experimental results using several commonly used models and datasets on the ARM-based Hikey970 demonstrate that our work outperforms state-of-the-art methods in inference efficiency, with no accuracy degradation. Yuyang Jin 0001, Runxin Zhong, Saiqin Long, Jidong Zhai |
IEEE Trans. Parallel Distributed Syst. | 4 |
| 2024 | G-Learned Index: Enabling Efficient Learned Index on GPUabstractAI and GPU technologies have been widely applied to solve big data problems. The total data volume worldwide reaches 200 zettabytes in 2022. How to efficiently index the required content among massive data becomes serious. Recently, a promising learned index has been proposed to address this challenge: It has extremely high efficiency while retaining marginal space overhead. However, we notice that previous learned indexes have mainly focused on CPU architecture, while ignoring the advantages of GPU. Because traditional indexes like B-Tree, LSM, and bitmap have greatly benefited from GPU acceleration, a combination of a learned index and GPU has great potentials to reach tremendous speedups. In this paper, we propose a GPU-based learned index, called G-Learned Index, to significantly improve the performance of learned index structures. The primary challenges in developing G-Learned Index lie in the use of thousands of GPU cores including minimization of synchronization and branch divergence, data structure design for parallel operations, and usage of memory bandwidth including limited memory transactions and multi-memory hierarchy. To overcome these challenges, a series of novel technologies are developed, including efficient thread organization, succinct data structures, and heterogeneous memory hierarchy utilization. Compared to the state-of-the-art learned index, the proposed G-Learned Index achieves an average of 174× speedup (and 107× of its parallel version). Meanwhile, we attain 2× less query time over the state-of-the-art GPU B-Tree. Our further exploration of range queries shows that G-Learned Index is 17× faster than CPU multi-dimensional learned index. We have made G-Learned Index available athttps://anonymous.4open.science/r/G-Learned-Index-8D89. Jiesong Liu, Feng Zhang 0007, Lv Lu, Chang Qi, Xiaoguang Guo, Dong Deng 0001, Guoliang Li 0001, Huanchen Zhang, Jidong Zhai, Hechen Zhang, Yuxing Chen 0003, Anqun Pan, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 9 |
| 2024 | Optimizing I/O Performance Through Effective vCPU Scheduling Interference ManagementabstractVirtual machines (VMs) heavily rely on virtual CPUs (vCPUs) scheduling to achieve efficient I/O performance. The vCPU scheduling interference can cause inconsistent scheduling latency and degraded I/O performance, potentially compromising the services provided by affected VMs. Existing solutions have limitations, such as inefficiency in diagnosing interference issues or imposing undesired side effects on cloud systems. To address these challenges, we present Otter, a holistic technique for optimizing I/O performance in the presence of vCPU scheduling interference. Otter employs innovative methods to enhance interference diagnosis efficiency. First, we propose lightweight methods to measure the dynamic changes in scheduling latencies for co-running vCPUs, ensuring both flexibility and accuracy. Second, we propose fine-grained quantification methods to timely determine the interference, with low false positive and false negative rates. Third, we identify interference patterns that aid in analyzing the root causes of interference and preventing similar issues from recurring. Otter has been operational for one year in the production cloud at the National Supercomputing Center (Wuxi). It diagnoses and helps fix more than 470 vCPU scheduling interference-related issues, resulting in a 19.6% improvement in cloud service I/O performance with negligible overhead in production. Jinzhe Yang, Jidong Zhai, Guangwen Yang 0002 |
IEEE Trans. Parallel Distributed Syst. | 3 |
| 2023 | Unveiling the Black Box of PLMs with Semantic Anchors: Towards Interpretable Neural Semantic ParsingabstractThe recent prevalence of pretrained language models (PLMs) has dramatically shifted the paradigm of semantic parsing, where the mapping from natural language utterances to structured logical forms is now formulated as a Seq2Seq task. Despite the promising performance, previous PLM-based approaches often suffer from hallucination problems due to their negligence of the structural information contained in the sentence, which essentially constitutes the key semantics of the logical forms. Furthermore, most works treat PLM as a black box in which the generation process of the target logical form is hidden beneath the decoder modules, which greatly hinders the model's intrinsic interpretability. To address these two issues, we propose to incorporate the current PLMs with a hierarchical decoder network. By taking the first-principle structures as the semantic anchors, we propose two novel intermediate supervision tasks, namely Semantic Anchor Extraction and Semantic Anchor Alignment, for training the hierarchical decoders and probing the model intermediate representations in a self-adaptive manner alongside the fine-tuning process. We conduct intensive experiments on several semantic parsing benchmarks and demonstrate that our approach can consistently outperform the baselines. More importantly, by analyzing the intermediate representations of the hierarchical decoders, our approach also makes a huge step toward the interpretability of PLMs in the domain of semantic parsing. Lunyiu Nie, Jiuding Sun, Yanlin Wang 0001, Lun Du, Shi Han, Dongmei Zhang 0001, Lei Hou 0001, Juan-Zi Li, Jidong Zhai |
AAAI | 9 |
| 2023 | Joint Geometrical and Statistical Domain Adaptation for Cross-domain Code Vulnerability DetectionabstractIn code vulnerability detection tasks, a detector trained on a label-rich source domain fails to provide accurate prediction on new or unseen target domains due to the lack of labeled training data on target domains.Previous studies mainly utilize domain adaptation to perform cross-domain vulnerability detection.But they ignore the negative effect of private semantic characteristics of the target domain for domain alignment, which easily causes the problem of negative transfer.In addition, these methods forcibly reduce the distribution discrepancy between domains and do not take into account the interference of irrelevant target instances for distributional domain alignment, which leads to the problem of excessive alignment.To address the above issues, we propose a novel cross-domain code vulnerability detection framework named MN-CRI.Specifically, we introduce mutual nearest neighbor contrastive learning to align the source domain and target domain geometrically, which could align the common semantic characteristics of two domains and separate out the private semantic characteristics of each domain.Furthermore, we introduce an instance re-weighting scheme to alleviate the problem of excessive alignment.This scheme dynamically assign different weights to instances, reducing the contribution of irrelevant instances so as to achieve better domain alignment.Finally, extensive experiments demonstrate that MNCRI significantly outperforms state-of-the-art cross-domain code vulnerability detection methods by a large margin. Qianjin Du, Shiji Zhou, Xiaohui Kuang, Jidong Zhai |
EMNLP | 5 |
| 2023 | GLM-130B: An Open Bilingual Pre-trained Model
Aohan Zeng, Xiao Liu 0036, Zhengxiao Du, Hanyu Lai, Ming Ding 0004, Zhuoyi Yang, Yifan Xu 0014, Wendi Zheng, Weng Lam Tam, Zixuan Ma, Jidong Zhai, Zhiyuan Liu 0001, Peng Zhang 0077, Yuxiao Dong, Jie Tang 0001 |
ICLR | 14 |
| 2023 | Cocktailer: Analyzing and Optimizing Dynamic Control Flow in Deep Learning
Chen Zhang 0001, Lingxiao Ma, Jilong Xue, Yining Shi 0001, Ziming Miao, Fan Yang 0024, Jidong Zhai, Zhi Yang 0001, Mao Yang 0004 |
OSDI | 7 |
| 2023 | EINNET: Optimizing Tensor Programs with Derivation-Based Transformations
Liyan Zheng 0001, Haojie Wang 0004, Jidong Zhai, Muyan Hu, Zixuan Ma, Tuowei Wang, Shuhong Huang, Xupeng Miao, Shizhi Tang, Kezhao Huang |
OSDI | 3 |
| 2023 | GraphSet: High Performance Graph Mining through Equivalent Set TransformationsabstractGraph mining is of critical use in a number of fields such as social networks, knowledge graphs, and fraud detection. As an NP-complete problem, accelerating computation performance is the main target for current optimizations. Due to excellent performance, state-of-the-art graph mining systems mainly rely on pattern-aware algorithms. Despite previous efforts, complex control flows introduced by pattern-aware algorithms bring significant overhead and also impede further acceleration on heterogeneous hardware. Tianhui Shi, Jidong Zhai, Haojie Wang 0004, Qiqian Chen, Mingshu Zhai, Zixu Hao |
SC | 2 |
| 2023 | SmartMoE: Efficiently Training Sparsely-Activated Models through Combining Offline and Online Parallelization
Mingshu Zhai, Jiaao He, Zixuan Ma, Zan Zong, Runqing Zhang, Jidong Zhai |
USENIX ATC | 6 |
| 2023 | Unified Programming Models for Heterogeneous High-Performance Computers
Zixuan Ma, Yuyang Jin 0001, Shizhi Tang, Haojie Wang 0004, Wei-Cheng Xue, Jidong Zhai |
J. Comput. Sci. Technol. | 6 |
| 2023 | CompressGraph: Efficient Parallel Graph Analytics with Rule-Based CompressionabstractModern graphs exert colossal time and space pressure on graph analytics applications. In 2022, Facebook social graph reaches 2.91 billion users with trillions of edges. Many compression algorithms have been developed to support direct processing on compressed graphs to address this challenge. However, previous graph compression algorithms do not focus on leveraging redundancy in repeated neighbor sequences, so they do not save the amount of computation for graph analytics. We develop CompressGraph, an efficient rule-based graph analytics engine that leverages data redundancy in graphs to achieve both performance boost and space reduction for common graph applications. CompressGraph has three advantages over previous works. First, the rule-based abstraction of CompressGraph supports the reuse of intermediate results during graph traversal, thus saving time. Second, CompressGraph has intense expressiveness to support a wide range of graph applications. Third, CompressGraph scales well under high parallelism because the context-free rules have few dependencies. Experiments show that CompressGraph provides significant performance and space benefits on both CPUs and GPUs. On evaluating six typical graph applications, CompressGraph can achieve 1.97× speedup on the CPU, while 3.95× speedup on the GPU, compared to the state-of-the-art CPU and GPU methods, respectively. Moreover, CompressGraph can save an average of 71.27% memory savings on CPU and 70.36 on GPU. Zheng Chen 0023, Feng Zhang 0007, Jiawei Guan, Jidong Zhai, Xipeng Shen, Huanchen Zhang, Wentong Shu, Xiaoyong Du 0001 |
Proc. ACM Manag. Data | 4 |
| 2023 | BladeDISC: Optimizing Dynamic Shape Machine Learning Workloads via Compiler ApproachabstractCompiler optimization plays an increasingly important role to boost the performance of machine learning models for data processing and management. With increasingly complex data, the dynamic tensor shape phenomenon emerges for ML models. However, existing ML compilers either can only handle static shape models or expose a series of performance problems for both operator fusion optimization and code generation in dynamic shape scenes. This paper tackles the main challenges of dynamic shape optimization: the fusion optimization without shape value, and code generation supporting arbitrary shapes. To tackle the fundamental challenge of the absence of shape values, it systematically abstracts and excavates the shape information and designs a cross-level symbolic shape representation. With the insight that what fusion optimization relies upon is tensor shape relationships between adjacent operators rather than exact shape values, it proposes the dynamic shape fusion approach based on shape information propagation. To generate code that adapts to arbitrary shapes efficiently, it proposes a compile-time and runtime combined code generation approach. Finally, it presents a complete optimization pipeline for dynamic shape models and implements an industrial-grade ML compiler, named BladeDISC. The extensive evaluation demonstrates that BladeDISC outperforms PyTorch, TorchScript, TVM, ONNX Runtime, XLA, Torch Inductor (dynamic shape), and TensorRT by up to 6.95×, 6.25×, 4.08×, 2.04×, 2.06×, 7.92×, and 4.16× (3.54×, 3.12×, 1.95×, 1.47×, 1.24×, 2.93×, and 1.46× on average) in terms of end-to-end inference speedup on the A10 and T4 GPU, respectively. BladeDISC's source code is publicly available at https://github.com/alibaba/BladeDISC. Zhen Zheng, Zaifeng Pan, Dalin Wang, Kai Zhu 0004, Wenyi Zhao, Tianyou Guo, Xiafei Qiu, Minmin Sun, Feng Zhang 0007, Xiaoyong Du 0001, Jidong Zhai, Wei Lin 0016 |
Proc. ACM Manag. Data | 12 |
| 2023 | Special issue on new trends in high-performance computing: Software systems and applicationsabstractHigh-performance computing (HPC) offers the computing power to continuously support the world's most important discoveries in various scientific and business domains such as chemistry, physics, biology, material science, drug discovery, and financial investment risk analysis. We are now in the exascale era with the Frontier exascale system that has been very recently revealed (June 2022). Researchers from across the HPC community have been developing software systems, tools, libraries, frameworks, application packages, and methods that can fully exploit these extremely powerful computing resources. Such extreme-scale computing will enable the solution of vastly more accurate predictive models and the analysis of massive quantities of data, producing quantum advances in areas of science and technology that are essential to the scientific community. New computational approaches such as machine learning/deep learning have also been heavily explored in recent years and have shown promising evidence for many problems that cannot be resolved by traditional computational simulation and engineering. Training deep neural networks with massive data is an extremely computing-intensive task that heavily relies on HPC power. The upcoming exascale computing era will be the essential basis for supporting new innovations in machine learning/deep learning-based exploration and will lead to new sciences in directions such as smart manufacturing, laboratory automation, and automatic programming. While the hardware architecture can generate extreme computing power, renovation in the software stack plays an essential role in effective performance delivery. The large supercomputers continue to move into the heterogeneous space, while the former fastest ARM-based system, Fugaku, and the many core Sunway Taihulight types of systems are marching towards the heterogeneous space. With systems equipped with GPUs, Advance RISC Machine (ARM) Support Vector Engine (SVEs), and many cores, there is a dire need for innovative software frameworks that can seamlessly migrate scientific code to these systems equipped with rich computing resources. We need innovation at different levels, including compiler tools and techniques, performance analysis tools, novel abstractions of the programming model, redesign of application-level algorithms, and so on. Furthermore, co-design of applications and low-level software frameworks can lead to more efficient use of the opportunities of exascale in many contexts. This special issue has selected 14 papers. We next present the summary of the papers presented in this special issue. The first paper titled “ParTransgrid: A scalable parallel pre-processing tool for unstructured-grid cell-centered Computational Fluid Dynamics (CFD) applications” by Jianqiang et al.1 proposes a parallel pre-processing tool, called ParTransgrid, that translates the general grid format such as the CFD General Notation System into an efficient distributed mesh data format for large-scale parallel computing. Experiment results reveal that ParTransgrid can be easily scaled to billion-level grid CFD applications and that the preparation time for parallel computing with hundreds of thousands of cores is reduced to a few minutes. The second paper titled “Generation of logic designs for efficiently solving ordinary differential equations on FPGAs” by Korch et al.2 proposes a framework that is able to automatically generate specific and optimized solver logic from easy-to-handle configuration files. No manual development and no special Field Programmable Gate Array (FPGA) or programming knowledge are required. The logic generated by this improved approach is up to 43 times faster than its hand-optimized High Level Synthesis (HLS) counterpart, depending on the solution method. The third paper titled “NAS Parallel Benchmarks with Compute Unified Device Architecture (CUDA) and Beyond” by Fernandes et al.3 provides a new CUDA implementation for NASA Parallel Benchmark (NPB). The performance results have shown up to 267% improvements over the best benchmark versions available. The authors also observe the best and worst design choices concerning code size and the performance tradeoff. Lastly, the authors highlight the challenges of implementing parallel CFD applications for Graphic Procesing Unit (GPUs) and how the computations impact the GPU's behavior. The fourth paper titled, “Using Ginkgo's Memory Accessor for Improving the Accuracy of Memory-Bound Low Precision Basic Linear Algebra Subprograms (BLAS)” by Quintana-Ortí et al.4 demonstrates that memory-bound applications operating on low precision data can increase their accuracy by relying on the memory accessor to perform all arithmetic operations in high precision. In particular, the authors demonstrate that memory-bound BLAS operations (including the sparse matrix-vector product) can be re-engineered with the memory accessor and that the resulting accessor-enabled BLAS routines achieve lower rounding errors while delivering the same performance as the fast low-precision BLAS. The fifth paper titled “Three Practical Workflow Schedulers for Easy Maximum Parallelism” by Rogers5 presents a complete characterization of the minimum effective task granularity for efficient scheduler usage scenarios. A separate job scheduler is implemented for three distinct workflow patterns involved in the preparation, execution, and analysis of computational chemistry simulations. It shows unique benefits, including simplicity of design, suitability for HPC centers, short startup time, and well-understood per-task overhead. All three new tools have been shown to scale to full utilization of Summit and have been made publicly available with tests and documentation. The sixth paper titled “LLAMA: The Low-Level Abstraction for Memory Access” by Bussmann et al.6 presents the Low-Level Abstraction of Memory Access (LLAMA), a C++ library that provides such a data structure abstraction layer with example implementations for multidimensional arrays of nested, structured data. LLAMA provides fully C++-compliant methods for defining and switching custom memory layouts for user-defined data types. The library is extensible with third-party allocators. LLAMA provides a novel tool for the development of high-performance C++ applications in a heterogeneous environment. The seventh paper titled “PAS: A new powerful and simple quantum computing simulator” by Wang et al.7 proposes a new powerful and simple CPU-based quantum computing simulator: PAS (Power And Simple). Compared with existing simulators, PAS introduces four novel optimization methods: efficient hybrid vectorization, fast bitwise operation, memory access filtering, and quantum tracking. Experiments were performed on the Intel Xeon E5-2670 v3 CPU and showed that PAS compared with the state-of-the-art simulator QuEST can achieve a mean speedup of 8.69x and 2.62x for the Quantum Field Theory (QFT) and Relativistic Quantum Chemistry (RQC) benchmarks, respectively. The eighth paper titled “Dynamics Signature based Anomaly Detection” by Bader et al.8 borrows the dynamics metrics and proposes the concept of Dynamics Signature (DS) in multi-dimensional feature space to efficiently distinguish the abnormal event from the normal behaviors of a variable star. Two datasets, parameterized sinusoidal dataset containing 262,440 light curves, and a real variable star-based dataset containing 462,996 light curves are used to evaluate the practical performance of the proposed DS algorithm. Experimental results show that their DS algorithm is highly accurate, sensitive to detecting weak microlensing events at very early stages, and fast enough to process 176,000 stars in less than 1 second on a commodity computer. The ninth paper titled “EESSI: A Cross-Platform Ready-To-Use Optimised Scientific Software Stack” by Röblitz et al.9 proposes the European Environment for Scientific Software Installations project that aims to provide a ready-to-use stack of scientific software installations that can be leveraged easily on a variety of platforms, ranging from personal workstations to cloud environments and supercomputer infrastructure, without making compromises with respect to performance. The authors provide a detailed overview of the project, highlight potential use cases, and demonstrate that the performance of the provided scientific software installations can be competitive with system-specific installations. The eleventh paper titled “A large scale parallel fluid-structure interaction computing platform for simulating structural responses to a detonation shock” by Yang et al.10 presents a partitioned fluid-structure interaction computing platform designed for parallel simulating structural responses to a detonation shock. The 3D numerical result of structural responses to a detonation shock is presented and analyzed. On 256 processor cores, the speedup ratio of the simulations for a detonation shock reaches 178.0 with 5.1 million mesh cells and the parallel efficiency achieves 69.5%. The results demonstrate the good potential of massively parallel simulations. Overall, a general-purpose fluid-structure interaction software platform with detonation support is proposed by integrating open source codes. The authors express their sincere gratitude and thanks to the Editor-in-Chief Dr. Rajkumar Buyya, for guiding them to organize this special issue. The authors appreciate the support from the editorial office. The authors are also thankful to all the authors who submitted their ideas to this special issue and to the reviewers for their thoughtful and critical suggestions to improve the quality of the submitted papers. Sunita Chandrasekaran, Min Si, Jidong Zhai, Lena Oden |
Softw. Pract. Exp. | 3 |
| 2023 | Optimizing DNNs With Partially Equivalent Transformations and Automated CorrectionsabstractDeep neural network (DNN) applications are typically represented by tensor programs. To boost the performance of DNN computations, existing works adopt fully equivalent transformations for tensor program optimization by guaranteeing the equivalence on each element of tensors. However, as there are thousands of elements in a tensor, such optimization misses the opportunities that allow the in-equivalence of minority elements. In this work, we proposePet, the first work that introduces partially equivalent transformations to optimize tensor programs. To maintain the functional equivalence of tensor programs,Petautomatically finds and corrects the in-equivalent positions by leveraging the multi-linearity of DNN computations.Petfurther uses a mutation manager to improve search efficiency. Evaluation results show thatPetcan achieve up to 1.98$\times$and 2.20$\times$speedups on NVIDIA Tesla A100 and V100 respectively compared with existing DNN frameworks by introducing new optimization opportunities of partially equivalent transformations. Haojie Wang 0004, Jidong Zhai, Mingyu Gao 0001, Feng Zhang 0007, Tuowei Wang, Zixuan Ma, Shizhi Tang, Liyan Zheng 0001, Kaiyuan Rong, Yuanyong Chen |
IEEE Trans. Computers | 2 |
| 2023 | Critique of "A Parallel Framework for Constraint-Based Bayesian Network Learning via Markov Blanket Discovery" by SCC Team From Tsinghua UniversityabstractSrivastava et al. propose a parallel framework to optimize Bayesian network learning in the SC20 article entitled “A Parallel Framework for Constraint-Based Bayesian Network Learning via Markov Blanket Discovery”. They parallelize all the phases in network constructing algorithms to achieve high performance and scalability. In this article, we reproduce the strong scaling and weak scaling experiments in that SC article. We conduct experiments on a 4-node cluster with Intel CPUs provided by the SCC committee. We further analyze the results of communication overhead. Our results show that the proposed method in that SC article scales well on the provided cluster, in accordance with the SC article.Author: Please confirm or add details for any funding or financial support for the research of this article. ?> Juncheng Cao, Kaiyuan Rong, Mingshu Zhai, Yanyu Ren, Yuxi Zhu, Jidong Zhai |
IEEE Trans. Parallel Distributed Syst. | 8 |
| 2023 | Enabling Efficient Random Access to Hierarchically Compressed Text Data on Diverse GPU PlatformsabstractThe tremendous computing capacity of GPU offers significant potential in processing hierarchically compressed text data without decompression. However, current GPU techniques offer only traversal-based text data analytics; random access is exceedingly inefficient, limiting their utility significantly. To address this issue, we develop a novel and widely applicable solution that prompts random access to hierarchically compressed text data without decompression in GPU memory. We address three main challenges for enabling efficient random access to compressed text data on GPUs. The first challenge is designing GPU data structures that facilitate random access. The second challenge is efficiently generating data structures on GPU. The CPU is inefficient when generating data structures for random access, and this inefficiency increases considerably when PCIe transmission is incorporated. The third challenge is query processing on compressed text data in GPU memory. Random accesses, such as data updates, cause massive conflicts among countless threads. In order to address the first challenge, we develop several compressed GPU data structures, including indexing within the intricate GPU memory hierarchy. To handle the second challenge, we propose a two-phase process for producing these data structures on GPU. For the third challenge, a double-parsing design is proposed as a solution to avoid conflicts. We evaluate our solution on three platforms, two server-grade GPU platforms and one edge-grade GPU platform, using five real-world datasets. Experimental results show that random access operations on GPU achieve an average speedup of 52.98× compared to the state-of-the-art solution. Yihua Hu 0003, Feng Zhang 0007, Zhiming Yao, Letian Zeng, Haipeng Ding 0002, Zhewei Wei, Xiao Zhang 0001, Jidong Zhai, Xiaoyong Du 0001, Siqi Ma 0001 |
IEEE Trans. Parallel Distributed Syst. | 9 |
| 2022 | Suppressing ZZ crosstalk of Quantum computers through pulse and scheduling co-optimizationabstractNoise is a significant obstacle to quantum computing, and ZZ crosstalk is one of the most destructive types of noise affecting superconducting qubits. Previous approaches to suppressing ZZ crosstalk have mainly relied on specific chip design that can complicate chip fabrication and aggravate decoherence. To some extent, special chip design can be avoided by relying on pulse optimization to suppress ZZ crosstalk. However, existing approaches are non-scalable, as their required time and memory grow exponentially with the number of qubits involved. Jidong Zhai, Jonathan Allcock, Shengyu Zhang 0002, Yicong Zheng |
ASPLOS | 2 |
| 2022 | AStitch: enabling a new multi-dimensional optimization space for memory-intensive ML training and inference on modern SIMT architecturesabstractThis work reveals that memory-intensive computation is a rising performance-critical factor in recent machine learning models. Due to a unique set of new challenges, existing ML optimizing compilers cannot perform efficient fusion under complex two-level dependencies combined with just-in-time demand. They face the dilemma of either performing costly fusion due to heavy redundant computation, or skipping fusion which results in massive number of kernels. Furthermore, they often suffer from low parallelism due to the lack of support for real-world production workloads with irregular tensor shapes. To address these rising challenges, we propose AStitch, a machine learning optimizing compiler that opens a new multi-dimensional optimization space for memory-intensive ML computations. It systematically abstracts four operator-stitching schemes while considering multi-dimensional optimization objectives, tackles complex computation graph dependencies with novel hierarchical data reuse, and efficiently processes various tensor shapes via adaptive thread mapping. Finally, AStitch provides just-in-time support incorporating our proposed optimizations for both ML training and inference. Although AStitch serves as a stand-alone compiler engine that is portable to any version of TensorFlow, its basic ideas can be generally applied to other ML frameworks and optimization compilers. Experimental results show that AStitch can achieve an average of 1.84x speedup (up to 2.73x) over the state-of-the-art Google's XLA solution across five production workloads. We also deploy AStitch onto a production cluster for ML workloads with thousands of GPUs. The system has been in operation for more than 10 months and saves about 20,000 GPU hours for 70,000 tasks per week. Zhen Zheng, Xuanda Yang, Pengzhan Zhao, Guoping Long, Kai Zhu 0004, Feiwen Zhu, Wenyi Zhao, Jun Yang 0052, Jidong Zhai, Shuaiwen Song, Wei Lin 0016 |
ASPLOS | 10 |
| 2022 | GraphQ IR: Unifying the Semantic Parsing of Graph Query Languages with One Intermediate RepresentationabstractSubject to the huge semantic gap between natural and formal languages, neural semantic parsing is typically bottlenecked by its complexity of dealing with both input semantics and output syntax.Recent works have proposed several forms of supplementary supervision but none is generalized across multiple formal languages.This paper proposes a unified intermediate representation (IR) for graph query languages, named GraphQ IR.It has a natural-language-like expression that bridges the semantic gap and formally defined syntax that maintains the graph structure.Therefore, a neural semantic parser can more precisely convert user queries into GraphQ IR, which can be later losslessly compiled into various downstream graph query languages.Extensive experiments on several benchmarks including KQA PRO, OVERNIGHT, GRAILQA and METAQA-Cypher under standard i.i.d., out-of-distribution and low-resource settings validate GraphQ IR's superiority over the previous state-of-the-arts with a maximum 11% accuracy improvement. Lunyiu Nie, Shulin Cao, Jiaxin Shi, Jiuding Sun, Lei Hou 0001, Juan-Zi Li, Jidong Zhai |
EMNLP | 8 |
| 2022 | Efficiently emulating high-bitwidth computation with low-bitwidth hardwareabstractDomain-Specific Accelerators (DSAs) are being rapidly developed to support high-performance domain-specific computation. Although DSAs provide massive computation capability, they often only support limited native data types. To mitigate this problem, previous works have explored software emulation for certain data types, which provides some compensation for hardware limitations. However, how to efficiently design more emulated data types and choose a high-performance one without hurting correctness or precision for a given application still remains an open problem. Zixuan Ma, Haojie Wang 0004, Guanyu Feng, Chen Zhang 0001, Jiaao He, Shengqi Chen 0001, Jidong Zhai |
ICS | 8 |
| 2022 | FreeTensor: a free-form DSL with holistic optimizations for irregular tensor programsabstractTensor programs are of critical use in many domains. Existing frameworks, such as PyTorch, TensorFlow, and JAX, adopt operator-based programming to ease programming, increase performance, and perform automatic differentiation. However, as the rapid development of tensor programs, operator-based programming shows significant limitations for irregular patterns since a large amount of redundant computation or memory access is introduced. Shizhi Tang, Jidong Zhai, Haojie Wang 0004, Liyan Zheng 0001, Zhenhao Yuan, Chen Zhang 0001 |
PLDI | 2 |
| 2022 | FasterMoE: modeling and optimizing training of large-scale dynamic pre-trained modelsabstractThe current trend in deep learning is to scale models to extremely large sizes with the objective of increasing their accuracy. Mixture-of-Expert (MoE) is the most popular pre-trained model that makes feasible the training of models with parameters beyond trillion-scale. Thanks to the dynamic activation of experts, i.e., shallow layers specialized in certain domains, it allows for sparse training of bigger models, removing the linearity between model size and computation. However, different from traditional deep learning models, it draws huge challenges to the efficiency of these training systems, including dynamic load imbalance, inefficient synchronous execution mode, and congested all-to-all communication. Jiaao He, Jidong Zhai, Tiago Antunes, Haojie Wang 0004, Fuwen Luo, Shangfeng Shi |
PPoPP | 2 |
| 2022 | PerFlow: a domain specific framework for automatic performance analysis of parallel applicationsabstractPerformance analysis is widely used to identify performance issues of parallel applications. However, complex communications and data dependence, as well as the interactions between different kinds of performance issues make high-efficiency performance analysis even harder. Although a large number of performance tools have been designed, accurately pinpointing root causes for such complex performance issues still needs specific in-depth analysis. To implement each such analysis, significant human efforts and domain knowledge are normally required. Yuyang Jin 0001, Haojie Wang 0004, Runxin Zhong, Chen Zhang 0001, Jidong Zhai |
PPoPP | 5 |
| 2022 | BaGuaLu: targeting brain scale pretrained models with over 37 million coresabstractLarge-scale pretrained AI models have shown state-of-the-art accuracy in a series of important applications. As the size of pretrained AI models grows dramatically each year in an effort to achieve higher accuracy, training such models requires massive computing and memory capabilities, which accelerates the convergence of AI and HPC. However, there are still gaps in deploying AI applications on HPC systems, which need application and system co-design based on specific hardware features. Zixuan Ma, Jiaao He, Jiezhong Qiu, Huanqi Cao, Yuanwei Wang, Zhenbo Sun, Liyan Zheng 0001, Haojie Wang 0004, Shizhi Tang, Tianyu Zheng, Junyang Lin, Guanyu Feng, Zeqiang Huang, Aohan Zeng, Jianwei Zhang 0012, Runxin Zhong, Tianhui Shi, Jie Tang 0001, Hongxia Yang, Xin Liu 0086, Jidong Zhai |
PPoPP | 24 |
| 2022 | Vapro: performance variance detection and diagnosis for production-run parallel applicationsabstractPerformance variance is a serious problem for parallel applications, which can cause performance degradation and make applications' behavior hard to understand. Therefore, detecting and diagnosing performance variance are of crucial importance for users and application developers. However, previous detection approaches either bring too large overhead and hurt applications' performance, or rely on nontrivial source code analysis that is impractical for production-run parallel applications. Liyan Zheng 0001, Jidong Zhai, Xiongchao Tang, Haojie Wang 0004, Yuyang Jin 0001, Shuaiwen Song |
PPoPP | 2 |
| 2022 | UniQ: A Unified Programming Model for Efficient Quantum Circuit SimulationabstractQuantum circuit simulation is critical for verifying quantum computers. Given exponential complexity in the simulation, existing simulators use different architectures to accelerate the simulation. However, due to the variety of both simulation methods and modern architectures, it is challenging to design a high-performance yet portable simulator. In this work, we propose UniQ, a unified programming model for multiple simulation methods on various hardware architectures. We provide a unified application abstraction to describe different applications, and a unified hierarchical hardware abstraction upon different hardware. Based on these abstractions, UniQ can perform various circuit transformations without being aware of either concrete application or architecture detail, and generate high-performance execution schedules on different platforms without much human effort. Evaluations on CPU, GPU, and Sunway platforms show that UniQ can accelerate quantum circuit simulation by up to 28.59× (4.47× on average) over state-of-the-art frameworks, and successfully scale to 399,360 cores on 1,024 nodes. Chen Zhang 0001, Haojie Wang 0004, Zixuan Ma, Jidong Zhai |
SC | 6 |
| 2022 | CompressDB: Enabling Efficient Compressed Data Direct Processing for Various DatabasesabstractIn modern data management systems, directly performing operations on compressed data has been proven to be a big success facing big data problems. These systems have demonstrated significant compression benefits and performance improvement for data analytics applications. However, current systems only focus on data queries, while a complete big data system must support both data query and data manipulation. Feng Zhang 0007, Weitao Wan, Chenyang Zhang 0005, Jidong Zhai, Yunpeng Chai, Haixiang Li, Xiaoyong Du 0001 |
SIGMOD Conference | 4 |
| 2022 | Zoro: A robotic middleware combining high performance and high reliability
Wei Liu 0143, Jiangming Jin, Hao Wu 0012, Yifan Gong 0003, Ziyue Jiang 0002, Jidong Zhai |
J. Parallel Distributed Comput. | 6 |
| 2022 | Periodic Weather-Aware LSTM With Event Mechanism for Parking Behavior PredictionabstractThere are plenty of parking spaces in big cities, but we often find nowhere to park. For example, New York has 1.4 million cars and 4.4 million on-street parking spaces, but it is still not easy to find a parking place near our destination, especially during peak hours. The reason is the lack of prediction of parking behavior. If we could provide parking behavior in advance, we can ease this parking problem that affects human well-being. We observe that parking lots have periodic parking patterns, which is an important factor for parking behavior prediction. Unfortunately, existing work ignores such periodic parking patterns in parking behavior prediction, and thus incurs low accuracy. To solve this problem, we propose PewLSTM, a novel periodic weather-aware LSTM model that successfully predicts the parking behavior based on historical records, weather, environments, weekdays, and events. PewLSTM includes a periodic weather-aware LSTM prediction module and an event prediction module, for predicting parking behaviors in regular days and events. PewLSTM is extremely useful for drivers and parking lot owners to improve customer experience. For example, the probability of parking space that will be available soon can be provided even if the parking lot is full. Based on 910,477 real parking records in 904 days from 13 parking lots, PewLSTM yields 93.84% parking prediction accuracy, which is about 30% higher than the state-of-the-art parking behavior prediction method. Additionally, we have analyzed parking behaviors in events like holidays and COVID-19. PewLSTM can handle parking behavior prediction in events and reaches 90.68 percent accuracy. Feng Zhang 0007, Yani Liu, Ningxuan Feng, Cheng Yang 0002, Jidong Zhai, Shuhao Zhang 0001, Bingsheng He, Jiazao Lin, Xiao Zhang 0001, Xiaoyong Du 0001 |
IEEE Trans. Knowl. Data Eng. | 5 |
| 2022 | Exploring Query Processing on CPU-GPU Integrated Edge DeviceabstractHuge amounts of data have been generated on edge devices every day, which requires efficient data analytics and management. However, due to the limited computing capacity of these edge devices, query processing at the edge faces tremendous pressure. Fortunately, in recent years, hardware vendors have integrated heterogeneous coprocessors, such as GPUs, into the edge device, which can provide much more computing power. Furthermore, the CPU-GPU integrated edge device has shown significant benefits in a variety of situations. Therefore, the exploration of query processing on such CPU-GPU integrated edge devices becomes an urgent need. In this article, we develop a fine-grained query processing engine, called FineQuery, which can perform efficient query processing on CPU-GPU integrated edge devices. Particularly, FineQuery can take advantage of both architectural features of edge devices and query characteristics by performing fine-grained workload scheduling between the CPU and the GPU. Experiments show that on TPC-H workloads, FineQuery reduces 42.81% latency and improves 2.39× bandwidth utilization on average compared to the implementation of using only GPU or CPU. Furthermore, query processing at the edge can bring significant performance-per-cost benefits and energy efficiency. On average, FineQuery at the edge brings 21× performance-per-cost ratio and 4× energy efficiency compared with processing the data on a discrete GPU platform. Jiesong Liu, Feng Zhang 0007, Hourun Li, Dalin Wang, Weitao Wan, Xiaokun Fang, Jidong Zhai, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 7 |
| 2022 | Exploring Data Analytics Without Decompression on Embedded GPU SystemsabstractWith the development of computer architecture, even for embedded systems, GPU devices can be integrated, providing outstanding performance and energy efficiency to meet the requirements of different industries, applications, and deployment environments. Data analytics is an important application scenario for embedded systems. Unfortunately, due to the limitation of the capacity of the embedded device, the scale of problems handled by the embedded system is limited. In this paper, we propose a novel data analytics method, called G-TADOC, for efficient text analytics directly on compression on embedded GPU systems. A large amount of data can be compressed and stored in embedded systems, and can be processed directly in the compressed state, which greatly enhances the processing capabilities of the systems. Particularly, G-TADOC has three innovations. First, a novel fine-grained thread-level workload scheduling strategy for GPU threads has been developed, which partitions heavily-dependent loads adaptively in a fine-grained manner. Second, a GPU thread-safe memory pool has been developed to handle inconsistency with low synchronization overheads. Third, a sequence-support strategy is provided to maintain high GPU parallelism while ensuring sequence information for lossless compression. Moreover, G-TADOC involves special optimizations for embedded GPUs, such as utilizing the CPU-GPU shared unified memory. Experiments show that G-TADOC provides 13.2× average speedup compared to the state-of-the-art TADOC. G-TADOC also improves performance-per-cost by 2.6× and energy efficiency by 32.5× over TADOC. Zaifeng Pan, Feng Zhang 0007, Yanliang Zhou, Jidong Zhai, Xipeng Shen, Onur Mutlu, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 4 |
| 2022 | Guest EditorialabstractThis special section focuses on the state-of-the-art technologies on parallel and distributed computing techniques for artificial intelligence (AI), machine learning (ML), and deep learning (DL). AI, ML, and DL can enable computers the ability to learn from a large amount of data and use the learned model to optimize a complex problem or discover rules in a complicated system. AI, ML and DL can be applied to push forward the boundaries for many domains and significantly influence our daily life. Jidong Zhai, Min Si, Antonio J. Peña |
IEEE Trans. Parallel Distributed Syst. | 1 |
| 2022 | Leveraging Code Snippets to Detect Variations in the Performance of HPC SystemsabstractVariations in the performance of parallel and distributed systems are becoming increasingly challenging. The runtimes of different executions can vary greatly even with a fixed number of computing nodes. Many HPC applications on supercomputers exhibit such variance. This not only leads to unpredictable execution times, but also renders the system’s behavior unintuitive. The efficient online detection of variations in performance is an open problem in HPC research. To solve it, we propose an approach, calledvSensor, to detect variations in the performance of systems. The key finding of this study is that the source code of programs can better represent performance at runtime than an external detector. Specifically, many HPC applications contain code snippets that are fixed workload patterns of execution, e.g., the workload of an invariant quantity and a linearly growing workload. This observation allows us to automatically identify these snippets of workload-related code and use them to detect variations in performance. We evaluatevSensoron the Tianhe-2A system with a large number of parallel applications, and the results indicate that it can efficiently identify variations in system performance. The average overhead of 4,096 processes is less than 6% for fixed-workload v-sensors. We identify a problematic node with slow memory by usingvSensorthat degrades the performance of the program by 21%. A serious issue with network performance is also detected that slows down the Tianhe-2A system by 3.37 times for an HPC kernel. Jidong Zhai, Liyan Zheng 0001, Jinghan Sun, Feng Zhang 0007, Xiongchao Tang, Xuehai Qian, Bingsheng He, Wei Xue 0003 |
IEEE Trans. Parallel Distributed Syst. | 1 |
| 2022 | Detecting Performance Variance for Parallel Applications Without Source CodeabstractFor parallel applications, performance variance is a critical issue that can degrade performance and make applications’ behavior difficult to explain. Therefore, users and application developers should be able to detect and diagnose performance variance. Previous detection methods either introduce too much overhead and slow down applications, or rely on nontrivial source code analysis, which is impractical for production-run parallel systems. In this article, we proposeVapro, a framework for detecting and diagnosing performance variance in production-run parallel systems. Our method is based on an observation that most parallel programs contain code snippets that are executed repeatedly with a fixed workload and can be utilized to detect performance variance. We present State Transition Graph (STG) to track program execution and then do light-weight workload analysis on STG to locate performance variance.Vaprois able to successfully identify these snippets at runtime even without program source code. To diagnose the discovered variation,Vaprouses a progressive diagnosis method based on a hybrid model combining variance breakdown and statistical analysis. According to evaluating results,Vapro's performance overhead is only 1.38% on average.Vaprocan identify performance variance in real applications caused by hardware issues, such as memory and IO. The standard deviation of the execution time is decreased by up to 73.5% when the identified variance is fixed.Vaproachieves 30.0% larger detection coverage than the state-of-the-art variance detection approach based on source code analysis. Jidong Zhai, Liyan Zheng 0001, Feng Zhang 0007, Xiongchao Tang, Haojie Wang 0004, Yuyang Jin 0001, Shuaiwen Song |
IEEE Trans. Parallel Distributed Syst. | 1 |
| 2022 | POCLib: A High-Performance Framework for Enabling Near Orthogonal Processing on CompressionabstractParallel technology boosts data processing in recent years, and parallel direct data processing on hierarchically compressed documents exhibits great promise. The high-performance direct data processing technique brings large savings in both time and space by removing the need for decompressing data. However, its benefits have been limited to data traversal operations; for random accesses, direct data processing is several times slower than the state-of-the-art baselines. This article proposes a novel concept, orthogonal processing on compression (orthogonal POC), which means that text analytics can be efficiently supported directly on compressed data, regardless of the type of the data processing – that is, the type of data processing is orthogonal to its capability of conducting POC. Previous proposals, such as TADOC, are not orthogonal POC. This article presents a set of techniques that successfully eliminate the limitation, and for the first time, establishes the near orthogonal POC feasibility of effectively handling both data traversal operations and random data accesses on hierarchically-compressed data. The work focuses on text data and yields a unified high-performance library, called POCLib. In a ten-node distributed Spark cluster on Amazon EC2, POCLib achieves 3.1× speedup over the state-of-the-art on random data accesses to compressed data, while preserving the capability of supporting traversal operations efficiently and providing large (3.9×) space savings. Feng Zhang 0007, Jidong Zhai, Xipeng Shen, Onur Mutlu, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2022 | Critique of "MemXCT: Memory-Centric X-Ray CT Reconstruction With Massive Parallelization" by SCC Team From Tsinghua UniversityabstractHidayetoğluet al.propose a novel memory-centric algorithm to reconstruct X-ray CT images in the SC19 article entitled “MemXCT: Memory-Centric X-ray CT Reconstruction with Massive Parallelization”. They formulate the reconstruction with several SpMVs, and propose two memory-centric optimizations to improve cache locality for better memory bandwidth utilization, i.e., a two-level pseudo-Hilbert ordering and a multi-stage input buffering. In this article, we present our results on reproducing that article to show its effectiveness and generality, as part of the SC20 Student Cluster Competition Reproducibility Challenge. We reproduce the execution time and memory bandwidth tests in that article on various architectures, including Intel CPUs, AMD CPUs, and NVIDIA GPUs. We further analyze the bottleneck on different architectures by comparing the achieved memory bandwidth with the peak bandwidth on those architectures. We then reproduce the strong scaling test on CPU and GPU clusters with different scales, and use the proposed algorithm to reconstruct three new X-ray computed tomograms. Runxin Zhong, Chen Zhang 0001, Mingshu Zhai, Lin Gan 0001, Jidong Zhai |
IEEE Trans. Parallel Distributed Syst. | 9 |
| 2022 | Payment behavior prediction on shared parking lots with TR-GCN
Feng Zhang 0007, Mingde Zhang, Jidong Zhai, Bingsheng He, Cheng Yang 0002, Shuhao Zhang 0001, Jiazao Lin, Haidi Liu, Xiaoyong Du 0001 |
VLDB J. | 4 |
| 2021 | Accelerating GPU Message Communication for Autonomous Navigation SystemsabstractAutonomous navigation systems consist of multiple software modules, such as sensing, object detection, and planning, to achieve traffic perception and fast decision making. Such a system generates a large amount of data and requires data processing and communication in real-time. Although accelerators, such as GPUs, have been exploited to speed up data processing, communicating GPU messages between modules is still lacking support, leading to high communication latency and resource contention. For such a latency-sensitive and resource-limited autonomous navigation system, high performance and lightweight message communication are crucial and demanding. To obtain both high performance and low resource usages, we first propose a novel pub-centric memory pool and an on-the-fly offset conversion algorithm to avoid unnecessary data movement. Secondly, we combine these two techniques and propose an efficient message communication on a single GPU. Finally, we extend this approach to multi-GPU and design a framework that natively supports GPU message communication for Inter-Process Communication. With comprehensive evaluation, results show our approach is able to reduce communication latency by 53.7% for PointCloud and Image messages compared to the state-of-the-art approach. Moreover, in the real autonomous navigation scenario, our approach reduces the end-to-end latency by 29.2% and decreases resource usage up to 58.9%. Hao Wu 0012, Jiangming Jin, Jidong Zhai, Yifan Gong 0003, Wei Liu 0143 |
CLUSTER | 3 |
| 2021 | Mitigating Crosstalk in Quantum Computers through Commutativity-Based Instruction ReorderingabstractCrosstalk is one of the major types of noise in quantum computers. To design high-fidelity quantum gates and large-scale quantum computers, effectively suppressing crosstalk is becoming increasingly important. Previous approaches to mitigate crosstalk rely on either hardware strategies, which are only applicable on limited platforms, or software techniques, which, however, cannot fully explore instruction parallelism. To address the above challenges, we propose a commutativity based compile-time instruction reordering approach. We first propose a complete set of generalized commutativity rules for main types of quantum gates, and then we design a two-level bidirectional reordering algorithm to reorder gates according to these rules for effective crosstalk mitigation. Our approach can not only capture both forward and backward instruction correlations but also reduce the effect of decoherence. We evaluate our approach with 117 quantum programs. Compared with the state-of-the-art, our approach can improve program fidelity by up to 120% (18% on average). Jidong Zhai |
DAC | 2 |
| 2021 | G-TADOC: Enabling Efficient GPU-Based Text Analytics without DecompressionabstractText analytics directly on compression (TADOC) has proven to be a promising technology for big data analytics. GPUs are extremely popular accelerators for data analytics systems. Unfortunately, no work so far shows how to utilize GPUs to accelerate TADOC. We describe G-TADOC, the first framework that provides GPU-based text analytics directly on compression, effectively enabling efficient text analytics on GPUs without decompressing the input data. G-TADOC solves three major challenges. First, TADOC involves a large amount of dependencies, which makes it difficult to exploit massive parallelism on a GPU. We develop a novel fine-grained thread-level workload scheduling strategy for GPU threads, which partitions heavily-dependent loads adaptively in a fine-grained manner. Second, in developing G-TADOC, thousands of GPU threads writing to the same result buffer leads to inconsistency while directly using locks and atomic operations lead to large synchronization overheads. We develop a memory pool with thread-safe data structures on GPUs to handle such difficulties. Third, maintaining the sequence information among words is essential for lossless compression. We design a sequence-support strategy, which maintains high GPU parallelism while ensuring sequence information. Our experimental evaluations show that G-TADOC provides 31.1× average speedup compared to state-of-the-art TADOC. Feng Zhang 0007, Zaifeng Pan, Yanliang Zhou, Jidong Zhai, Xipeng Shen, Onur Mutlu, Xiaoyong Du 0001 |
ICDE | 4 |
| 2021 | HyQuas: hybrid partitioner based quantum circuit simulation system on GPUabstractQuantum computing has shown its strong potential in solving certain important problems. Due to the intrinsic limitations of current real quantum computers, quantum circuit simulation still plays an important role in both research and development of quantum computing. GPU-based quantum circuit simulation has been explored due to GPU's high computation capability. Despite previous efforts, existing quantum circuit simulation systems usually rely on a single method to improve poor data locality caused by complex quantum entanglement. However, we observe that existing simulation methods show significantly different performance for different circuit patterns. The optimal performance cannot be obtained only with any single method. Chen Zhang 0001, Haojie Wang 0004, Kaiyuan Rong, Jidong Zhai |
ICS | 5 |
| 2021 | PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
Haojie Wang 0004, Jidong Zhai, Mingyu Gao 0001, Zixuan Ma, Shizhi Tang, Liyan Zheng 0001, Yuanzhi Li, Kaiyuan Rong, Yuanyong Chen |
OSDI | 2 |
| 2021 | Understanding and bridging the gaps in current GNN performance optimizationsabstractGraph Neural Network (GNN) has recently drawn a rapid increase of interest in many domains for its effectiveness in learning over graphs. Maximizing its performance is essential for many tasks, but remains preliminarily understood. In this work, we provide an in-depth examination of the state-of-the-art GNN frameworks, revealing five major gaps in the current frameworks in optimizing GNN performance, especially in handling the special complexities of GNN over traditional graph or DNN operations. Based on the insights, we put together a set of optimizations to fill the gaps. These optimizations leverage the state-of-the-art GPU optimization techniques and tailor them to the special properties of GNN. Experimental results show that these optimizations achieve 1.37×--15.5× performance improvement over the state-of-the-art frameworks on various GNN models. Kezhao Huang, Jidong Zhai, Zhen Zheng, Youngmin Yi, Xipeng Shen |
PPoPP | 2 |
| 2021 | Preface
Xian-He Sun, Dong Li 0001, Wen-Guang Chen, Tao Li 0006, Jiwu Shu, Bo Wu 0002, Jin Xiong, Jinging Xue, Feng Zhang 0007, Jidong Zhai, Zhiia Zhao |
J. Comput. Sci. Technol. | 10 |
| 2021 | A Fast Lock for Explicit Message Passing ArchitecturesabstractSynchronization is a crucial issue for multi-threaded programs. Mutex locks are widely used in legacy programs and are still popular for the intuition semantics. The SW26010 architecture, deployed on the supercomputer Sunway TaihuLight, introduces a hardware-supported inter-core message passing mechanism and exposes explicit interfaces for developers to use its fast on-chip network. This emerging architectural feature brings both opportunities and challenges for mutex lock implementation. However, there is still no general lock mechanism, especially designed and optimized for architectures with this new feature. In this article, we propose mLock, a fast lock designed and optimized for architectures that support Explicit inter-core Message Passing (EMP). mLock uses partial cores as lock servers and leverages the fast on-chip network to implement high-performance mutual exclusive locks. In this article, we propose a series of novel techniques to improve the performance of EMP locks. First, we propose the concepts ofchaining lockandhierarchical lockto reduce message count and mitigate network congestion. Second, we propose afair lockapproach to improve the fairness of EMP locks. Third,server reusingis introduced to reduce the number of lock servers. We implement and evaluate mLock on an SW26010 processor. Experimental results show that our proposed techniques can improve the performance of EMP locks by up to$16.2\times$over a basic design. Xiongchao Tang, Chen Zhang 0001, Jidong Zhai, Xuehai Qian, Yong Jiang 0001 |
IEEE Trans. Computers | 3 |
| 2021 | Automatic Irregularity-Aware Fine-Grained Workload Partitioning on Integrated ArchitecturesabstractThe integrated architecture that features both CPU and GPU on the same die is an emerging and promising architecture for fine-grained CPU-GPU collaboration. However, the integration also brings forward several programming and system optimization challenges, especially for irregular applications such as graph processing. The complex interplay between heterogeneity and irregularity leads to very low processor utilization of running irregular applications on integrated architectures. Furthermore, fine-grained co-processing on the CPU and GPU is still an open problem. Particularly, in this paper, we show that the previous workload partitioning for CPU-GPU co-processing is far from ideal in terms of resource utilization and performance. To solve this problem, we propose a system software called FinePar, which considers architectural differences of the CPU and GPU and leverages fine-grained collaboration enabled by integrated architectures. Through irregularity-aware performance modeling and online auto-tuning, FinePar partitions irregular workloads and achieves both device-level and thread-level load balance. We evaluate FinePar with eight irregular applications in graphs and sparse matrices on two integrated architectures and compare it with state-of-the-art partitioning approaches. Results show that FinePar demonstrates better resource utilization and achieves an average of 1.6X speedup over the optimal coarse-grained partitioning method. Feng Zhang 0007, Jidong Zhai, Bo Wu 0002, Bingsheng He, Xiaoyong Du 0001 |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2021 | Guest Editorial
Pavan Balaji, Jidong Zhai, Min Si |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2021 | Collaborative Heterogeneity-Aware OS Scheduler for Asymmetric Multicore ProcessorsabstractAsymmetric multicore processors (AMP) offer multiple types of cores under the same programming interface. Extracting the full potential of AMPs requires intelligent scheduling decisions, matching each thread with the right kind of core, the core that will maximize performance or minimize wasted energy for this thread. Existing OS schedulers are not up to this task. While they may handle certain aspects of asymmetry in the system, none can handle all runtime factors affecting AMPs for the general case of multi-threaded multi-programmed workloads. We address this problem by introducing COLAB, a general purpose asymmetry-aware scheduler targeting multi-threaded multi-programmed workloads. It estimates the performance and power of each thread on each type of core and identifies communication patterns and bottleneck threads. With this information, the scheduler makes coordinated core assignment and thread selection decisions that still provide each application its fair share of the processor's time. We evaluate our approach using both the GEM5 simulator on four distinct big.LITTLE configurations and a development board with ARM Cortex-A73/A53 processors and mixed workloads composed of PARSEC and SPLASH2 benchmarks. Compared to the state-of-the art Linux CFS and AMP-aware schedulers, we demonstrate performance gains of up to 25 and 5 to 15 percent on average, together with an average 5 percent energy saving depending on the hardware setup. Runxin Zhong, Vladimir Janjic, Pavlos Petoumenos, Jidong Zhai, Hugh Leather, John Thomson |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2021 | An Efficient Parallel Secure Machine Learning Framework on GPUsabstractMachine learning is widely used in our daily lives. Large amounts of data have been continuously produced and transmitted to the cloud for model training and data processing, which raises a problem: how to preserve the security of the data. Recently, a secure machine learning system named SecureML has been proposed to solve this issue using two-party computation. However, due to the excessive computation expenses of two-party computation, the secure machine learning is about 2× slower than the original machine learning methods. Previous work on secure machine learning mostly focused on novel protocols or improving accuracy, while the performance metric has been ignored. In this article, we propose a GPU-based framework ParSecureML to improve the performance of secure machine learning algorithms based on two-party computation. The main challenges of developing ParSecureML lie in the complex computation patterns, frequent intra-node data transmission between CPU and GPU, and complicated inter-node data dependence. To handle these challenges, we propose a series of novel solutions, including profiling-guided adaptive GPU utilization, fine-grained double pipeline for intra-node CPU-GPU cooperation, and compressed transmission for inter-node communication. Moreover, we integrate architecture specific optimizations, such as Tensor Cores, into ParSecureML. As far as we know, this is the first GPU-based secure machine learning framework. Compared to the state-of-the-art framework, ParSecureML achieves an average of 33.8× speedup. ParSecureML can also be applied to inferences, which achieves 31.7× speedup on average. Feng Zhang 0007, Zheng Chen 0023, Chenyang Zhang 0005, Amelie Chi Zhou, Jidong Zhai, Xiaoyong Du 0001 |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2021 | Critique of "Planetary Normal Mode Computation: Parallel Algorithms, Performance, and Reproducibility" by SCC Team From Tsinghua UniversityabstractIn this article we present our results from the SC19 Student Cluster Competition Reproducibility Challenge. The challenge entails reproducing the article entitled “Computing Planetary Interior Normal Modes with A Highly Parallel Polynomial Filtering Eigensolver” presented at SC'18, which proposes a parallel polynomial filtered Lanczos algorithm to directly calculate the planetary normal modes of heterogeneous planets. The proposed algorithm showed excellent performance with relatively low memory consumption and high parallel efficiency. In this work, we reproduce the scaling tests in that article on a cluster using Intel Cascade Lake architecture and use the proposed algorithm to illustrate specific normal modes of Mars. We compare the results obtained on our cluster with those in the original article. We also design a new metric to better analyze the results. In addition, we use the profiling tool Intel VTune Amplifier to explain our discoveries. Our results demonstrate that the given models show great scalability, which is similar to the original article. The required normal modes of Mars are also successfully calculated and visualized. Chen Zhang 0001, Chenggang Zhao, Jiaao He, Shengqi Chen 0001, Liyan Zheng 0001, Kezhao Huang, Jidong Zhai |
IEEE Trans. Parallel Distributed Syst. | 8 |
| 2021 | TADOC: Text analytics directly on compression
Feng Zhang 0007, Jidong Zhai, Xipeng Shen, Dalin Wang, Zheng Chen 0023, Onur Mutlu, Xiaoyong Du 0001 |
VLDB J. | 2 |
| 2020 | GOPipe: A Granularity-Oblivious Programming Framework for Pipelined Stencil Executions on GPUabstractRecent studies have shown promising performance benefits when multiple stages of a pipelined stencil application are mapped to different parts of a GPU to run concurrently. An important factor for the computing efficiency of such pipelines is the granularity of a task. In previous programming frameworks that support true pipelined computations on GPU, the choice has to be made by the programmers during the application development time. Due to many difficulties, programmers' decisions are often far from optimal, causing inferior performance and performance portability. Chanyoung Oh, Zhen Zheng, Xipeng Shen, Jidong Zhai, Youngmin Yi |
PACT | 4 |
| 2020 | Elan: Towards Generic and Efficient Elastic Training for Deep LearningabstractShowing a promising future in improving resource utilization and accelerating training, elastic deep learning training has been attracting more and more attention recently. Nevertheless, existing approaches to provide elasticity have certain limitations. They either fail to fully explore the parallelism of deep learning training when scaling out or lack an efficient mechanism to replicate training states among different devices.To address these limitations, we design Elan, a generic and efficient elastic training system for deep learning. In Elan, we propose a novel hybrid scaling mechanism to make a good trade-off between training efficiency and model performance when exploring more parallelism. We exploit the topology of underlying devices to perform concurrent and IO-free training state replication. To avoid the high overhead of start and initialization, we further propose an asynchronous coordination mechanism. Powered by the above innovations, Elan can provide high-performance (~1s) migration, scaling in and scaling out support with negligible runtime overhead (<3‰). For elastic training of ResNet-50 on ImageNet, Elan improves the time to solution by 20%. For elastic scheduling, with the help of Elan, resource utilization is improved by 21%+ and job pending time is reduced by 43%+. Jidong Zhai, Baodong Wu, Xingcheng Zhang, Peng Sun 0006, Shengen Yan |
ICDCS | 2 |
| 2020 | Enabling Efficient Random Access to Hierarchically-Compressed DataabstractRecent studies have shown the promise of direct data processing on hierarchically-compressed text documents. By removing the need for decompressing data, the direct data processing technique brings large savings in both time and space. However, its benefits have been limited to data traversal operations; for random accesses, direct data processing is several times slower than the state-of-the-art baselines. This paper presents a set of techniques that successfully eliminate the limitation, and for the first time, establishes the feasibility of effectively handling both data traversal operations and random data accesses on hierarchically-compressed data. The work yields a new library, which achieves 3.1× speedup over the state-of-the-art on random data accesses to compressed data, while preserving the capability of supporting traversal operations efficiently and providing large (3.9×) space savings. Feng Zhang 0007, Jidong Zhai, Xipeng Shen, Onur Mutlu, Xiaoyong Du 0001 |
ICDE | 2 |
| 2020 | ParSecureML: An Efficient Parallel Secure Machine Learning Framework on GPUsabstractMachine learning has been widely used in our daily lives. Large amounts of data have been continuously produced and transmitted to the cloud for model training and data processing, which raises a problem: how to preserve the security of the data. Recently, a secure machine learning system named SecureML has been proposed to solve this issue using two-party computation. However, due to the excessive computation expenses of two-party computation, the secure machine learning is about 2x slower than the original machine learning methods. Previous work on secure machine learning mostly focused on novel protocols or improving accuracy, while the performance metric has been ignored. In this paper, we propose a GPU-based framework ParSecureML to improve the performance of secure machine learning algorithms based on two-party computation. The main challenges of developing ParSecureML lie in the complex computation patterns, frequent intra-node data transmission between CPU and GPU, and complicated inter-node data dependence. To handle these challenges, we propose a series of novel solutions, including profiling-guided adaptive GPU utilization, fine-grained double pipeline for intra-node CPU-GPU cooperation, and compressed transmission for inter-node communication. As far as we know, this is the first GPU-based secure machine learning framework. Compared to the state-of-the-art framework, ParSecureML achieves an average of 32.2x speedup. ParSecureML can be downloaded from https://github.com/ZhengChenCS/ParSecureML. Zheng Chen 0023, Feng Zhang 0007, Amelie Chi Zhou, Jidong Zhai, Chenyang Zhang 0005, Xiaoyong Du 0001 |
ICPP | 4 |
| 2020 | Memory-Centric Communication Mechanism for Real-time Autonomous Navigation ApplicationsabstractThere has been a remarkable increase in the speed of AI development over the past few years. Artificial intelligence and deep learning techniques are blooming and expanding in all forms to every sector possible. With the emerging intelligent autonomous navigation systems, both memory allocation and data movement are becoming the main bottlenecks in inter-process communication procedures, especially in supporting various types of messages between multiple programming languages. To reduce significant memory allocation and data movement cost, we propose a novel memory-centric mechanism, which includes a virtual layer based architecture and a pre-record memory allocation algorithm. Furthermore, we implement a memory-centric communication framework named Z-framework based on the proposed mechanism to achieve high efficient IPC procedures in autonomous navigation systems. Experimental results show that Z-framework is able to gain up to 41% and 35% performance improvement compared with the approach used in ROS2, which is an industry standard and the state-of-the-art approach used in CyberRT, respectively. Wei Liu 0143, Yifan Gong 0003, Hao Wu 0012, Jidong Zhai, Jiangming Jin |
ICPP | 4 |
| 2020 | Edge-Stream: a Stream Processing Approach for Distributed Applications on a Hierarchical Edge-computing SystemabstractWith the rapid growth of IoT devices, the traditional cloud computing scheme is inefficient for many IoT based applications, mainly due to network data flood, long latency, and privacy issues. To this end, the edge computing scheme is proposed to mitigate these problems. However, in an edge computing system, the application development becomes more complicated as it involves increasing levels of edge nodes. Although some efforts have been introduced, existing edge computing frameworks still have some limitations in various application scenarios. To overcome these limitations, we propose a new programming model called Edge-Stream. It is a simple and programmer-friendly model, which can cover typical scenarios in edge-computing. Besides, we address several new issues, such as data sharing and area awareness, in this model. We also implement a prototype of edge-computing framework based on the Edge-Stream model. A comprehensive evaluation is provided based on the prototype. Experimental results demonstrate the effectiveness of the model. Xiaoyang Wang 0006, Zhe Zhou 0002, Ping Han, Tong Meng, Guangyu Sun 0003, Jidong Zhai |
SEC | 6 |
| 2020 | PewLSTM: Periodic LSTM with Weather-Aware Gating Mechanism for Parking Behavior PredictionabstractIn big cities, there are plenty of parking spaces, but we often find nowhere to park. For example, New York has 1.4 million cars and 4.4 million on-street parking spaces, but it is still not easy to find a parking place near our destination, especially during peak hours. The reason is the lack of prediction of parking behavior. If we could provide parking behavior in advance, we can ease this parking problem that affects human well-being. We observe that parking lots have periodic parking patterns, which is an important factor for parking behavior prediction. Unfortunately, existing work ignores such periodic parking patterns in parking behavior prediction, and thus incurs low accuracy. To solve this problem, we propose PewLSTM, a novel periodic weather-aware LSTM model that successfully predicts the parking behavior based on historical records, weather, environments, and weekdays. PewLSTM has been successfully integrated into a real parking space reservation system, ThsParking, which is one of the top smart parking platforms in China. Based on 452,480real parking records in 683 days from 10 parking lots, PewLSTM yields 85.3% parking prediction accuracy, which is about 20% higher than the state-of-the-art parking behavior prediction method. The code and data can be obtained fromhttps://github.com/NingxuanFeng/PewLSTM. Feng Zhang 0007, Ningxuan Feng, Yani Liu, Cheng Yang 0002, Jidong Zhai, Shuhao Zhang 0001, Bingsheng He, Jiazao Lin, Xiaoyong Du 0001 |
IJCAI | 5 |
| 2020 | Payment Behavior Prediction and Statistical Analysis for Shared Parking Lots
Feng Zhang 0007, Mingde Zhang, Jidong Zhai, Jiazao Lin, Haidi Liu, Xiaoyong Du 0001 |
NPC | 4 |
| 2020 | Identifying scalability bottlenecks for large-scale parallel programs with graph analysisabstractScaling a parallel program to modern supercomputers is challenging due to inter-process communication, code serialization, and resource contention. Performance analysis tools for finding such scaling bottlenecks either base on profiling or tracing. Profiling incurs lower overheads but does not capture detailed dependencies needed for root-cause analyses. Tracing collects all information at prohibitive overheads. In this work, we develop ScalAna that uses static analysis techniques to achieve the best of both worlds---it enables the analyzability of traces at a cost similar to profiling. We leverage compiler and runtime lightweight techniques to generate performance graph and perform graph analysis algorithm to detect the root cause of scaling issues. We evaluate ScalAna with real applications on the Tianhe-2 supercomputer. Results show that our approach can effectively locate the root cause of scalability bottlenecks for real applications and incur less than 6.38% overhead (1.89% on average) for up to 2,048 processes. Yuyang Jin 0001, Haojie Wang 0004, Xiongchao Tang, Torsten Hoefler, Xu Liu 0001, Jidong Zhai |
PPoPP | 6 |
| 2020 | ScalAna: automating scaling loss detection with graph analysisabstractScaling a parallel program to modern supercomputers is challenging due to inter-process communication, Amdahl’s law, and resource contention. Performance analysis tools for finding such scaling bottlenecks either base on profiling or tracing. Profiling incurs low overheads but does not capture detailed dependencies needed for root-cause analysis. Tracing collects all information at prohibitive overheads. In this work, we design SCALANA that uses static analysis techniques to achieve the best of both worlds - it enables the analyzability of traces at a cost similar to profiling. SCALANA first leverages static compiler techniques to build a Program Structure Graph, which records the main computation and communication patterns as well as the program’s control structures. At runtime, we adopt lightweight techniques to collect performance data according to the graph structure and generate a Program Performance Graph. With this graph, we propose a novel approach, called backtracking root cause detection, which can automatically and efficiently detect the root cause of scaling loss. We evaluate SCALANA with real applications. Results show that our approach can effectively locate the root cause of scaling loss for real applications and incurs 1.73parcent overhead on average for up to 2,048 processes. We achieve up to 11.11parcent performance improvement by fixing the root causes detected by SCALANA on 2,048 processes. Yuyang Jin 0001, Haojie Wang 0004, Xiongchao Tang, Torsten Hoefler, Xu Liu 0001, Jidong Zhai |
SC | 7 |
| 2020 | GraphPi: high performance graph pattern matching through effective redundancy eliminationabstractGraph pattern matching, which aims to discover structural patterns in graphs, is considered one of the most fundamental graph mining problems in many real applications. Despite previous efforts, existing systems face two main challenges. First, inherent symmetry existing in patterns can introduce a large amount of redundant computation. Second, different matching orders for a pattern have significant performance differences and are quite hard to predict. When these factors are mixed, this problem becomes extremely complicated. High efficient pattern matching remains an open problem currently. To address these challenges, we propose GraphPi, a high performance distributed pattern matching system. GraphPi utilizes a new algorithm based on 2-cycles in group theory to generate multiple sets of asymmetric restrictions, where each set can eliminate redundant computation completely. We further design an accurate performance model to determine the optimal matching order and asymmetric restriction set for efficient pattern matching. We evaluate GraphPi on Tianhe-2A supercomputer. Results show that GraphPi outperforms the state-of-the-art system, by up to $ 105\times$ for 6 real-world graph datasets on a single node. We also scale GraphPi to 1,024 computing nodes (24,576 cores). Tianhui Shi, Mingshu Zhai, Jidong Zhai |
SC | 4 |
| 2019 | pLock: A Fast Lock for Architectures with Explicit Inter-core Message PassingabstractSynchronization is a significant issue for multi-threaded programs. Mutex lock, as a classic solution, is widely used in legacy programs and is still popular for its intuition. The SW26010 architecture, deployed on the supercomputer Sunway Taihulight, introduces hardware-supported inter-core message passing mechanism and exposes explicit interfaces for developers to use its fast on-chip network. This emerging architectural feature brings both opportunities and challenges for mutex lock implementation. However, there is still no general lock mechanism optimized for architectures with this new feature. In this paper, we propose pLock, a fast lock designed for architectures that support Explicit inter-core Message Passing (EMP). pLock uses partial cores as lock servers and leverages the fast on-chip network to implement high-performance mutual exclusive locks. We propose two new techniques -- chaining lock and hierarchical lock -- to reduce message count and mitigate network congestion. We implement and evaluate pLock on an SW26010 processor. The experimental results show that our proposed techniques improve the performance of EMP-lock by up to 19.4x over a basic design. Xiongchao Tang, Jidong Zhai, Xuehai Qian |
ASPLOS | 2 |
| 2019 | HiWayLib: A Software Framework for Enabling High Performance Communications for Heterogeneous Pipeline ComputationsabstractPipeline is a parallel computing model underpinning a class of important applications running on CPU-GPU heterogeneous systems. A critical aspect for the efficiency of such applications is the support of communications among pipeline stages that may reside on CPU and different parts of a GPU. Existing libraries of concurrent data structures do not meet the needs, due to the massive parallelism on GPU and the complexities in CPU-GPU memory and connections. This work gives an in-depth study on the communication problem. It identifies three key issues, namely, slow and error-prone detection of the end of pipeline processing, intensive queue contentions on GPU, and cumbersome inter-device data movements. This work offers solutions to each of the issues, and integrates all together to form a unified library named HiWayLib. Experiments show that HiWayLib significantly boosts the efficiency of pipeline communications in CPU-GPU heterogeneous applications. For real-world applications, HiWayLib produces 1.22~2.13× speedups over the state-of-art implementations with little extra programming effort required. Zhen Zheng, Chanyoung Oh, Jidong Zhai, Xipeng Shen, Youngmin Yi |
ASPLOS | 3 |
| 2019 | Automatic, Application-Aware I/O Forwarding Resource Allocation
Bin Yang 0043, Xiaosong Ma, Xiupeng Zhu, Xiyang Wang 0003, Nosayba El-Sayed, Jidong Zhai, Wei Xue 0003 |
FAST | 8 |
| 2019 | Statistical Analysis and Prediction of Parking Behavior
Ningxuan Feng, Feng Zhang 0007, Jiazao Lin, Jidong Zhai, Xiaoyong Du 0001 |
NPC | 4 |
| 2019 | End-to-end I/O Monitoring on a Leading Supercomputer
Bin Yang 0043, Xiaosong Ma, Xiyang Wang 0003, Xiupeng Zhu, Nosayba El-Sayed, Haidong Lan, Jidong Zhai, Wei Xue 0003 |
NSDI | 10 |
| 2019 | GOPipe: a granularity-oblivious programming framework for pipelined stencil executions on GPUabstractRecent studies have shown promising performance benefits of pipelined stencil applications. An important factor for the computing efficiency of such pipelines is the granularity of a task. We presents GOPipe, the first granularity-oblivious programming framework for efficient pipelined stencil executions. With GOPipe, programmers no longer need to specify the appropriate task granularity. GOPipe automatically finds it, and schedules tasks of that granularity while observing all inter-task and inter-stage data dependencies. In our experiments on four real-life applications, GOPipe outperforms the state-of-the-art by up to 4.57× with a much better programming productivity. Chanyoung Oh, Zhen Zheng, Xipeng Shen, Jidong Zhai, Youngmin Yi |
PPoPP | 4 |
| 2019 | Spread-n-share: improving application performance and cluster throughput with resource-aware job placementabstractTraditional batch job schedulers adopt the Compact-n-Exclusive (CE) strategy, packing processes of a parallel job into as few compute nodes as possible. While CE minimizes inter-node network communication, it often brings self-contention among tasks of a resource-intensive application. Recent studies have used virtual containers to balance CPU utilization and memory capacity across physical nodes, but the imbalance in cache and memory bandwidth usage is still under-investigated. Xiongchao Tang, Haojie Wang 0004, Xiaosong Ma, Nosayba El-Sayed, Jidong Zhai, Ashraf Aboulnaga |
SC | 5 |
| 2019 | Performance evaluation and analysis of sparse matrix and graph kernels on heterogeneous processors
Feng Zhang 0007, Weifeng Liu 0002, Ningxuan Feng, Jidong Zhai, Xiaoyong Du 0001 |
CCF Trans. High Perform. Comput. | 4 |
| 2019 | Student Cluster Competition 2018, Team Tsinghua University: Reproducing performance of multi-physics simulations of the Tsunamigenic 2004 Sumatra megathrust earthquake on the Intel Skylake Architecture
Jiaao He, Chenggang Zhao, Jiping Yu, Xinjian Yu, Liyan Zheng 0001, Chenyao Lou, Shizhi Tang, Jidong Zhai |
Parallel Comput. | 9 |
| 2019 | Privacy Regulation Aware Process Mapping in Geo-Distributed Cloud Data CentersabstractRecently, various applications including data analytics and machine learning have been developed for geo-distributed cloud data centers. For those applications, the ways of mapping parallel processes to physical nodes (i.e., “process mapping”) could significantly impact the performance of the applications because of non-uniform communication cost in geo-distributed environments. What's more, the different data privacy requirements in geo-distributed data centers pose additional constraints on process mapping solutions. While process mapping has been widely studied in grid/cluster environments, few of the existing studies have considered the problem in geo-distributed cloud environment, which is a challenging task due to the multi-level data privacy constraints, heterogeneous network performance and process failures. In this paper, we introduce the special privacy requirements in geo-distributed data centers and formulate the geo-distributed process mapping problem as an optimization problem with multiple constraints. We develop a new method to efficiently find good process mapping solutions to the problem. Experimental results on real clouds (including Amazon EC2 and Windows Azure) and simulations demonstrate that our proposed approach can achieve significant performance improvement compared to the state-of-the-art algorithms. Amelie Chi Zhou, Yifan Gong 0003, Bingsheng He, Jidong Zhai, Rui Mao 0001 |
IEEE Trans. Parallel Distributed Syst. | 5 |
| 2018 | Zwift: A Programming Framework for High Performance Text Analytics on Compressed DataabstractToday's rapidly growing document volumes pose pressing challenges to modern document analytics frameworks, in both space usage and processing time. Recently, a promising method, called text analytics directly on compressed data (TADOC), was proposed for improving both the time and space efficiency of text analytics. The main idea of the technique is to enable direct document analytics on compressed data. This paper focuses on the programming challenges for developing efficient TADOC programs. It presents Zwift, the first programming framework for TADOC, which consists of a Domain Specific Language, a compiler and runtime, and a utility library. Experiments show that Zwift significantly improves programming productivity, while effectively unleashing the power of TADOC, producing code that reduces storage usage by 90.8% and execution time by 41.0% on six text analytics problems. Feng Zhang 0007, Jidong Zhai, Xipeng Shen, Onur Mutlu |
ICS | 2 |
| 2018 | BitFlow: Exploiting Vector Parallelism for Binary Neural Networks on CPUabstractDeep learning has revolutionized computer vision and other fields since its big bang in 2012. However, it is challenging to deploy Deep Neural Networks (DNNs) into real-world applications due to their high computational complexity. Binary Neural Networks (BNNs) dramatically reduce computational complexity by replacing most arithmetic operations with bitwise operations. Existing implementations of BNNs have been focusing on GPU or FPGA, and using the conventional image-to-column method that doesn't perform well for binary convolution due to low arithmetic intensity and unfriendly pattern for bitwise operations. We propose BitFlow, a gemm-operator-network three-level optimization framework for fully exploiting the computing power of BNNs on CPU. BitFlow features a new class of algorithm named PressedConv for efficient binary convolution using locality-aware layout and vector parallelism. We evaluate BitFlow with the VGG network. On a single core of Intel Xeon Phi, BitFlow obtains 1.8x speedup over unoptimized BNN implementations, and 11.5x speedup over counterpart full-precision DNNs. Over 64 cores, BitFlow enables BNNs to run 1.1x faster than counterpart full-precision DNNs on GPU (GTX 1080). Jidong Zhai, Dinghua Li, Yifan Gong 0003, Yuhao Zhu 0001, Wei Liu 0143, Jiangming Jin |
IPDPS | 2 |
| 2018 | CSE: Parallel Finite State Machines with Convergence Set EnumerationabstractFinite State Machine (FSM) is known to be “embarrassingly sequential” because the next state depends on the current state and input symbol. Enumerative FSM breaks the data dependencies by cutting the input symbols into segments and processing all segments in parallel. With unknown starting state (except the first segment), each segment needs to calculate the state transitions, i.e., state state, for all states, each one is called an enumeration path. The current software and hardware implementations suffer from two drawbacks: 1) large amount of state state computation overhead for the enumeration paths; and 2) the optimizations are restricted by the need to correctly performing state state and only achieve limited improvements. This paper proposes CSE, a Convergence Set based Enumeration based parallel FSM. Unlike prior approaches, CSE is based on a novel computation primitive set(N) set(M), which maps N states to M states without giving the specific state state mappings (which state is mapped to which). The set(N) set(M) has two key properties: 1) if M is equal to 1, i.e., all N states are mapped to the same state, the state state for all the N states are computed; 2) using one-hot encoding, the hardware implementation cost of state state is the same as set(N) set(M). The convergence property ensures that M is always less than N. The key idea of CSE is to partition the original all S states into n state sets CS1,CS2,...,CSn, i.e., convergence sets. Using set(N) set(M) to process each CS, if the states converge to a single state, then we have successfully computed the enumeration path for each state in CS; otherwise, we may need to re-execute the stage when the outcome of the previous stage falls in CS. CSE is realized by two techniques: convergence set prediction, which generates the convergence sets with random input based profiling that maximizes the probability of each CS z converging to one state; global re-execution algorithm, which ensures the correctness by re-executing the non-converging stages with known input state. Essentially, CSE reformulates the enumeration paths as setbased rather than singleton-based. We evaluate CSE with 13 benchmarks. It achieved on average 2.0x/2.4x and maximum 8.6x/2.7x speedup compared to Lookback Enumeration (LBE) and Parallel Automata Processor (PAP), respectively. Youwei Zhuo, Jinglei Cheng, Qinyi Luo, Jidong Zhai, Yanzhi Wang 0001, Zhongzhi Luan, Xuehai Qian |
MICRO | 4 |
| 2018 | vSensor: leveraging fixed-workload snippets of programs for performance variance detectionabstractPerformance variance becomes increasingly challenging on current large-scale HPC systems. Even using a fixed number of computing nodes, the execution time of several runs can vary significantly. Many parallel programs executing on supercomputers suffer from such variance. Performance variance not only causes unpredictable performance requirement violations, but also makes it unintuitive to understand the program behavior. Despite prior efforts, efficient on-line detection of performance variance remains an open problem. Xiongchao Tang, Jidong Zhai, Xuehai Qian, Bingsheng He, Wei Xue 0003 |
PPoPP | 2 |
| 2018 | Spindle: Informed Memory Access Monitoring
Haojie Wang 0004, Jidong Zhai, Xiongchao Tang, Bowen Yu 0003, Xiaosong Ma |
USENIX ATC | 2 |
| 2018 | A vision of post-exascale programmingabstractExascale systems have been under development for quite some time and will be available for use in a few years. It is time to think about future post-exascale systems. There are many main challenges with regard to future post-exascale systems, such as processor architecture, programming, storage, and interconnect. In this study, we discuss three significant programming challenges for future post-exascale systems: heterogeneity, parallelism, and fault tolerance. Based on our experience of programming on current large-scale systems, we propose several potential solutions for these challenges. Nevertheless, more research efforts are needed to solve these problems. Jidong Zhai, Wen-Guang Chen |
Frontiers Inf. Technol. Electron. Eng. | 1 |
| 2018 | Student cluster competition 2017, team Tsinghua University: Reproducing vectorization of the tersoff multi-body potential on the Intel Skylake and NVIDIA Volta architectures
Ka Cheong Jason Lau, Qian Xie 0005, Beichen Li 0005, Guanyu Feng, Jiping Yu, Xinjian Yu, Jidong Zhai |
Parallel Comput. | 12 |
| 2018 | Efficient Document Analytics on Compressed Data: Method, Challenges, Algorithms, InsightsabstractToday's rapidly growing document volumes pose pressing challenges to modern document analytics, in both space usage and processing time. In this work, we propose the concept of compression-based direct processing to alleviate issues in both dimensions. The main idea is to enable direct document analytics on compressed data. We present how the concept can be materialized on Sequitur, a compression algorithm that produces hierarchical grammar-like representations. We discuss the major challenges in applying the idea to various document analytics tasks, and reveal a set of guidelines and also assistant software modules for developers to effectively apply compression-based direct processing . Experiments show that our proposed techniques save 90.8% storage space and 77.5% memory usage, while speeding up data processing significantly, i.e., by 1.6X on sequential systems, and 2.2X on distributed clusters, on average. Feng Zhang 0007, Jidong Zhai, Xipeng Shen, Onur Mutlu |
Proc. VLDB Endow. | 2 |
| 2018 | An adaptive breadth-first search algorithm on integrated architectures
Feng Zhang 0007, Heng Lin, Jidong Zhai, Dingyi Xiang, Jizhong Li, Yunpeng Chai, Xiaoyong Du 0001 |
J. Supercomput. | 3 |
| 2018 | An Efficient In-Memory Checkpoint Method and its Practice on Fault-Tolerant HPLabstractFault tolerance is increasingly important in high-performance computing due to the substantial growth of system scale and decreasing system reliability. In-memory/diskless checkpoint has gained extensive attention as a solution to avoid the IO bottleneck of traditional disk-based checkpoint methods. However, applications using previous in-memory checkpoint suffer from little available memory space. To provide high reliability, previous in-memory checkpoint methods either need to keep two copies of checkpoints to tolerate failures while updating old checkpoints or trade performance for space by flushing in-memory checkpoints into disk. In this paper, we propose a novel in-memory checkpoint method, called self-checkpoint, which can not only achieve the same reliability of previous in-memory checkpoint methods, but also increase the available memory space for applications by almost 50 percent. To validate our method, we apply self-checkpoint method to an important problem: High-Performance Linpack (HPL) with fault tolerance. We implement a scalable and fault tolerant HPL based on this new method, called SKT-HPL, and validate it on two large-scale systems. Experimental results with 24,576 processes show that SKT-HPL achieves over 95 percent of the performance of the original HPL. Compared to the state-of-the-art in-memory checkpoint method, it improves the available memory size by 47 percent and the performance by 5 percent. Xiongchao Tang, Jidong Zhai, Bowen Yu 0003, Keqin Li 0001 |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2017 | FinePar: irregularity-aware fine-grained workload partitioning on integrated architectures
Feng Zhang 0007, Bo Wu 0002, Jidong Zhai, Bingsheng He |
CGO | 3 |
| 2017 | Algorithm-Directed Crash Consistence in Non-volatile Memory for HPCabstractFault tolerance is one of the major design goals for HPC. The emergence of non-volatile memories (NVM) provides a solution to build fault tolerant HPC. Data in NVM-based main memory are not lost when the system crashes because of the non-volatility nature of NVM. However, because of volatile caches, data must be logged and explicitly flushed from caches into NVM to ensure consistence and correctness before crashes, which can cause large runtime overhead. In this paper, we introduce an algorithm-based method to establish crash consistence in NVM for HPC applications. We slightly extend application data structures or sparsely flush cache blocks, which introduce ignorable runtime overhead. Such extension or cache flushing allows us to use algorithm knowledge to reason data consistence or correct inconsistent data when the application crashes. We demonstrate the effectiveness of our method for three algorithms, including an iterative solver, dense matrix multiplication, and Monte-Carlo simulation. Based on comprehensive performance evaluation on a variety of test environments, we demonstrate that our approach has very small runtime overhead (at most 8.2% and less than 3% in most cases), much smaller than that of traditional checkpoint, while having the same or less recomputation cost. Shuo Yang 0011, Kai Wu 0006, Yifan Qiao 0002, Dong Li 0001, Jidong Zhai |
CLUSTER | 5 |
| 2017 | Scalable Graph Traversal on Sunway TaihuLight with Ten Million CoresabstractInterest has recently grown in efficiently analyzing unstructured data such as social network graphs and protein structures. A fundamental graph algorithm for doing such task is the Breadth-First Search (BFS) algorithm, the foundation for many other important graph algorithms such as calculating the shortest path or finding the maximum flow in graphs. In this paper, we share our experience of designing and implementing the BFS algorithm on Sunway TaihuLight, a newly released machine with 40,960 nodes and 10.6 million accelerator cores. It tops the Top500 list of June 2016 with a 93.01 petaflops Linpack performance [1]. Designed for extremely large-scale computation and power efficiency, processors on Sunway TaihuLight employ a unique heterogeneous many-core architecture and memory hierarchy. With its extremely large size, the machine provides both opportunities and challenges for implementing high-performance irregular algorithms, such as BFS. We propose several techniques, including pipelined module mapping, contention-free data shuffling, and group-based message batching, to address the challenges of efficiently utilizing the features of this large scale heterogeneous machine. We ultimately achieved 23755.7 giga-traversed edges per second (GTEPS), which is the best among heterogeneous machines and the second overall in the Graph500s June 2016 list [2]. Heng Lin, Xiongchao Tang, Bowen Yu 0003, Youwei Zhuo, Jidong Zhai, Wanwang Yin |
IPDPS | 6 |
| 2017 | Versapipe: a versatile programming framework for pipelined computing on GPUabstractPipeline is an important programming pattern, while GPU, designed mostly for data-level parallel executions, lacks an efficient mechanism to support pipeline programming and executions. This paper provides a systematic examination of various existing pipeline execution models on GPU, and analyzes their strengths and weaknesses. To address their shortcomings, this paper then proposes three new execution models equipped with much improved controllability, including a hybrid model that is capable of getting the strengths of all. These insights ultimately lead to the development of a software programming framework named VersaPipe. With VersaPipe, users only need to write the operations for each pipeline stage. VersaPipe will then automatically assemble the stages into a hybrid execution model and configure it to achieve the best performance. Experiments on a set of pipeline benchmarks and a real-world face detection application show that VersaPipe produces up to 6.90X (2.88X on average) speedups over the original manual implementations. Zhen Zheng, Chanyoung Oh, Jidong Zhai, Xipeng Shen, Youngmin Yi |
MICRO | 3 |
| 2017 | Self-Checkpoint: An In-Memory Checkpoint Method Using Less Space and Its Practice on Fault-Tolerant HPLabstractFault tolerance is increasingly important in high performance computing due to the substantial growth of system scale and decreasing system reliability. In-memory/diskless checkpoint has gained extensive attention as a solution to avoid the IO bottleneck of traditional disk-based checkpoint methods. However, applications using previous in-memory checkpoint suffer from little available memory space. To provide high reliability, previous in-memory checkpoint methods either need to keep two copies of checkpoints to tolerate failures while updating old checkpoints or trade performance for space by flushing in-memory checkpoints into disk. Xiongchao Tang, Jidong Zhai, Bowen Yu 0003 |
PPoPP | 2 |
| 2017 | Efficient process mapping in geo-distributed cloud data centersabstractRecently, various applications including data analytics and machine learning have been developed for geo-distributed cloud data centers. For those applications, the ways to map parallel processes to physical nodes (i.e., "process mapping") could significantly impact the performance of the applications because of non-uniform communication cost in such geo-distributed environments. While process mapping has been widely studied in grid/cluster environments, few of the existing studies have considered the problem in geo-distributed cloud environments. In this paper, we propose a novel model to formulate the geo-distributed process mapping problem and develop a new method to efficiently find the near optimal solution. Our algorithm considers both the network communication performance of geo-distributed data centers as well as the communication matrix of the target application. Evaluation results with real experiments on Amazon EC2 and simulations demonstrate that our proposal achieves significant performance improvement (50% on average) compared to the state-of-the-art algorithms. Amelie Chi Zhou, Yifan Gong 0003, Bingsheng He, Jidong Zhai |
SC | 4 |
| 2017 | Understanding Co-Running Behaviors on Integrated CPU/GPU ArchitecturesabstractArchitecture designers tend to integrate both CPUs and GPUs on the same chip to deliver energy-efficient designs. It is still an open problem to effectively leverage the advantages of both CPUs and GPUs on integrated architectures. In this work, we port 42 programs in Rodinia, Parboil, and Polybench benchmark suites and analyze the co-running behaviors of these programs on both AMD and Intel integrated architectures. We find that co-running performance is not always better than running the program only with CPUs or GPUs. Among these programs, only eight programs can benefit from the co-running, while 24 programs only using GPUs and seven programs only using CPUs achieve the best performance. The remaining three programs show little performance preference for different devices. Through extensive workload characterization analysis, we find that architecture differences between CPUs and GPUs and limited shared memory bandwidth are two main factors affecting current co-running performance. Since not all the programs can benefit from integrated architectures, we build an automatic decision-tree-based model to help application developers predict the co-running performance for a given CPU-only or GPU-only program. Results show that our model correctly predicts 14 programs out of 15 for evaluated programs. For a co-run friendly program, we further propose a profiling-based method to predict the optimal workload partition ratio between CPUs and GPUs. Results show that our model can achieve 87.7 percent of the optimal performance relative to the best partition. The co-running programs acquired with our method outperform the original CPU-only and GPU-only programs by 34.5 and 20.9 percent respectively. Feng Zhang 0007, Jidong Zhai, Bingsheng He, Shuhao Zhang 0001 |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2016 | A Fast Tridiagonal Solver for Intel MIC ArchitectureabstractThe tridiagonal solver is an important kernel in many scientific and engineering applications. Although quite a few parallel algorithms have been exploited recently, challenges still remain when solving tridiagonal systems on many-core architectures. In this paper, quantitative analysis is conducted to guide the selection of algorithms on different architectures, and a fast register-PCR-pThomas algorithm for Intel MIC architecture is presented to achieve the best utilization of in-core vectorization and registers. By choosing the most suitable number of PCR reductions, we further propose an improved algorithm, named register-PCR-half-pThomas algorithm, which minimizes the computational cost and the number of registers for use. The optimizations of manual prefetching and strength reduction are also discussed for the new algorithm. Evaluation on Intel Xeon Phi 7110P shows that our register-PCR-pThomas can outperform the MKL solver by 7.7 times in single precision and 3.4 times in double precision. Moreover, our optimized register-PCR-half-pThomas can further boost the performance by another 43.9% in single precision and 20.4% in double precision. Our best algorithm can outperform the latest official GPU library (cuSPARSE) on Nvidia K40 by 3.7 times and 2.4 times in single and double precision, respectively. Wei Xue 0003, Jidong Zhai, Yangtong Xu, Hai-Xiang Lin |
IPDPS | 3 |
| 2016 | Characterizing and optimizing TPC-C workloads on large-scale systems using SSD arrays
Jidong Zhai, Feng Zhang 0007 |
Sci. China Inf. Sci. | 1 |
| 2016 | A survey of cloud resource management for complex engineering applications
Haibao Chen, Song Wu 0001, Hai Jin 0001, Jidong Zhai, Yingwei Luo, Xiaolin Wang 0001 |
Frontiers Comput. Sci. | 5 |
| 2016 | Performance Prediction for Large-Scale Parallel Applications Using Representative ReplayabstractAutomatically predicting performance of parallel applications has been a long-standing goal in the area of high performance computing. However, accurate performance prediction is challenging, since the execution time of parallel applications is determined by several factors, such as sequential computation time, communication time and their complex interactions. Despite previous efforts, accurately estimating the sequential computation time in each process for large-scale parallel applications remains an open problem. In this paper, we propose a novel approach to acquiring accurate sequential computation time using a parallel debugging technique called deterministic replay. The main advantage of our approach is that we only need a single node of a target platform but the whole target platform does not need to be available. Therefore, with this approach we can simply measure the real sequential computation time on a target node for each process on by one. Moreover, we observe that there is great computation similarity in parallel applications, not only within each process but also among different processes. Based on this observation, we further propose representative replay that can significantly reduce replay overhead, because we only need to replay partial iterations for representative processes instead of all of them. Finally, we implement a complete performance prediction system, called Phantom, which combines the above computation-time acquisition approach and a trace-driven simulator. We validate our approach on both traditional HPC platforms and the latest Amazon EC2 cloud platform. On both types of platforms, prediction error of our approach is less than 7 percent on average up to 2,500 processes. Jidong Zhai, Keqin Li 0001 |
IEEE Trans. Computers | 1 |
| 2016 | Building Semi-Elastic Virtual Clusters for Cost-Effective HPC Cloud Resource ProvisioningabstractRecent studies have found cloud environments increasingly appealing for executing HPC applications, including tightly coupled parallel simulations. At the same time, while public clouds offer elastic, on-demand resource provisioning and pay-as-you-go pricing, individual users setting up their on-demand virtual clusters may not be able to take full advantage of common cost-saving opportunities, such as reserved instances. In this paper, we propose a Semi-Elastic Cluster (SEC) computing model for organizations to reserve and dynamically resize a virtual cloud-based cluster. We present a set of integrated batch scheduling plus resource scaling strategies uniquely enabled by SEC, as well as an online reserved instance provisioning algorithm based on job history. Our trace-driven simulation results show that such a model has a 61.0 percent cost saving than individual users acquiring and managing cloud resources without causing longer average job wait time. Moreover, to exploit the advantages of different public clouds, we also extend SEC to a multi-cloud environment, where SEC can get a lower cost than on any single cloud. We design and implement a prototype system of the SEC model and evaluate it in terms of management overhead and average job wait time. Experimental results show that the management overhead is negligible with respect to the job wait time. Shuangcheng Niu, Jidong Zhai, Xiaosong Ma, Xiongchao Tang |
IEEE Trans. Parallel Distributed Syst. | 2 |
| 2015 | A Power-Conserving Online Scheduling Scheme for Video Streaming Services
Yunyun Jiang, Tian Xiao, Jidong Zhai |
ICA3PP (1) | 3 |
| 2015 | To Co-run, or Not to Co-run: A Performance Study on Integrated ArchitecturesabstractArchitecture designers tend to integrate both CPU and GPU on the same chip to deliver energy-efficient designs. To effectively leverage the power of both CPUs and GPUs on integrated architectures, researchers have recently put substantial efforts into co-running a single application on both the CPU and the GPU of such architectures. However, few studies have been performed to analyze a wide range of parallel computation patterns on such architectures. In this paper, we port all programs in Rodinia benchmark suite and co-run these programs on the integrated architecture. We find that co-running results are not always better than running the application on the CPU only or the GPU only. Among the 20 programs, 3 programs can benefit from co-running, 12 programs using GPU only and 2 programs using CPU only achieve the best performance. The remaining 3 programs show no performance preference for different devices. We also characterize the workload and summarize the patterns for the system insights of co-running on integrated architectures. Feng Zhang 0007, Jidong Zhai, Bingsheng He, Shuhao Zhang 0001 |
MASCOTS | 2 |
| 2015 | Optimizing seam carving on multi-GPU systems for real-time content-aware image resizing
Ikjoon Kim, Jidong Zhai |
J. Supercomput. | 2 |
| 2015 | Automatic Cloud I/O Configurator for I/O Intensive Parallel ApplicationsabstractAs the cloud platform becomes a promising alternative to traditional HPC (high performance computing) centers or in-house clusters, the I/O bottleneck problem is highlighted in this new environment, typically with top-of-the-line compute instances but sub-par communication and I/O facilities. It has been observed that changing the cloud I/O system configurations, such as choices of file systems, number of I/O servers and their placement strategies, etc., will lead to a considerable variation in the performance and cost efficiency of I/O intensive parallel applications. However, storage system configuration is tedious and error-prone to do manually, even for expert users, leading to solutions that are grossly over-provisioned (low cost inefficiency), substantially under-performing (poor performance) or, in the worst case, both. This paper proposes ACIC, a system which automatically searches for optimized I/O system configurations from many candidates for each individual application running on a given cloud platform. ACIC takes advantage of machine learning models to perform performance/cost predictions. To tackle the high-dimensional parameter exploration space, we enable affordable, reusable, and incremental training on cloud platforms, guided by the Plackett and Burman Matrices for experiment design. Our evaluation results with four representative parallel applications indicate that ACIC consistently identifies optimal or near-optimal configurations among a large group of candidate settings. The top ACIC-recommended configuration is capable of improving the applications' performance by a factor of up to 10.5 (3.1 on average), and cost saving of up to 89 percent (51 percent on average), compared with a commonly used baseline I/O configuration. In addition, we carried out a small-scale user study for one of the test applications, which found that ACIC consistently beat the user and even the application's developer, often by a significant margin, in selecting optimized configurations. Jidong Zhai, Xiaosong Ma |
IEEE Trans. Parallel Distributed Syst. | 1 |
| 2014 | Optimizing Seam Carving on multi-GPU systems for real-time image resizingabstractImage resizing is increasingly important for picture sharing and exchanging between various personal electronic equipments. Seam Carving is a state-of-the-art approach for effective image resizing because of its content-aware characteristic. However, complex computation and memory access patterns make it time-consuming and prevent its wide usage in real-time image processing. To address these problems, we propose a novel algorithm, called Non-Cumulative Seam Carving (NCSC), which removes main computation bottleneck. Furthermore, we also propose an adaptive multi-seam algorithm for better parallelism on GPU platforms. Finally, we implement our algorithm on a multi-GPU platform. Results show that our approach achieves a maximum 140× speedup on a two-GPU system over the sequential version. It only takes 0.11 second to resize a 1024×640 image by half in width compared to 15.5 seconds with the traditional seam carving. Ikjoon Kim, Jidong Zhai |
ICPADS | 2 |
| 2014 | CYPRESS: Combining Static and Dynamic Analysis for Top-Down Communication Trace CompressionabstractCommunication traces are increasingly important, both for parallel applications' performance analysis/optimization, and for designing next-generation HPC systems. Meanwhile, the problem size and the execution scale on supercomputers keep growing, producing prohibitive volume of communication traces. To reduce the size of communication traces, existing dynamic compression methods introduce large compression overhead with the job scale. We propose a hybrid static-dynamic method that leverages information acquired from static analysis to facilitate more effective and efficient dynamic trace compression. Our proposed scheme, Cypress, extracts a program communication structure tree at compile time using inter-procedural analysis. This tree naturally contains crucial iterative computing features such as the loop structure, allowing subsequent runtime compression to "fill in", in a "top-down" manner, event details into the known communication template. Results show that Cypress reduces intra-process and inter-process compression overhead up to 5× and 9× respectively over state-of-the-art dynamic methods, while only introducing very low compiling overhead. Jidong Zhai, Jianfei Hu, Xiongchao Tang, Xiaosong Ma |
SC | 1 |
| 2013 | ACIC: automatic cloud I/O configurator for parallel applications
Jidong Zhai, Yan Zhai, Xiaosong Ma |
HPDC | 3 |
| 2013 | ACIC: automatic cloud I/O configurator for HPC applicationsabstractThe cloud has become a promising alternative to traditional HPC centers or in-house clusters. This new environment highlights the I/O bottleneck problem, typically with top-of-the-line compute instances but sub-par communication and I/O facilities. It has been observed that changing cloud I/O system configurations leads to significant variation in the performance and cost efficiency of I/O intensive HPC applications. However, storage system configuration is tedious and error-prone to do manually, even for experts. Jidong Zhai, Yan Zhai, Xiaosong Ma |
SC | 3 |
| 2013 | Cost-effective cloud HPC resource provisioning by building semi-elastic virtual clustersabstractRecent studies have found cloud environments increasingly appealing for executing HPC applications, including tightly coupled parallel simulations. While public clouds offer elastic, on-demand resource provisioning and pay-as-you-go pricing, individual users setting up their on-demand virtual clusters may not be able to take full advantage of common cost-saving opportunities, such as reserved instances. Shuangcheng Niu, Jidong Zhai, Xiaosong Ma, Xiongchao Tang |
SC | 2 |
| 2012 | Employing Checkpoint to Improve Job Scheduling in Large-Scale Systems
Shuangcheng Niu, Jidong Zhai, Xiaosong Ma, Yan Zhai |
JSSPP | 2 |
| 2011 | Efficiently Acquiring Communication Traces for Large-Scale Parallel ApplicationsabstractCommunication patterns of parallel applications are important to optimize application performance and design better communication subsystems. Communication patterns can be extracted from communication traces. However, existing approaches to generate communication traces need to execute the entire parallel applications on full-scale systems that are time consuming and expensive. We propose a novel technique, called Fact, which can perform FAst Communication Traces collection for large-scale parallel applications on small-scale systems. Our idea is to reduce the original program to obtain a program slice through static analysis, and to execute the program slice to acquire the communication traces. Our idea is based on an observation that most computation and message contents in parallel applications are not relevant to their spatial and volume communication attributes, and therefore can be removed for the purpose of communication trace collection. We have implemented Fact and evaluated it with NPB programs and Sweep3D. The results show that Fact can reduce resource consumptions by two orders of magnitude in most cases. Jidong Zhai, Tianwei Sheng, Jiangzhou He |
IEEE Trans. Parallel Distributed Syst. | 1 |
| 2010 | PHANTOM: predicting performance of parallel applications on large-scale parallel machines using a single nodeabstractFor designers of large-scale parallel computers, it is greatly desired that performance of parallel applications can be predicted at the design phase. However, this is difficult because the execution time of parallel applications is determined by several factors, including sequential computation time in each process, communication time and their convolution. Despite previous efforts, it remains an open problem to estimate sequential computation time in each process accurately and efficiently for large-scale parallel applications on non-existing target machines. Jidong Zhai |
PPoPP | 1 |
| 2009 | Process Mapping for MPI Collective Communications
Jidong Zhai |
Euro-Par | 2 |
| 2009 | FACT: fast communication trace collection for parallel applications through program slicingabstractA proper understanding of communication patterns of parallel applications is important to optimize application performance and design better communication subsystems. Communication patterns can be obtained by analyzing communication traces. However, existing approaches to generate communication traces need to execute the entire parallel applications on full-scale systems that are time-consuming and expensive. Jidong Zhai, Tianwei Sheng, Jiangzhou He |
SC | 1 |
| 2009 | LogGPO: An accurate communication model for performance prediction of MPI programs
Jidong Zhai |
Sci. China Ser. F Inf. Sci. | 2 |