Fix job name checking for AVX tests (#8135)

sungin-h · Jun 4, 2018 · a5ce012 · a5ce012
1 parent c0a419e
commit a5ce012
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.jenkins/pytorch/test.sh b/.jenkins/pytorch/test.sh
@@ -36,10 +36,10 @@ fi
 
 export ATEN_DISABLE_AVX=
 export ATEN_DISABLE_AVX2=
-if [[ "${JOB_BASE_NAME}" == *NO_AVX* ]]; then
+if [[ "${JOB_BASE_NAME}" == *-NO_AVX-* ]]; then
   export ATEN_DISABLE_AVX=1
 fi
-if [[ "${JOB_BASE_NAME}" == *NO_AVX2* ]]; then
+if [[ "${JOB_BASE_NAME}" == *-NO_AVX2-* ]]; then
   export ATEN_DISABLE_AVX2=1
 fi