From ecceae966d67bd900bb9b82b18d1f88894feecd8 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Fri, 17 Nov 2023 21:22:37 +0800 Subject: [PATCH] add pytorch test --- .github/workflows/test_pytorch.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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