From 4fc5a534e2e88cddb02032b7813a974cebf5fefc Mon Sep 17 00:00:00 2001
From: Vladimir Vereschaka <vvereschaka@accesssoftek.com>
Date: Sat, 22 Feb 2025 11:31:55 -0800
Subject: [PATCH] Allow Lit to use 'ptxas' tool to validate generated PTX on
 the NVPTX builders.

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.
---
 buildbot/osuosl/master/config/builders.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/buildbot/osuosl/master/config/builders.py b/buildbot/osuosl/master/config/builders.py
index 180fe312..ab3afcba 100644
--- a/buildbot/osuosl/master/config/builders.py
+++ b/buildbot/osuosl/master/config/builders.py
@@ -3100,6 +3100,8 @@
                         # TMP/TEMP within the build dir (to utilize a ramdisk).
                         'TMP'        : util.Interpolate("%(prop:builddir)s/build"),
                         'TEMP'       : util.Interpolate("%(prop:builddir)s/build"),
+                        # Allow Lit to use 'ptxas' tool to validate generated PTX.
+                        'LLVM_PTXAS_EXECUTABLE' : "/usr/local/cuda/bin/ptxas",
                     })},
 
     {'name' : "llvm-nvptx64-nvidia-ubuntu",
@@ -3125,6 +3127,8 @@
                         # TMP/TEMP within the build dir (to utilize a ramdisk).
                         'TMP'        : util.Interpolate("%(prop:builddir)s/build"),
                         'TEMP'       : util.Interpolate("%(prop:builddir)s/build"),
+                        # Allow Lit to use 'ptxas' tool to validate generated PTX.
+                        'LLVM_PTXAS_EXECUTABLE' : "/usr/local/cuda/bin/ptxas",
                     })},
 
     {'name' : "llvm-nvptx-nvidia-win",
@@ -3148,6 +3152,8 @@
                         # TMP/TEMP within the build dir (to utilize a ramdisk).
                         'TMP'        : util.Interpolate("%(prop:builddir)s/build"),
                         'TEMP'       : util.Interpolate("%(prop:builddir)s/build"),
+                        # Allow Lit to use 'ptxas' tool to validate generated PTX.
+                        'LLVM_PTXAS_EXECUTABLE' : "c:/buildbot/latest-cuda/bin/ptxas.exe",
                     })},
 
     {'name' : "llvm-nvptx64-nvidia-win",
@@ -3171,6 +3177,8 @@
                         # TMP/TEMP within the build dir (to utilize a ramdisk).
                         'TMP'        : util.Interpolate("%(prop:builddir)s/build"),
                         'TEMP'       : util.Interpolate("%(prop:builddir)s/build"),
+                        # Allow Lit to use 'ptxas' tool to validate generated PTX.
+                        'LLVM_PTXAS_EXECUTABLE' : "c:/buildbot/latest-cuda/bin/ptxas.exe",
                     })},
 
     # flang FortranRuntime CUDA Offload builders.