Skip to content

Commit 03519a3

Browse files
Laurawlywweic
authored andcommitted
[TOPI] Intel graphics conv2d autotvm template added (apache#3839)
* update lint * lint fixed * lint updated * lint fixed * lint fixed * lint fixed * updates * add intel graphics as a package * remove print info * depthwise conv2d schedule added for intel graphics * asdf * fix lint * fix lint * fix ci * add channels
1 parent 4a5f339 commit 03519a3

File tree

4 files changed

+611
-154
lines changed

4 files changed

+611
-154
lines changed

python/tvm/autotvm/tophub.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -36,15 +36,16 @@
3636

3737
# the version of each package
3838
PACKAGE_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

5051
logger = logging.getLogger('autotvm')

tests/python/frontend/nnvm_to_relay/test_alter_conv2d.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff 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

8586
if __name__ == "__main__":
87+
import numpy as np
8688
np.random.seed(42)
8789
test_alter_layout_conv2d()

0 commit comments

Comments
 (0)