[Mlir-commits] [mlir] [MLIR][ROCDL] Add conversion for gpu.subgroup_id to ROCDL (PR #136405)
Alan Li
llvmlistbot at llvm.org
Fri Apr 25 18:48:57 PDT 2025
================
@@ -608,6 +608,10 @@ def ConvertGpuOpsToROCDLOps : Pass<"convert-gpu-to-rocdl", "gpu::GPUModuleOp"> {
clEnumValN(::mlir::gpu::amd::Runtime::HIP, "HIP", "HIP"),
clEnumValN(::mlir::gpu::amd::Runtime::OpenCL, "OpenCL",
"OpenCL"))}]>,
+ Option<"subgroupSize", "subgroup-size", "unsigned",
+ "0",
+ "specify subgroup size for the kernel, if left empty, the default "
+ "value will be decided by the target chipset.">,
----------------
lialan wrote:
@kuhar subgroup size op is added, so I have changed this to divide by using subgroup size op
https://github.com/llvm/llvm-project/pull/136405
More information about the Mlir-commits
mailing list