t

openvinotoolkit · Apr 18, 2024 · ff756cf · ff756cf
1 parent 2d21ef3
commit ff756cf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/precommit.yml b/.github/workflows/precommit.yml
@@ -92,6 +92,7 @@ jobs:
 
 
   pytorch-cpu:
+    timeout-minutes: 150
     defaults:
       run:
         shell: bash
@@ -109,6 +110,7 @@ jobs:
         run: make test-torch-cpu
 
   pytorch-cuda:
+    timeout-minutes: 150
     defaults:
       run:
         shell: bash
@@ -121,8 +123,6 @@ jobs:
       - name: Download CUDA
         run: |
           wget -q https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
-      - name: Install CUDA
-        run: |
           sudo sh cuda_12.1.1_530.30.02_linux.run --toolkit
       - name: Runner info
         continue-on-error: true

diff --git a/Makefile b/Makefile
@@ -146,7 +146,7 @@ test-torch:
 	pytest ${COVERAGE_ARGS} tests/torch -m "not weekly and not nightly and not models_hub" --junitxml ${JUNITXML_PATH} $(DATA_ARG)
 
 test-torch-cpu:
-	pytest ${COVERAGE_ARGS} tests/torch -n4 -rS -m "not cuda and not weekly and not nightly and not models_hub"
+	pytest ${COVERAGE_ARGS} tests/torch -n2 -rS -m "not cuda and not weekly and not nightly and not models_hub"
 
 test-torch-cuda:
 	pytest ${COVERAGE_ARGS} tests/torch -rS -m "cuda and not weekly and not nightly and not models_hub"