mirrors/llvm-project
0
0
Fork 0
mirror of https://github.com/llvm/llvm-project.git synced 2025-04-21 16:16:51 +00:00
Code Issues Projects Releases Wiki Activity

fix an error after mering with main

Browse Source
This commit is contained in:
Chao Chen 2025-04-04 18:08:07 +00:00
parent 2a16d11663
commit d281a149c2
1 changed files with 3 additions and 10 deletions

13
mlir/lib/Dialect/XeGPU/IR/XeGPUDialect.cpp

@ -298,17 +298,10 @@ LogicalResult TensorDescType::verify(
"contiguous elements";
}
// For 1D tensor, pad the shape with an outer unit dimension to allow common
// validation logic.
SmallVector<int64_t> tensorShape(shape);
if (rank == 1)
tensorShape = {1, tensorShape.back()};
size_t dims = tensorShape.size();
for (size_t i = 0; i < dims; ++i) {
for (size_t i = 0; i < shape.size(); ++i) {
uint32_t numElemPerWi = laneLayout[i] * laneData[i];
if (tensorShape[i] < numElemPerWi || tensorShape[i] % numElemPerWi != 0)
return emitError() << "cannot distribute " << tensorShape[i] << " over "
if (shape[i] < numElemPerWi || shape[i] % numElemPerWi != 0)
return emitError() << "cannot distribute " << shape[i] << " over "
<< laneLayout[i] << " work items with "
<< laneData[i] << " elements each";
}