Skip to content

Commit dc9c954

Browse files
committed
[MLIR][NVVM] Print ptxas path
1 parent 44b9f5e commit dc9c954

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

mlir/lib/Target/LLVM/NVVM/Target.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -455,6 +455,7 @@ NVPTXSerializer::compileToBinary(const std::string &ptxCode) {
455455
LLVM_DEBUG({
456456
llvm::dbgs() << "Tool invocation for module: "
457457
<< getOperation().getNameAttr() << "\n";
458+
llvm::dbgs() << "ptxas executable:" << ptxasCompiler.value() << "\n";
458459
llvm::interleave(ptxasArgs, llvm::dbgs(), " ");
459460
llvm::dbgs() << "\n";
460461
if (createFatbin) {

mlir/test/Integration/GPU/CUDA/command-line-arg.mlir

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ func.func @host_function(%arg0 : f32, %arg1 : memref<?xf32>) {
1616
return
1717
}
1818

19+
// CHECK: ptxas executable:
1920
// CHECK: ptxas -arch sm_80
2021
// CHECK-SAME: -v
2122
// CHECK-SAME: --register-usage-level=8

0 commit comments

Comments
 (0)