|
3103 | 3103 | # TMP/TEMP within the build dir (to utilize a ramdisk). |
3104 | 3104 | 'TMP' : util.Interpolate("%(prop:builddir)s/build"), |
3105 | 3105 | 'TEMP' : util.Interpolate("%(prop:builddir)s/build"), |
| 3106 | + # Allow Lit to use 'ptxas' tool to validate generated PTX. |
| 3107 | + 'LLVM_PTXAS_EXECUTABLE' : "/usr/local/cuda/bin/ptxas", |
3106 | 3108 | })}, |
3107 | 3109 |
|
3108 | 3110 | {'name' : "llvm-nvptx64-nvidia-ubuntu", |
|
3128 | 3130 | # TMP/TEMP within the build dir (to utilize a ramdisk). |
3129 | 3131 | 'TMP' : util.Interpolate("%(prop:builddir)s/build"), |
3130 | 3132 | 'TEMP' : util.Interpolate("%(prop:builddir)s/build"), |
| 3133 | + # Allow Lit to use 'ptxas' tool to validate generated PTX. |
| 3134 | + 'LLVM_PTXAS_EXECUTABLE' : "/usr/local/cuda/bin/ptxas", |
3131 | 3135 | })}, |
3132 | 3136 |
|
3133 | 3137 | {'name' : "llvm-nvptx-nvidia-win", |
|
3151 | 3155 | # TMP/TEMP within the build dir (to utilize a ramdisk). |
3152 | 3156 | 'TMP' : util.Interpolate("%(prop:builddir)s/build"), |
3153 | 3157 | 'TEMP' : util.Interpolate("%(prop:builddir)s/build"), |
| 3158 | + # Allow Lit to use 'ptxas' tool to validate generated PTX. |
| 3159 | + 'LLVM_PTXAS_EXECUTABLE' : "c:/buildbot/latest-cuda/bin/ptxas.exe", |
3154 | 3160 | })}, |
3155 | 3161 |
|
3156 | 3162 | {'name' : "llvm-nvptx64-nvidia-win", |
|
3174 | 3180 | # TMP/TEMP within the build dir (to utilize a ramdisk). |
3175 | 3181 | 'TMP' : util.Interpolate("%(prop:builddir)s/build"), |
3176 | 3182 | 'TEMP' : util.Interpolate("%(prop:builddir)s/build"), |
| 3183 | + # Allow Lit to use 'ptxas' tool to validate generated PTX. |
| 3184 | + 'LLVM_PTXAS_EXECUTABLE' : "c:/buildbot/latest-cuda/bin/ptxas.exe", |
3177 | 3185 | })}, |
3178 | 3186 |
|
3179 | 3187 | # flang FortranRuntime CUDA Offload builders. |
|
0 commit comments