[XPU][TritonGPUToLLVM] Use reqd_work_group_size

Use `llvm.func`'s `reqd_work_group_size` instead of `max_work_group_size` to express known work-group size information. Signed-off-by: victor-eds <[email protected]>
intel · Nov 27, 2024 · 109de74 · 109de74
1 parent ee78046
commit 109de74
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/third_party/intel/lib/TritonIntelGPUToLLVM/PipelineManager.h b/third_party/intel/lib/TritonIntelGPUToLLVM/PipelineManager.h
@@ -116,9 +116,8 @@ struct FuncOpConversion : public ConvertOpToLLVMPattern<triton::FuncOp> {
       newFuncOp.setLinkage(LLVM::Linkage::External);
     }
 
-    newFuncOp->setAttr(
-        TritonGEN::TritonGENDialect::getMaxWorkGroupSizeAttrName(),
-        rewriter.getDenseI32ArrayAttr({threadsPerWarp * numWarps, 1, 1}));
+    newFuncOp.setReqdWorkGroupSize(
+        ArrayRef<int>{threadsPerWarp * numWarps, 1, 1});
     newFuncOp.setIntelReqdSubGroupSize(threadsPerWarp);
 
     if (!LLVM::isKernel(funcOp)) {