[Mlir-commits] [mlir] [XeGPU] Add sg_map for scatter verification (PR #124300)
Adam Siemieniuk
llvmlistbot at llvm.org
Tue Jan 28 12:02:14 PST 2025
================
@@ -453,7 +453,16 @@ LogicalResult CreateDescOp::verify() {
if (shape != tdescShape)
return emitOpError("Incorrect TensorDesc shape. ")
<< "Expected is " << makeString(shape) << "\n";
-
+ if (auto sgMap = tdescTy.getSGMapAttr()) {
+ if (sgMap.getWiData()[0] > 1)
+ return emitOpError("TensorDesc cannot have wi_data[0] > 1.");
+ if (chunkSize != static_cast<int>(sgMap.getWiData()[1]))
----------------
adam-smnk wrote:
I'm not sure about rules here. Wouldn't distribution still be valid when `chunk_size >= wi_data[1]`?
https://github.com/llvm/llvm-project/pull/124300
More information about the Mlir-commits
mailing list