From 368ebb7e26f86e7807ef4b33f41c4d6c77541eb2 Mon Sep 17 00:00:00 2001 From: agibsonccc Date: Fri, 12 Mar 2021 19:23:32 +0900 Subject: [PATCH] Update run-gpu-tests-sanity-checks.yml --- .github/workflows/run-gpu-tests-sanity-checks.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/run-gpu-tests-sanity-checks.yml b/.github/workflows/run-gpu-tests-sanity-checks.yml index 077ed1943..75b82b712 100644 --- a/.github/workflows/run-gpu-tests-sanity-checks.yml +++ b/.github/workflows/run-gpu-tests-sanity-checks.yml @@ -35,5 +35,5 @@ jobs: protoc --version bash ./change-cuda-versions.sh 11.2 export OMP_NUM_THREADS=1 - mvn -DskipTestResourceEnforcement=true -Ptestresources -pl ":deeplearning4j-modelimport,:deeplearning4j-core,:nd4j-cuda-11.2,:samediff-import,:libnd4j" -Dlibnd4j.helper=cudnn -Dlibnd4j.compute="5.0 5.2 5.3 6.0 8.0" -Ptest-nd4j-cuda --also-make -Dlibnd4j.chip=cuda clean test + mvn -DskipTestResourceEnforcement=true -Ptestresources -pl ":deeplearning4j-modelimport,:deeplearning4j-core,:nd4j-cuda-11.2,:samediff-import,:libnd4j" -Dlibnd4j.helper=cudnn -Ptest-nd4j-cuda --also-make -Dlibnd4j.chip=cuda clean test