fixes

huggingface · Dec 10, 2024 · 4629a4e · 4629a4e
1 parent 1e1d900
commit 4629a4e
Show file tree

Hide file tree

Showing 9 changed files with 11 additions and 11 deletions.
diff --git a/.github/workflows/test_api_cuda.yaml b/.github/workflows/test_api_cuda.yaml
@@ -53,7 +53,7 @@ jobs:
 
       - name: Run examples
         run: |
-          pytest tests/test_examples.py -x -s -k "api and cuda"
+          pytest tests/test_examples.py -x -s -k "api and cuda and pytorch"
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
           PUSH_REPO_ID: optimum-benchmark/cuda
diff --git a/.github/workflows/test_cli_cpu_llama_cpp.yaml b/.github/workflows/test_cli_cpu_llama_cpp.yaml
@@ -48,7 +48,7 @@ jobs:
           pip install -e .[testing,llama-cpp]
 
       - name: Run tests
-        run: pytest tests/test_cli.py -s -k "llama_cpp"
+        run: pytest tests/test_cli.py -s -k "cli and cpu and llama_cpp"
 
       - name: Run examples
-        run: pytest tests/test_examples.py -s -k "llama_cpp"
+        run: pytest tests/test_examples.py -s -k "cli and cpu and llama_cpp"
diff --git a/.github/workflows/test_cli_cpu_py_txi.yaml b/.github/workflows/test_cli_cpu_py_txi.yaml
@@ -51,4 +51,4 @@ jobs:
         run: pytest tests/test_cli.py -s -k "cli and cpu and py_txi"
 
       # - name: Run examples
-      #   run: pytest tests/test_examples.py -s -k "cli and cpu and py_txi"
+      #   run: pytest tests/test_examples.py -s -k "cli and cpu and tgi"
diff --git a/.github/workflows/test_cli_cuda_onnxruntime.yaml b/.github/workflows/test_cli_cuda_onnxruntime.yaml
@@ -49,5 +49,5 @@ jobs:
         run: |
           pytest tests/test_cli.py -x -s -k "cli and cuda and onnxruntime"
 
-      - name: Run examples
-        run: pytest tests/test_examples.py -x -s -k "cli and cuda and onnxruntime"
+      # - name: Run examples
+      #   run: pytest tests/test_examples.py -x -s -k "cli and cuda and onnxruntime"
diff --git a/.github/workflows/test_cli_cuda_py_txi.yaml b/.github/workflows/test_cli_cuda_py_txi.yaml
@@ -51,4 +51,4 @@ jobs:
         run: pytest tests/test_cli.py -x -s -k "cli and cuda and py_txi"
 
       - name: Run examples
-        run: pytest tests/test_examples.py -x -s -k "cli and cuda and (tgi or tei)"
+        run: pytest tests/test_examples.py -x -s -k "cli and cuda and tgi"
diff --git a/.github/workflows/test_cli_cuda_pytorch.yaml b/.github/workflows/test_cli_cuda_pytorch.yaml
@@ -51,7 +51,7 @@ jobs:
           pytest tests/test_cli.py -x -s -k "cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed)"
 
       - name: Run examples
-        run: pytest tests/test_examples.py -x -s -k "cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed)"
+        run: pytest tests/test_examples.py -x -s -k "cli and cuda and pytorch"
 
   run_cli_cuda_pytorch_multi_gpu_tests:
     if: ${{

diff --git a/.github/workflows/test_cli_cuda_tensorrt_llm.yaml b/.github/workflows/test_cli_cuda_tensorrt_llm.yaml
@@ -51,7 +51,7 @@ jobs:
           pytest tests/test_cli.py -x -s -k "cli and cuda and tensorrt_llm and not (tp or pp)"
 
       - name: Run examples
-        run: pytest tests/test_examples.py -x -s -k "cli and cuda and tensorrt_llm and not (tp or pp)"
+        run: pytest tests/test_examples.py -x -s -k "cli and cuda and trt"
 
   cli_cuda_tensorrt_llm_multi_gpu_tests:
     if: ${{

diff --git a/.github/workflows/test_cli_cuda_vllm.yaml b/.github/workflows/test_cli_cuda_vllm.yaml
@@ -50,9 +50,9 @@ jobs:
         run: |
           FORCE_SEQUENTIAL=1 pytest tests/test_cli.py -x -s -k "cli and cuda and vllm and not (tp or pp)"
 
-      - name: Run examples (sequential)
+      - name: Run examples
         run: |
-          FORCE_SEQUENTIAL=1 pytest tests/test_examples.py -x -s -k "cli and cuda and vllm and not (tp or pp)"
+          pytest tests/test_examples.py -x -s -k "cli and cuda and vllm"
 
   run_cli_cuda_vllm_multi_gpu_tests:
     if: ${{

diff --git a/examples/cuda_pytorch_llama.py → examples/cuda_pytorch_llama_quants.py b/examples/cuda_pytorch_llama.py → examples/cuda_pytorch_llama_quants.py