[ROCm] Only run core tests for ROCm post-build checks

jax-ml · Feb 10, 2025 · d50206b · d50206b
1 parent cf308a8
commit d50206b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/rocm-ci.yml b/.github/workflows/rocm-ci.yml
@@ -55,5 +55,5 @@ jobs:
       - name: Run tests
         run: |
           cd $WORKSPACE_DIR
-          python3 build/rocm/ci_build test $TEST_IMAGE
+          python3 build/rocm/ci_build test $TEST_IMAGE --test-cmd "pytest tests/core_test.py"