7 citations · 14 across the 5 of their papers we have counts for
Showing 2022Show all
2 papers · 1 filter
cs.DC2022★ 3 cited
ALCOP: Automatic Load-Compute Pipelining in Deep Learning Compiler for AI-GPUs
Guyue Huang, Yang Bai, Liu Liu +4
Pipelining between data loading and computation is a critical tensor program optimization for GPUs. In order to unleash the high performance of latest GPUs, we must perform a syner…
cs.LG2022★ 2 cited
Moses: Efficient Exploitation of Cross-device Transferable Features for Tensor Program Optimization
Zhihe Zhao, Xian Shuai, Yang Bai +4
Achieving efficient execution of machine learning models has attracted significant attention recently. To generate tensor programs efficiently, a key component of DNN compilers is…