[TOPI] Intel graphics conv2d autotvm template added (#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
apache · Sep 7, 2019 · 70042b7 · 70042b7
1 parent 02ddb5a
commit 70042b7
Show file tree

Hide file tree

Showing 4 changed files with 677 additions and 163 deletions.
diff --git a/python/tvm/autotvm/tophub.py b/python/tvm/autotvm/tophub.py
@@ -36,15 +36,16 @@
 
 # the version of each package
 PACKAGE_VERSION = {
-    'arm_cpu': "v0.04",
-    'llvm':    "v0.03",
+    'arm_cpu':          "v0.04",
+    'llvm':             "v0.03",
 
-    'cuda':    "v0.05",
-    'rocm':    "v0.03",
-    'opencl':  "v0.03",
-    'mali':    "v0.05",
+    'cuda':             "v0.05",
+    'rocm':             "v0.03",
+    'opencl':           "v0.03",
+    'mali':             "v0.05",
+    'intel_graphics':   "v0.01",
 
-    'vta':     "v0.06",
+    'vta':              "v0.06",
 }
 
 logger = logging.getLogger('autotvm')

diff --git a/tests/python/frontend/nnvm_to_relay/test_alter_conv2d.py b/tests/python/frontend/nnvm_to_relay/test_alter_conv2d.py
@@ -69,6 +69,7 @@ def convnet():
     targets=['cuda',
              'opencl -device=mali',
              'opencl -device=intel_graphics',
+
              'llvm -device=arm_cpu',
              'llvm -device=core-avx-ii']
 
@@ -83,5 +84,6 @@ def convnet():
                 assert not relay.analysis.alpha_equal(N, O)
 
 if __name__ == "__main__":
+    import numpy as np
     np.random.seed(42)
     test_alter_layout_conv2d()