Commit 70042b78 by Leyuan Wang Committed by Yao Wang

[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
parent 02ddb5a9
...@@ -43,6 +43,7 @@ PACKAGE_VERSION = { ...@@ -43,6 +43,7 @@ PACKAGE_VERSION = {
'rocm': "v0.03", 'rocm': "v0.03",
'opencl': "v0.03", 'opencl': "v0.03",
'mali': "v0.05", 'mali': "v0.05",
'intel_graphics': "v0.01",
'vta': "v0.06", 'vta': "v0.06",
} }
......
...@@ -69,6 +69,7 @@ def test_alter_layout_conv2d(): ...@@ -69,6 +69,7 @@ def test_alter_layout_conv2d():
targets=['cuda', targets=['cuda',
'opencl -device=mali', 'opencl -device=mali',
'opencl -device=intel_graphics', 'opencl -device=intel_graphics',
'llvm -device=arm_cpu', 'llvm -device=arm_cpu',
'llvm -device=core-avx-ii'] 'llvm -device=core-avx-ii']
...@@ -83,5 +84,6 @@ def test_alter_layout_conv2d(): ...@@ -83,5 +84,6 @@ def test_alter_layout_conv2d():
assert not relay.analysis.alpha_equal(N, O) assert not relay.analysis.alpha_equal(N, O)
if __name__ == "__main__": if __name__ == "__main__":
import numpy as np
np.random.seed(42) np.random.seed(42)
test_alter_layout_conv2d() test_alter_layout_conv2d()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment