[Mlir-commits] [mlir] [MLIR][ROCDL] Add conversion for gpu.subgroup_id to ROCDL (PR #136405)
Alan Li
llvmlistbot at llvm.org
Fri Apr 25 09:05:09 PDT 2025
================
@@ -406,15 +504,11 @@ void mlir::populateGpuToROCDLConversionPatterns(
patterns.add<GPUDynamicSharedMemoryOpLowering>(converter);
patterns.add<GPUShuffleOpLowering, GPULaneIdOpToROCDL>(converter);
-
+ patterns.add<GPUSubgroupIdOpToROCDL>(converter, chipset, subgroupSize);
populateMathToROCDLConversionPatterns(converter, patterns);
}
std::unique_ptr<OperationPass<gpu::GPUModuleOp>>
-mlir::createLowerGpuOpsToROCDLOpsPass(const std::string &chipset,
- unsigned indexBitwidth,
- bool useBarePtrCallConv,
- gpu::amd::Runtime runtime) {
----------------
lialan wrote:
We are adding a new input argument so might as well use the constructor emitted by tablegen here.
https://github.com/llvm/llvm-project/pull/136405
More information about the Mlir-commits
mailing list