Skip CUDA benchmarks on nogpu configs (#42704)

malfet · facebook-github-bot · commit eaace3e10e1e · 2020-08-06T15:47:48.000-07:00
Summary: Avoids timeouts when the benchmark is launched on nogpu configs Pull Request resolved: #42704 Reviewed By: mruberry Differential Revision: D22987725 Pulled By: malfet fbshipit-source-id: aa9aece16557c0af8e05e612277ae1d9e0173a51
diff --git a/.jenkins/pytorch/test.sh b/.jenkins/pytorch/test.sh
@@ -341,7 +341,7 @@ test_bazel() {
 }
 
 test_benchmarks() {
-  if [[ "$BUILD_ENVIRONMENT" == *cuda* ]]; then
+  if [[ "$BUILD_ENVIRONMENT" == *cuda* && "$BUILD_ENVIRONMENT" != *nogpu* ]]; then
     pip_install --user "pytest-benchmark==3.2.3"
     pip_install --user "requests"
     BENCHMARK_DATA="benchmarks/.data"

Original file line number	Diff line number	Diff line change
`@@ -341,7 +341,7 @@ test_bazel() {`
`341`	`341`	`}`
`342`	`342`
`343`	`343`	`test_benchmarks() {`
`344`		`- if [[ "$BUILD_ENVIRONMENT" == cuda ]]; then`
	`344`	`+ if [[ "$BUILD_ENVIRONMENT" == cuda && "$BUILD_ENVIRONMENT" != nogpu ]]; then`
`345`	`345`	`pip_install --user "pytest-benchmark==3.2.3"`
`346`	`346`	`pip_install --user "requests"`
`347`	`347`	`BENCHMARK_DATA="benchmarks/.data"`