Skip to content

Commit f5e03e0

Browse files
authored
[MetaSchedule] TE Workloads (#502)
1 parent fc902d3 commit f5e03e0

File tree

2 files changed

+745
-0
lines changed

2 files changed

+745
-0
lines changed

python/tvm/meta_schedule/testing/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,4 @@
1717
"""Testing utilities in meta schedule"""
1818
from .local_rpc import LocalRPC
1919
from .relay_workload import get_torch_model, MODEL_TYPE, MODEL_TYPES
20+
from .te_workload import create_te_workload

0 commit comments

Comments
 (0)