Skip to content

Commit

Permalink
[Pytest][TensorRT] Mark the TensorRT tests with tvm.testing.requires_…
Browse files Browse the repository at this point in the history
…cuda

Previously, the tests had an early bailout if tensorrt was disabled,
or if there was no cuda device present.  However, the tests were not
marked with `pytest.mark.gpu` and so they didn't run during
`task_python_integration_gpuonly.sh`.  This commit adds the
`requires_cuda` mark, and maintains the same behavior of testing the
tensorrt compilation steps if compilation is enabled, and running the
results if tensorrt is enabled.

In addition, some of the tests result in failures when run.  These
have been marked with `pytest.mark.xfail`, and are being tracked in
issue apache#8901.
  • Loading branch information
Lunderberg committed Sep 1, 2021
1 parent f029798 commit 5c78c43
Showing 1 changed file with 285 additions and 238 deletions.
Loading

0 comments on commit 5c78c43

Please sign in to comment.