fix cputest job

HabanaAI · Aug 13, 2024 · ebcb4ab · ebcb4ab
1 parent 08c9cf3
commit ebcb4ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/cpu-test.yml b/.github/workflows/cpu-test.yml
@@ -12,7 +12,7 @@ on:
 
 
 jobs:
-  ruff:
+  cputest:
     runs-on: ubuntu-latest
     strategy:
       matrix:
@@ -31,4 +31,4 @@ jobs:
         VLLM_TARGET_DEVICE=hpu python setup.py develop
     - name: cpu-test
       run: |
-        mypy tests --config-file pyproject.toml
+        VLLM_SKIP_WARMUP=true VLLM_PROMPT_SEQ_BUCKET_MAX=128 python examples/offline_inference_fakehpu.py