[Mlir-commits] [mlir] [MLIR][NVVM] Print ptxas path (PR #132373)

[Guray Ozen]

https://github.com/grypp created https://github.com/llvm/llvm-project/pull/132373

None

>From dc9c9549135b09c2c0c12e06143721b4f186ec36 Mon Sep 17 00:00:00 2001
From: Guray Ozen <gozen at nvidia.com>
Date: Fri, 21 Mar 2025 11:51:54 +0100
Subject: [PATCH] [MLIR][NVVM] Print ptxas path

---
 mlir/lib/Target/LLVM/NVVM/Target.cpp                 | 1 +
 mlir/test/Integration/GPU/CUDA/command-line-arg.mlir | 1 +
 2 files changed, 2 insertions(+)

diff --git a/mlir/lib/Target/LLVM/NVVM/Target.cpp b/mlir/lib/Target/LLVM/NVVM/Target.cpp
index fa8c597da58b1..69a247b06bf16 100644
--- a/mlir/lib/Target/LLVM/NVVM/Target.cpp
+++ b/mlir/lib/Target/LLVM/NVVM/Target.cpp
@@ -455,6 +455,7 @@ NVPTXSerializer::compileToBinary(const std::string &ptxCode) {
   LLVM_DEBUG({
     llvm::dbgs() << "Tool invocation for module: "
                  << getOperation().getNameAttr() << "\n";
+    llvm::dbgs() << "ptxas executable:" << ptxasCompiler.value() << "\n";
     llvm::interleave(ptxasArgs, llvm::dbgs(), " ");
     llvm::dbgs() << "\n";
     if (createFatbin) {
diff --git a/mlir/test/Integration/GPU/CUDA/command-line-arg.mlir b/mlir/test/Integration/GPU/CUDA/command-line-arg.mlir
index 34dde6e03c80e..f65eda6a2a88d 100644
--- a/mlir/test/Integration/GPU/CUDA/command-line-arg.mlir
+++ b/mlir/test/Integration/GPU/CUDA/command-line-arg.mlir
@@ -16,6 +16,7 @@ func.func @host_function(%arg0 : f32, %arg1 : memref<?xf32>) {
     return
 }
 
+// CHECK: ptxas executable:
 // CHECK: ptxas -arch sm_80
 // CHECK-SAME: -v 
 // CHECK-SAME: --register-usage-level=8