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.

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

[Testing] Add decorator tvm.testing.requires_cuda_compute_version #12778

Merged
merged 2 commits into from
Sep 16, 2022

Commits on Sep 14, 2022

  1. [Testing] Add decorator tvm.testing.requires_cuda_compute_version

    Previously, individual unit tests would call
    `tvm.contrib.nvcc.get_target_compute_version` and return early.  This
    was repeated boilerplate in many tests, and incorrectly reported a
    test as `PASSED` if the required infrastructure wasn't present.
    
    This commit introduces `tvm.testing.requires_cuda_compute_version`, a
    decorator that checks the CUDA compute version and applies
    `pytest.mark.skipif`.  If required infrastructure isn't present, a
    test will be reported as `SKIPPED`.
    Lunderberg committed Sep 14, 2022
    Configuration menu
    Copy the full SHA
    d0548c2 View commit details
    Browse the repository at this point in the history

Commits on Sep 15, 2022

  1. Configuration menu
    Copy the full SHA
    1862459 View commit details
    Browse the repository at this point in the history