[Mlir-commits] [mlir] [MLIR][ROCDL] Add conversion for gpu.subgroup_id to ROCDL (PR #136405)
Jakub Kuderski
llvmlistbot at llvm.org
Fri Apr 25 08:59:25 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.">,
----------------
kuhar wrote:
+1
https://github.com/llvm/llvm-project/pull/136405
More information about the Mlir-commits
mailing list