update

intel · Jan 15, 2024 · de4e13e · de4e13e
1 parent 7ad7918
commit de4e13e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
@@ -78,7 +78,6 @@ jobs:
         env:
           # TRITON_INTERPRET: "1"
           CUA_VISIBLE_DEVICES: ""
-        if: ${{ env.BACKEND == 'XPU'}}
         run: |
           cd python/test/unit
           python3 -m pytest -vs operators/test_flash_attention.py
@@ -87,7 +86,7 @@ jobs:
         if: ${{ env.BACKEND == 'XPU'}}
         run: |
           cd python/test/unit
-          python3 -m pytest -n auto --verbose operators
+          python3 -m pytest -n 8 --verbose operators
 
       - name: Run XPU python tests
         if: ${{ env.BACKEND == 'XPU'}}

diff --git a/.github/workflows/build_and_test_2.yaml b/.github/workflows/build_and_test_2.yaml
@@ -176,7 +176,7 @@ jobs:
       - name: Run partial operators tests
         run: |
           cd python/test/unit
-          python3 -m pytest -n auto --verbose operators
+          python3 -m pytest -n 8 --verbose operators
 
       - name: Run XPU python tests
         run: |

diff --git a/scripts/test-triton.sh b/scripts/test-triton.sh
@@ -118,7 +118,7 @@ function run_core_tests {
     echo "FAILED: return code $?" ; exit $?
   fi
   cd $CORE_TEST_DIR/operators
-  TRITON_DISABLE_LINE_INFO=1 python3 -m pytest -n auto --verbose
+  TRITON_DISABLE_LINE_INFO=1 python3 -m pytest -n 8 --verbose
   if [ $? -ne 0 ]; then
     echo "FAILED: return code $?" ; exit $?
   fi