02/02/2021

OpEvo: An Evolutionary Method for Tensor Operator Optimization

Xiaotian Gao, Wei Cui, Lintao Zhang, Mao Yang

Keywords:

Abstract Paper Similar Papers

Abstract: Training and inference efficiency of deep neural networks highly rely on the performance of tensor operators on hardware platforms. Manually optimizing tensor operators has limitations in terms of supporting new operators or hardware platforms. Therefore, automatically optimizing device code configurations of tensor operators is getting increasingly attractive. However, current methods for tensor operator optimization usually suffer from poor sample-efficiency due to the combinatorial search space. In this work, we propose a novel evolutionary method, OpEvo, which efficiently explores the search spaces of tensor operators by introducing a topology-aware mutation operation based on q-random walk to leverage the topological structures over the search spaces. Our comprehensive experiment results show that compared with state-of-the-art(SOTA) methods OpEvo can find the best configuration with the lowest variance and least efforts in the number of trials and wall-clock time. All code of this work is available online.

The video of this talk cannot be embedded. You can watch it here:
https://slideslive.com/38949362
(Link will open in new window)
 0
 0
 0
 0
  Share    
This is an embedded video. Talk and the respective paper are published at AAAI 2021 virtual conference. If you are one of the authors of the paper and want to manage your upload, see the question "My papertalk has been externally embedded..." in the FAQ section.

Comments

Post Comment
no comments yet
code of conduct: tbd Characters remaining: 140

Similar Papers

 0
 0
 0
 0
 20:10 
 0
 0
 0
 0
 5:01 
 0
 0
 0
 0
 4:59 
 0
 0
 0
 0
 12:53 
 0
 0
 0
 0
 1:00 
 0
 0
 0
 1
 20:37 
 0
 0
 0
 0
 11:59 
 0
 0
 0
 0
 13:21 
 0
 0
 0
 0
 16:07 
 0
 0
 0
 0
 5:07 
 0
 0
 0
 0
 3:19 
 0
 0
 0
 0
 13:50 
 0
 0
 0
 0
 16:53 
 0
 0
 0
 0
 14:38 
 0
 0
 0
 0
 14:59 
 0
 0
 0
 0
 4:53 
 0
 0
 0
 0
 5:11 
 1
 0
 0
 0
 14:01 
 0
 0
 0
 0
 4:52 
 0
 0
 0
 0
 14:56 
 0
 0
 0
 0
 16:43 
 0
 0
 0
 0
 12:16 
 0
 0
 0
 0
 3:20 
 0
 0
 0
 0
 4:56 
 0
 0
 0
 0
 5:01 
 0
 0
 0
 0
 3:16 
 0
 0
 0
 0
 15:48 
 0
 0
 0
 0
 5:19 
 0
 0
 0
 0
 3:16 
 0
 0
 0
 0
 12:58 
 0
 0
 0
 0
 20:19 
 0
 0
 0
 0
 0:59 
 0
 0
 0
 0
 10:00 
 0
 0
 0
 0
 9:24 
 0
 0
 0
 0
 6:16 
 1
 0
 0
 0
 3:19 
 0
 0
 0
 0
 4:36 
 0
 0
 0
 0
 14:26 
 0
 0
 0
 0
 4:21 
 0
 0
 0
 0
 5:09