Skip to content

Commit c16b052

Browse files
authored
Allow Lit to use 'ptxas' tool to validate generated PTX on the NVPTX builders. (#390)
Specify LLVM_PTXAS_EXECUTABLE environment variable with a full path to 'ptxas' tool from installed NVIDIA CUDA Toolkit. LIT will be automatically check and use this tool to validate PTX during the tests.
1 parent 0ac8f9d commit c16b052

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

buildbot/osuosl/master/config/builders.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3103,6 +3103,8 @@
31033103
# TMP/TEMP within the build dir (to utilize a ramdisk).
31043104
'TMP' : util.Interpolate("%(prop:builddir)s/build"),
31053105
'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",
31063108
})},
31073109

31083110
{'name' : "llvm-nvptx64-nvidia-ubuntu",
@@ -3128,6 +3130,8 @@
31283130
# TMP/TEMP within the build dir (to utilize a ramdisk).
31293131
'TMP' : util.Interpolate("%(prop:builddir)s/build"),
31303132
'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",
31313135
})},
31323136

31333137
{'name' : "llvm-nvptx-nvidia-win",
@@ -3151,6 +3155,8 @@
31513155
# TMP/TEMP within the build dir (to utilize a ramdisk).
31523156
'TMP' : util.Interpolate("%(prop:builddir)s/build"),
31533157
'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",
31543160
})},
31553161

31563162
{'name' : "llvm-nvptx64-nvidia-win",
@@ -3174,6 +3180,8 @@
31743180
# TMP/TEMP within the build dir (to utilize a ramdisk).
31753181
'TMP' : util.Interpolate("%(prop:builddir)s/build"),
31763182
'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",
31773185
})},
31783186

31793187
# flang FortranRuntime CUDA Offload builders.

0 commit comments

Comments
 (0)