Skip to content

[CI] Refactor of tvm.testing.requires_* annotations #11313

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

Merged
merged 9 commits into from
Jun 3, 2022
Merged

[CI] Refactor of tvm.testing.requires_* annotations #11313

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
150ce46
[CI] Improved skip messages when using @tvm.testing.requires_*
Lunderberg May 13, 2022
9659562
Refactor to specify repeated feature marks, compile-only markers
Lunderberg May 13, 2022
2cd0654
Fixed lint errors
Lunderberg May 16, 2022
fff0404
Import from contrib, not from a different import
Lunderberg May 17, 2022
ced22ee
Removed use of requires_llvm() as a list of marks
Lunderberg May 17, 2022
d8090fa
Corrected mark from requires_gpu to requires_cuda
Lunderberg May 17, 2022
7bd8211
Adding missing "not"
Lunderberg May 18, 2022
3dd37a4
Added USE_CMSISNN as a requirement for corstone300.
Lunderberg May 18, 2022
8a4505b
Merge branch 'main' into requires_feature_refactor
Lunderberg May 27, 2022
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
25 changes: 13 additions & 12 deletions python/tvm/testing/plugin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,8 +56,8 @@
def pytest_configure(config):
"""Runs at pytest configure time, defines marks to be used later."""

for markername, desc in MARKERS.items():
config.addinivalue_line("markers", "{}: {}".format(markername, desc))
for feature in utils.Feature._all_features.values():
feature._register_marker(config)

print("enabled targets:", "; ".join(map(lambda x: x[0], utils.enabled_targets())))
print("pytest marker:", config.option.markexpr)
Expand Down Expand Up @@ -269,25 +269,26 @@ def _target_to_requirement(target):

# mapping from target to decorator
if target.kind.name == "cuda" and "cudnn" in target.attrs.get("libs", []):
return utils.requires_cudnn()
return utils.requires_cudnn.marks()
if target.kind.name == "cuda" and "cublas" in target.attrs.get("libs", []):
return utils.requires_cublas()
return utils.requires_cublas.marks()
if target.kind.name == "cuda":
return utils.requires_cuda()
return utils.requires_cuda.marks()
if target.kind.name == "rocm":
return utils.requires_rocm()
return utils.requires_rocm.marks()
if target.kind.name == "vulkan":
return utils.requires_vulkan()
return utils.requires_vulkan.marks()
if target.kind.name == "nvptx":
return utils.requires_nvptx()
return utils.requires_nvptx.marks()
if target.kind.name == "metal":
return utils.requires_metal()
return utils.requires_metal.marks()
if target.kind.name == "opencl":
return utils.requires_opencl()
return utils.requires_opencl.marks()
if target.kind.name == "llvm":
return utils.requires_llvm()
return utils.requires_llvm.marks()
if target.kind.name == "hexagon":
return utils.requires_hexagon()
return utils.requires_hexagon.marks()

return []


Expand Down
Loading