File tree Expand file tree Collapse file tree 4 files changed +611
-154
lines changed
tests/python/frontend/nnvm_to_relay
topi/python/topi/intel_graphics Expand file tree Collapse file tree 4 files changed +611
-154
lines changed Original file line number Diff line number Diff line change 3636
3737# the version of each package
3838PACKAGE_VERSION = {
39- 'arm_cpu' : "v0.04" ,
40- 'llvm' : "v0.03" ,
39+ 'arm_cpu' : "v0.04" ,
40+ 'llvm' : "v0.03" ,
4141
42- 'cuda' : "v0.05" ,
43- 'rocm' : "v0.03" ,
44- 'opencl' : "v0.03" ,
45- 'mali' : "v0.05" ,
42+ 'cuda' : "v0.05" ,
43+ 'rocm' : "v0.03" ,
44+ 'opencl' : "v0.03" ,
45+ 'mali' : "v0.05" ,
46+ 'intel_graphics' : "v0.01" ,
4647
47- 'vta' : "v0.06" ,
48+ 'vta' : "v0.06" ,
4849}
4950
5051logger = logging .getLogger ('autotvm' )
Original file line number Diff line number Diff line change @@ -69,6 +69,7 @@ def convnet():
6969 targets = ['cuda' ,
7070 'opencl -device=mali' ,
7171 'opencl -device=intel_graphics' ,
72+
7273 'llvm -device=arm_cpu' ,
7374 'llvm -device=core-avx-ii' ]
7475
@@ -83,5 +84,6 @@ def convnet():
8384 assert not relay .analysis .alpha_equal (N , O )
8485
8586if __name__ == "__main__" :
87+ import numpy as np
8688 np .random .seed (42 )
8789 test_alter_layout_conv2d ()
You can’t perform that action at this time.
0 commit comments