@@ -268,11 +268,11 @@ test-cli-cpu-ipex-examples:
268268# ## CUDA tests
269269test-cli-cuda-pytorch-single :
270270 uv sync --dev --extra bitsandbytes --extra deepspeed
271- uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed)"
271+ uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and not (tp or dp or ddp or device_map or deepspeed)"
272272
273273test-cli-cuda-pytorch-multi :
274274 uv sync --dev --extra bitsandbytes --extra deepspeed
275- uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and (dp or ddp or device_map or deepspeed)"
275+ uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and (tp or dp or ddp or device_map or deepspeed)"
276276
277277test-cli-cuda-vllm-single :
278278 uv sync --dev --extra vllm
@@ -301,11 +301,11 @@ test-cli-cuda-py-txi:
301301# ## CUDA examples
302302test-cli-cuda-pytorch-single-examples :
303303 uv sync --dev --extra bitsandbytes --extra deepspeed
304- uv run pytest tests/test_examples.py -s -k " cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed)"
304+ uv run pytest tests/test_examples.py -s -k " cli and cuda and pytorch and not (tp or dp or ddp or device_map or deepspeed)"
305305
306306test-cli-cuda-pytorch-multi-examples :
307307 uv sync --dev --extra bitsandbytes --extra deepspeed
308- uv run pytest tests/test_examples.py -s -k " cli and cuda and pytorch and (dp or ddp or device_map or deepspeed)"
308+ uv run pytest tests/test_examples.py -s -k " cli and cuda and pytorch and (tp or dp or ddp or device_map or deepspeed)"
309309
310310test-cli-cuda-onnxruntime-examples :
311311 uv sync --dev --extra onnxruntime-gpu
@@ -334,16 +334,20 @@ test-cli-cuda-py-txi-examples:
334334# ## ROCm tests
335335test-cli-rocm-pytorch-single :
336336 uv sync --dev
337- uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and not (dp or ddp or device_map or deepspeed)"
337+ uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and not (tp or dp or ddp or device_map or deepspeed)"
338338
339339test-cli-rocm-pytorch-multi :
340340 uv sync --dev
341- uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and (dp or ddp or device_map or deepspeed)"
341+ uv run pytest tests/test_cli.py -s -k " cli and cuda and pytorch and (tp or dp or ddp or device_map or deepspeed)"
342342
343343# ## ROCm examples
344- test-cli-rocm-pytorch-examples :
344+ test-cli-rocm-pytorch-single- examples :
345345 uv sync --dev
346- uv run pytest tests/test_examples.py -s -k " cli and rocm and pytorch"
346+ uv run pytest tests/test_examples.py -s -k " cli and rocm and pytorch and not (tp or dp or ddp or device_map or deepspeed)"
347+
348+ test-cli-rocm-pytorch-multi-examples :
349+ uv sync --dev
350+ uv run pytest tests/test_examples.py -s -k " cli and rocm and pytorch and (tp or dp or ddp or device_map or deepspeed)"
347351
348352# ## MPS tests
349353test-cli-mps-pytorch :
0 commit comments