@@ -40,31 +40,28 @@ case "$suite" in
40
40
gpu)
41
41
echo " -- Run Python tests, using a single GPU"
42
42
python -c ' from cupy.cuda import jitify; jitify._init_module()'
43
- pytest -v -s -rxXs --fulltrace -- durations=0 -m ' not mgpu' tests/python-gpu
43
+ pytest -v -s -rxXs --durations=0 -m ' not mgpu' tests/python-gpu
44
44
;;
45
45
mgpu)
46
46
echo " -- Run Python tests, using multiple GPUs"
47
47
python -c ' from cupy.cuda import jitify; jitify._init_module()'
48
48
export NCCL_RAS_ENABLE=0
49
- pytest -v -s -rxXs --fulltrace --durations=0 -m ' mgpu' tests/python-gpu
50
- pytest -v -s -rxXs --fulltrace --durations=0 -m ' mgpu' \
51
- tests/test_distributed/test_gpu_with_dask
52
- pytest -v -s -rxXs --fulltrace --durations=0 -m ' mgpu' \
53
- tests/test_distributed/test_gpu_with_spark
54
- pytest -v -s -rxXs --fulltrace --durations=0 -m ' mgpu' \
55
- tests/test_distributed/test_gpu_federated
49
+ pytest -v -s -rxXs --durations=0 -m ' mgpu' tests/python-gpu
50
+ pytest -v -s -rxXs --durations=0 tests/test_distributed/test_gpu_with_dask
51
+ pytest -v -s -rxXs --durations=0 tests/test_distributed/test_gpu_with_spark
52
+ pytest -v -s -rxXs --durations=0 tests/test_distributed/test_gpu_federated
56
53
;;
57
54
cpu)
58
55
echo " -- Run Python tests (CPU)"
59
56
export RAY_OBJECT_STORE_ALLOW_SLOW_STORAGE=1
60
- pytest -v -s -rxXs --fulltrace -- durations=0 tests/python
61
- pytest -v -s -rxXs --fulltrace -- durations=0 tests/test_distributed/test_with_dask
62
- pytest -v -s -rxXs --fulltrace -- durations=0 tests/test_distributed/test_with_spark
63
- pytest -v -s -rxXs --fulltrace -- durations=0 tests/test_distributed/test_federated
57
+ pytest -v -s -rxXs --durations=0 tests/python
58
+ pytest -v -s -rxXs --durations=0 tests/test_distributed/test_with_dask
59
+ pytest -v -s -rxXs --durations=0 tests/test_distributed/test_with_spark
60
+ pytest -v -s -rxXs --durations=0 tests/test_distributed/test_federated
64
61
;;
65
62
cpu-arm64)
66
63
echo " -- Run Python tests (CPU, ARM64)"
67
- pytest -v -s -rxXs --fulltrace -- durations=0 \
64
+ pytest -v -s -rxXs --durations=0 \
68
65
tests/python/test_basic.py tests/python/test_basic_models.py \
69
66
tests/python/test_model_compatibility.py
70
67
;;
0 commit comments