Skip to content

Commit 88cc25c

Browse files
committed
Try diff constructor
1 parent 3d95a81 commit 88cc25c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

third_party/intel/lib/TritonIntelGPUToLLVM/PipelineManager.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,8 +116,9 @@ struct FuncOpConversion : public ConvertOpToLLVMPattern<triton::FuncOp> {
116116
newFuncOp.setLinkage(LLVM::Linkage::External);
117117
}
118118

119-
newFuncOp.setReqdWorkGroupSize(
120-
ArrayRef<int>{1, 1, threadsPerWarp * numWarps});
119+
constexpr size_t numDims = 3;
120+
std::array<int, numDims> localSize{threadsPerWarp * numWarps, 1, 1};
121+
newFuncOp.setReqdWorkGroupSize(localSize);
121122
newFuncOp.setIntelReqdSubGroupSize(threadsPerWarp);
122123

123124
if (!LLVM::isKernel(funcOp)) {

0 commit comments

Comments
 (0)