Skip to content

Commit

Permalink
add pytorch test
Browse files Browse the repository at this point in the history
  • Loading branch information
youkaichao committed Nov 17, 2023
1 parent 4d4c685 commit ecceae9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/test_pytorch.yml
Original file line number Diff line number Diff line change
Expand Up @@ -41,5 +41,5 @@ jobs:
echo "success"
- name: Test with pytest
run: |
pytest tests/test.py
for TORCH_COMPILE_BACKEND in eager aot_eager inductor; do for REQUIRES_GRAD in 0 1; do for DYNAMIC_SHAPE in 0 1; do for COMPILE_TYPE in function module; do for USAGE_TYPE in debug dump; do export TORCH_COMPILE_BACKEND REQUIRES_GRAD DYNAMIC_SHAPE COMPILE_TYPE USAGE_TYPE; echo input | python tests/test_pytorch/test_pytorch.py; done; done; done; done; done;
python tests/assert.py

0 comments on commit ecceae9

Please sign in to comment.