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.

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

[ONNX] Enable GPU in ONNX importer tests #7438

Merged
merged 17 commits into from
Mar 30, 2021
Merged

[ONNX] Enable GPU in ONNX importer tests #7438

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
1595841
remove hardcoded target and ctx
electriclilies Feb 10, 2021
43668a9
fix c-codgen for floating point mod
Feb 12, 2021
8655ce5
Merge pull request #3 from mbrookhart/onnx_gpu_tests
Feb 12, 2021
c02d9fb
MDisable onnx gpu test for argmin / argmax so we can get this fix mer…
electriclilies Mar 17, 2021
371faff
lint
electriclilies Mar 17, 2021
fcbc54e
fix black
electriclilies Mar 17, 2021
dc4b083
Add flag to task_python_frontend.sh to only run GPU enabled tests on GPU
electriclilies Mar 17, 2021
2f60c43
black again
electriclilies Mar 17, 2021
b58dc24
Enable GPU for test_nonzero
electriclilies Mar 18, 2021
e765689
Respond to comments
electriclilies Mar 18, 2021
e583e9d
Don't test batch matmul on CUDA
electriclilies Mar 19, 2021
e6e5cf0
Turn cuda off for dynamic batch matmul test
electriclilies Mar 22, 2021
4f5a41c
Fix task script
electriclilies Mar 23, 2021
9f31e99
Merge
electriclilies Mar 23, 2021
dabc571
Flaky test
electriclilies Mar 24, 2021
96bf1ba
another flaky test
electriclilies Mar 29, 2021
9268af2
Merge remote-tracking branch 'upstream/main' into onnx_gpu_enabled
electriclilies Mar 29, 2021
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
Prev Previous commit
Next Next commit
Merge
  • Loading branch information
@electriclilies
electriclilies committed Mar 23, 2021
commit 9f31e99a2c926d93a2c03ba594d4af11be414467
7 changes: 3 additions & 4 deletions tests/scripts/task_python_frontend.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,14 +31,13 @@ find . -type f -path "*.pyc" | xargs rm -f
# Rebuild cython
make cython3

# Enable tvm.testing decorators in the ONNX importer test (not enabling in the other tests because we
# they do not consistently use the decorators to indicate that tests should run on GPU)
# In the future, we should enable tvm.testing decorators for all the test files.

echo "Running relay MXNet frontend test..."
run_pytest cython python-frontend-mxnet tests/python/frontend/mxnet

echo "Running relay ONNX frontend test..."
# Enable tvm.testing decorators in the ONNX importer test (not enabling in the other tests because we
# they do not consistently use the decorators to indicate that tests should run on GPU)
# In the future, we should enable tvm.testing decorators for all the test files.
PYTEST_ADDOPTS="-m gpu $PYTEST_ADDOPTS" run_pytest cython python-frontend-onnx tests/python/frontend/onnx

echo "Running relay CoreML frontend test..."
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.