Ok dokey

Signed-off-by: Huy Do <huydhn@gmail.com>

Author: huydhn

.github/workflows/test_linux_job.yml

@@ -38,15 +38,15 @@ jobs:
   test-cpu:
     uses: ./.github/workflows/linux_job.yml
     with:
-      job-name: "linux-py3.10-cpu"
+      job-name: "linux-py3.9-cpu"
       runner: linux.2xlarge
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
       submodules: "recursive"
       gpu-arch-type: cpu
       gpu-arch-version: ""
       script: |
-        conda create --yes --quiet -n test python=3.10
+        conda create --yes --quiet -n test python=3.9
         conda activate test
         python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cpu --pre torch
         # Can import pytorch
@@ -55,50 +55,50 @@ jobs:
     uses: ./.github/workflows/linux_job.yml
     strategy:
       matrix:
-        runner_type: ["linux.g5.4xlarge.nvidia.gpu"]
+        runner_type: ["linux.4xlarge.nvidia.gpu", "linux.g5.4xlarge.nvidia.gpu"]
     with:
-      job-name: "linux-py3.10-cu128"
+      job-name: "linux-py3.9-cu121"
       runner: ${{ matrix.runner_type }}
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
       submodules: ${{ 'true' }}
       gpu-arch-type: cuda
-      gpu-arch-version: "12.8"
+      gpu-arch-version: "12.1"
       timeout: 60
       script: |
         nvidia-smi
-        nvcc --version | grep "cuda_12.8"
-        conda create --yes --quiet -n test python=3.10
+        nvcc --version | grep "cuda_12.1"
+        conda create --yes --quiet -n test python=3.9
         conda activate test
-        python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu128 --pre torch
+        python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu121 --pre torch
         # Can import pytorch, cuda is available
         python3 -c 'import torch;cuda_avail = torch.cuda.is_available();print("CUDA available: " + str(cuda_avail));assert(cuda_avail)'
         python3 -c 'import torch;t = torch.ones([2,2], device="cuda:0");print(t);print("tensor device:" + str(t.device))'
-        python3 -c 'import torch;assert(torch.version.cuda == "12.8")'
+        python3 -c 'import torch;assert(torch.version.cuda == "12.1")'
   test-gpu-containers:
     uses: ./.github/workflows/linux_job.yml
     strategy:
       matrix:
         runner_type: ["linux.aws.a100"]
     with:
-      job-name: "linux-py3.10-cu126-container"
+      job-name: "linux-py3.9-cu121-container"
       runner: ${{ matrix.runner_type }}
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
       submodules: ${{ 'true' }}
       gpu-arch-type: cuda
-      gpu-arch-version: "12.6"
+      gpu-arch-version: "12.1"
       timeout: 60
       script: |
         nvidia-smi
-        nvcc --version | grep "cuda_12.6"
-        conda create --yes --quiet -n test python=3.10
+        nvcc --version | grep "cuda_12.1"
+        conda create --yes --quiet -n test python=3.9
         conda activate test
-        python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu126 --pre torch
+        python3 -m pip install --index-url https://download.pytorch.org/whl/nightly/cu121 --pre torch
         # Can import pytorch, cuda is available
         python3 -c 'import torch;cuda_avail = torch.cuda.is_available();print("CUDA available: " + str(cuda_avail));assert(cuda_avail)'
         python3 -c 'import torch;t = torch.ones([2,2], device="cuda:0");print(t);print("tensor device:" + str(t.device))'
-        python3 -c 'import torch;assert(torch.version.cuda == "12.6")'
+        python3 -c 'import torch;assert(torch.version.cuda == "12.1")'
   test-docker-image:
     uses: ./.github/workflows/linux_job.yml
     with:
@@ -180,7 +180,7 @@ jobs:
     uses: ./.github/workflows/linux_job.yml
     strategy:
       matrix:
-        python_version: ["3.10", "3.11", "3.12", "3.13"]
+        python_version: ["3.9", "3.10", "3.11", "3.12"]
     with:
       runner: linux.2xlarge
       test-infra-repository: ${{ github.repository }}
@@ -203,6 +203,6 @@ jobs:
       gpu-arch-version: ""
       run-with-docker: false
       script: |
-        export PYTHON_VERSION="3.10"
+        export PYTHON_VERSION="3.9"
         docker run -e PYTHON_VERSION -t -v $PWD:$PWD -w $PWD "${DOCKER_IMAGE}" echo "hello from inside docker"
         echo "hello from outside docker"