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
......@@ -36,15 +36,16 @@ AUTOTVM_TOPHUB_ROOT_PATH = os.path.join(os.path.expanduser('~'), ".tvm", "tophub
# 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')
......
......@@ -69,6 +69,7 @@ def test_alter_layout_conv2d():
targets=['cuda',
'opencl -device=mali',
'opencl -device=intel_graphics',
'llvm -device=arm_cpu',
'llvm -device=core-avx-ii']
......@@ -83,5 +84,6 @@ def test_alter_layout_conv2d():
assert not relay.analysis.alpha_equal(N, O)
if __name__ == "__main__":
import numpy as np
np.random.seed(42)
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