activity
20232026
most citedCatch-Up Distillation: You Only Need to Train Once for Accelerating Sampling

8 citations · 9 across the 9 of their papers we have counts for

collaborators

10 papers

cs.AR2026

Fovea: Physical-Implication-Aware Wafer-Scale DSE with Decision-Domain-Guided Cross-Fidelity Refinement

Jinxi Li, Huizheng Wang, Jinyi Deng +2

Modern pre-silicon design-space exploration (DSE) follows a coarse-to-fine workflow: low-cost evaluators screen candidate spaces, while detailed evaluation is reserved for a shortl…

cs.AR2026

MOCAP: Wafer-Scale-Chip-Oriented Memory-Orchestrated Chunked Pipelining Framework for Prefill-Only LLM Inference

Zichuan Wang, Huizheng Wang, Yuheng Xiao +6

Large language models (LLMs) are increasingly used in prefill-only workloads, where end-to-end latency is dominated by the prefill phase. For long-context prefill, communication ov…

cs.AR2025

Designing Spatial Architectures for Sparse Attention: STAR Accelerator via Cross-Stage Tiling

Huizheng Wang, Taiquan Wei, Hongbin Wang +6

Large language models (LLMs) rely on self-attention for contextual understanding, demanding high-throughput inference and large-scale token parallelism (LTPP). Existing dynamic spa…

cs.AI2025

cuPilot: A Strategy-Coordinated Multi-agent Framework for CUDA Kernel Evolution

Jinwu Chen, Qidie Wu, Bin Li +5

Optimizing CUDA kernels is a challenging and labor-intensive task, given the need for hardware-software co-design expertise and the proprietary nature of high-performance kernel li…

cs.DC2025

HPCTransCompile: An AI Compiler Generated Dataset for High-Performance CUDA Transpilation and LLM Preliminary Exploration

Jiaqi Lv, Xufeng He, Yanchen Liu +7

The rapid growth of deep learning has driven exponential increases in model parameters and computational demands. NVIDIA GPUs and their CUDA-based software ecosystem provide robust…

cs.AI2024

Tackling the Dynamicity in a Production LLM Serving System with SOTA Optimizations via Hybrid Prefill/Decode/Verify Scheduling on Efficient Meta-kernels

Mingcong Song, Xinru Tang, Fengfan Hou +9

Meeting growing demands for low latency and cost efficiency in production-grade large language model (LLM) serving systems requires integrating advanced optimization techniques. Ho…

cs.DC2024

PALM: A Efficient Performance Simulator for Tiled Accelerators with Large-scale Model Training

Jiahao Fang, Huizheng Wang, Qize Yang +5

Deep learning (DL) models are piquing high interest and scaling at an unprecedented rate. To this end, a handful of tiled accelerators have been proposed to support such large-scal…

cs.AI2024

Efficient Orchestrated AI Workflows Execution on Scale-out Spatial Architecture

Jinyi Deng, Xinru Tang, Zhiheng Yue +8

Given the increasing complexity of AI applications, traditional spatial architectures frequently fall short. Our analysis identifies a pattern of interconnected, multi-faceted task…

cs.AR2023★ 1 cited

Wafer-scale Computing: Advancements, Challenges, and Future Perspectives

Yang Hu, Xinhan Lin, Huizheng Wang +12

Nowadays, artificial intelligence (AI) technology with large models plays an increasingly important role in both academia and industry. It also brings a rapidly increasing demand f…

cs.LG2023★ 8 cited

Catch-Up Distillation: You Only Need to Train Once for Accelerating Sampling

Shitong Shao, Xu Dai, Lujun Li +3

Diffusion Probability Models (DPMs) have made impressive advancements in various machine learning domains. However, achieving high-quality synthetic samples typically involves perf…