Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[TOPI] Intel graphics conv2d autotvm template added #3839

Merged
merged 15 commits into from
Sep 7, 2019
Merged

[TOPI] Intel graphics conv2d autotvm template added #3839

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
6c22d45
update lint
Laurawly Aug 26, 2019
62b511c
lint fixed
Laurawly Aug 26, 2019
a194deb
lint updated
Laurawly Aug 26, 2019
c46aa7a
lint fixed
Laurawly Aug 26, 2019
8f6565c
lint fixed
Laurawly Aug 27, 2019
e791302
lint fixed
Laurawly Aug 27, 2019
80512d7
updates
Laurawly Aug 27, 2019
f66d93e
add intel graphics as a package
Laurawly Aug 29, 2019
364b1fb
remove print info
Laurawly Sep 3, 2019
966f1d4
depthwise conv2d schedule added for intel graphics
Laurawly Sep 3, 2019
745af6b
asdf
Laurawly Sep 3, 2019
7cf3365
fix lint
Laurawly Sep 4, 2019
9f4df30
fix lint
Laurawly Sep 4, 2019
2789011
fix ci
Laurawly Sep 4, 2019
282f3f3
add channels
Laurawly Sep 5, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 8 additions & 7 deletions python/tvm/autotvm/tophub.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand Down
2 changes: 2 additions & 0 deletions tests/python/frontend/nnvm_to_relay/test_alter_conv2d.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,7 @@ def convnet():
targets=['cuda',
'opencl -device=mali',
'opencl -device=intel_graphics',

'llvm -device=arm_cpu',
'llvm -device=core-avx-ii']

Expand All @@ -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()
Loading