diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml index 657494167a0f..19d8af559440 100644 --- a/.github/workflows/cicd-main.yml +++ b/.github/workflows/cicd-main.yml @@ -142,7 +142,7 @@ jobs: TIMEOUT: 20 # TODO: remove this hack SCRIPT: | - python -c "from nemo.collections.asr.models import ASRModel" && NEMO_NUMBA_MINVER=0.53 pytest tests/collections/asr -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + python -c "from nemo.collections.asr.models import ASRModel" && NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/asr -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Audio: needs: [pre-flight, cicd-test-container-build] @@ -153,7 +153,7 @@ jobs: TIMEOUT: 20 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/collections/audio -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/audio -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Common: needs: [pre-flight, cicd-test-container-build] @@ -163,7 +163,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/collections/common -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/common -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_LLM: needs: [pre-flight, cicd-test-container-build] @@ -173,7 +173,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/collections/llm -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/llm -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Multimodal: needs: [pre-flight, cicd-test-container-build] @@ -183,7 +183,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/collections/multimodal -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/multimodal -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_NLP: needs: [pre-flight, cicd-test-container-build] @@ -193,7 +193,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/collections/nlp -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/nlp -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_TTS: needs: [pre-flight, cicd-test-container-build] @@ -203,7 +203,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/collections/tts -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/collections/tts -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Core: needs: [pre-flight, cicd-test-container-build] @@ -214,7 +214,7 @@ jobs: TIMEOUT: 20 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/core -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/core -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Hydra: needs: [pre-flight, cicd-test-container-build] @@ -224,7 +224,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/hydra -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest tests/hydra -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Lightning: needs: [pre-flight, cicd-test-container-build] @@ -234,7 +234,7 @@ jobs: RUNNER: self-hosted-azure SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest tests/lightning -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0,1 pytest tests/lightning -m "not pleasefixme" --with_downloads --cov-branch --cov-report=xml --cov=nemo L0_Unit_Tests_GPU_Others: needs: [pre-flight, cicd-test-container-build] @@ -244,7 +244,7 @@ jobs: RUNNER: self-hosted-azure-gpus-1 SAVE_COVERAGE_REPORT: true SCRIPT: | - NEMO_NUMBA_MINVER=0.53 pytest -m "not pleasefixme" --with_downloads \ + NEMO_NUMBA_MINVER=0.53 CUDA_VISIBLE_DEVICES=0 pytest -m "not pleasefixme" --with_downloads \ --ignore=tests/collections/asr \ --ignore=tests/collections/audio \ --ignore=tests/collections/common \