Fixup missing python command

pytorch · Feb 28, 2024 · 8042a2a · 8042a2a
1 parent 79deda5
commit 8042a2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.circleci/common.sh b/.circleci/common.sh
@@ -137,7 +137,7 @@ function run_torch_xla_python_tests() {
 
         # single-host-multi-process
         num_devices=$(nvidia-smi --list-gpus | wc -l)
-        test/test_train_mp_imagenet.py --fake_data --batch_size=16 --num_epochs=1 --num_steps=25 --model=resnet18
+        python3 test/test_train_mp_imagenet.py --fake_data --batch_size=16 --num_epochs=1 --num_steps=25 --model=resnet18
         torchrun --nnodes=1 --node_rank=0 --nproc_per_node=$num_devices test/test_train_mp_imagenet.py --fake_data --pjrt_distributed --batch_size=16 --num_epochs=1  --num_steps=25 --model=resnet18
 
         # single-host-SPMD