Skip to content

Commit 25c235c

Browse files
committed
fixed target
1 parent ebce1e7 commit 25c235c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/python/unittest/test_meta_schedule_relay_integration.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,8 @@ def test_task_extraction_winograd_tensorcore():
120120
with tvm.transform.PassContext(opt_level=3):
121121
mod = seq(mod)
122122

123-
extracted_tasks = ms.relay_integration.extract_tasks(mod, target="cuda", params=params)
123+
target = tvm.target.Target("nvidia/geforce-rtx-3070")
124+
extracted_tasks = ms.relay_integration.extract_tasks(mod, target=target, params=params)
124125

125126
assert len([t for t in extracted_tasks if "winograd" in t.task_name]) == 4
126127

0 commit comments

Comments
 (0)