GEVO-ML: a proposal for optimizing ML code with evolutionary computation
GEVO-ML: a proposal for optimizing ML code with evolutionary computation
复制标题
DOI:
10.1145/3377929.3398139
复制
发表时间:
2020-07
期刊:
影响因子:
--
通讯作者:
Jhe-Yu Liou;Xiaodong Wang;S. Forrest;Carole-Jean Wu
中科院分区:
文献类型:
--
作者:
Jhe-Yu Liou;Xiaodong Wang;S. Forrest;Carole-Jean Wu
Parallel accelerators, such as GPUs, are a key enabler of large-scale Machine Learning (ML) applications. However, programmers often lack detailed knowledge of the underlying architecture and fail to fully leverage their computational power. This paper proposes GEVO-ML, a tool for automatically discovering optimization opportunities and tuning the performance of ML kernels. GEVO-ML extends earlier work on GEVO (Gpu optimization using EVOlutionary computation) by focusing directly on ML frameworks, intermediate languages, and target architectures. It retains the multi-objective evolutionary search developed for GEVO, which searches for edits to GPU code compiled to LLVM-IR and improves performance on desired criteria while retaining required functionality. In earlier work, we studied some ML workloads in GPU settings and found that GEVO could improve kernel speeds by factors ranging from 1.7X to 2.9X, even with access to only a small portion of the overall ML framework. This workshop paper examines the limitations and constraints of GEVO for ML workloads and discusses our GEVO-ML design, which we are currently implementing.