[NVPTX] Fix ptxas tests from #135444

[modiking]

ptxas needs to be updated to the correct arch to match what llc is outputting

[llvmbot]

@llvm/pr-subscribers-backend-nvptx

Author: None (modiking)

Changes

ptxas needs to be updated to the correct arch to match what llc is outputting

---

Full diff: https://github.com/llvm/llvm-project/pull/136782.diff

2 Files Affected:

• (modified) llvm/test/CodeGen/NVPTX/addrspacecast-ptx64.ll (+2-2)
• (modified) llvm/test/CodeGen/NVPTX/distributed-shared-cluster.ll (+1-1)

diff --git a/llvm/test/CodeGen/NVPTX/addrspacecast-ptx64.ll b/llvm/test/CodeGen/NVPTX/addrspacecast-ptx64.ll
index afd0a7fded64e..01326db9a8b16 100644
--- a/llvm/test/CodeGen/NVPTX/addrspacecast-ptx64.ll
+++ b/llvm/test/CodeGen/NVPTX/addrspacecast-ptx64.ll
@@ -1,8 +1,8 @@
 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
 ; RUN: llc -O0 < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx78 | FileCheck %s -check-prefixes=NOPTRCONV
 ; RUN: llc -O0 < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx78 --nvptx-short-ptr | FileCheck %s -check-prefixes=PTRCONV
-; RUN: %if ptxas-12.8 %{ llc -O0 < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx78 | %ptxas-verify %}
-; RUN: %if ptxas-12.8 %{ llc -O0 < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx78 --nvptx-short-ptr | %ptxas-verify %}
+; RUN: %if ptxas-12.8 %{ llc -O0 < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx78 | %ptxas-verify -arch=sm_90 %}
+; RUN: %if ptxas-12.8 %{ llc -O0 < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx78 --nvptx-short-ptr | %ptxas-verify -arch=sm_90 %}
 
 ; ALL-LABEL: conv_shared_cluster_to_generic
 define i32 @conv_shared_cluster_to_generic(ptr addrspace(7) %ptr) {
diff --git a/llvm/test/CodeGen/NVPTX/distributed-shared-cluster.ll b/llvm/test/CodeGen/NVPTX/distributed-shared-cluster.ll
index 8b6c554aeb9f2..10f3b9f36f95a 100644
--- a/llvm/test/CodeGen/NVPTX/distributed-shared-cluster.ll
+++ b/llvm/test/CodeGen/NVPTX/distributed-shared-cluster.ll
@@ -1,6 +1,6 @@
 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
 ; RUN: llc < %s -o - -mcpu=sm_90 -mattr=+ptx78 | FileCheck %s
-; RUN: %if ptxas-12.0 %{ llc < %s -mcpu=sm_90 -mattr=+ptx78| %ptxas-verify -arch=sm_90 %}
+; RUN: %if ptxas-12.8 %{ llc < %s -mcpu=sm_90 -mattr=+ptx78| %ptxas-verify -arch=sm_90 %}
 
 target triple = "nvptx64-nvidia-cuda"
 

[AlexMaclean]

LGTM, assuming this has been verified locally with LLVM_PTXAS_EXECUTABLE

[modiking]

LGTM, assuming this has been verified locally with LLVM_PTXAS_EXECUTABLE

Yep. Thanks for the quick review!

/data/mmo/llvm-project/build-rel$ LLVM_PTXAS_EXECUTABLE=/data/mmo/cuda-12.8/bin/ptxas /data/mmo/llvm-project/build-rel/bin/llvm-lit /data/mmo/llvm-project/llvm/test/CodeGen/NVPTX/distributed-shared-cluster.ll /data/mmo/llvm-project/llvm/tes
t/CodeGen/NVPTX/addrspacecast-ptx64.ll
-- Testing: 2 tests, 2 workers --
PASS: LLVM :: CodeGen/NVPTX/distributed-shared-cluster.ll (1 of 2)
PASS: LLVM :: CodeGen/NVPTX/addrspacecast-ptx64.ll (2 of 2)

Testing Time: 0.44s

Total Discovered Tests: 2

Passed: 2 (100.00%)