File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -102,19 +102,19 @@ jobs:
102102 echo "H100 detected--setting Hopper90 architecture"
103103 echo "Hopper=ON" >> $GITHUB_ENV
104104 echo "CUDA_ARCH=90" >> $GITHUB_ENV
105- echo "CUDA_DEVICE=Hopper90 " >> $GITHUB_ENV
105+ echo "CUDA_DEVICE=HOPPER90 " >> $GITHUB_ENV
106106 ;;
107107 *"A100"*)
108108 echo "A100 detected--setting Ampere80 architecture"
109109 echo "Ampere=ON" >> $GITHUB_ENV
110110 echo "CUDA_ARCH=80" >> $GITHUB_ENV
111- echo "CUDA_DEVICE=Ampere80 " >> $GITHUB_ENV
111+ echo "CUDA_DEVICE=AMPERE80 " >> $GITHUB_ENV
112112 ;;
113113 *"V100"*)
114114 echo "V100 detected--setting Volta70 architecture"
115115 echo "Volta=ON" >> $GITHUB_ENV
116116 echo "CUDA_ARCH=70" >> $GITHUB_ENV
117- echo "CUDA_DEVICE=Volta70 " >> $GITHUB_ENV
117+ echo "CUDA_DEVICE=VOLTA70 " >> $GITHUB_ENV
118118 ;;
119119 *)
120120 echo "Unsupported GPU model: $gpu_model"
You can’t perform that action at this time.
0 commit comments