Skip to content

Commit 9d54ae8

Browse files
authored
[mlir] Add opt-level to test-lower-to-nvvm Pipeline (#68183)
This PR adds the `opt-level` parameter to control code optimization for NVIDIA GPU targets in the `test-lower-to-nvvm` pipeline.
1 parent 8823e96 commit 9d54ae8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

mlir/test/lib/Dialect/GPU/TestLowerToNVVM.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,10 @@ struct TestLowerToNVVMOptions
7474
*this, "cubin-format",
7575
llvm::cl::desc("Compilation format to use to serialize to cubin."),
7676
llvm::cl::init("isa")};
77+
PassOptions::Option<int> optLevel{
78+
*this, "opt-level",
79+
llvm::cl::desc("Optimization level for NVVM compilation"),
80+
llvm::cl::init(2)};
7781
};
7882

7983
//===----------------------------------------------------------------------===//
@@ -242,6 +246,7 @@ void buildLowerToNVVMPassPipeline(OpPassManager &pm,
242246
nvvmTargetOptions.triple = options.cubinTriple;
243247
nvvmTargetOptions.chip = options.cubinChip;
244248
nvvmTargetOptions.features = options.cubinFeatures;
249+
nvvmTargetOptions.optLevel = options.optLevel;
245250
pm.addPass(createGpuNVVMAttachTarget(nvvmTargetOptions));
246251

247252
// Convert GPU to LLVM.

0 commit comments

Comments
 (0)