diff --git a/.github/workflows/test_pytorch.yml b/.github/workflows/test_pytorch.yml index 13d03912..a3a2b780 100644 --- a/.github/workflows/test_pytorch.yml +++ b/.github/workflows/test_pytorch.yml @@ -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