================
@@ -1508,19 +1517,20 @@ void XeGPUWgToSgDistributePass::runOnOperation() {
vector::ConstantMaskOp, vector::CreateMaskOp>(
[=](Operation *op) -> bool {
// Check for either a SliceAttr or LayoutAttr on the result.
- auto layout = xegpu::getDistributeLayoutAttr(op->getResult(0));
+ auto layout =
+ xegpu::getTempLayout(dyn_cast<OpResult>(op->getResult(0)));
return isLegal(layout);
});
target.addDynamicallyLegalOp<xegpu::LoadGatherOp>(
[=](xegpu::LoadGatherOp op) -> bool {
- auto layout = xegpu::getDistributeLayoutAttr(op.getResult());
+ auto layout = op.getLayoutAttr();
return isLegal(layout);
});
target.addDynamicallyLegalOp<xegpu::StoreScatterOp>(
[=](xegpu::StoreScatterOp op) -> bool {
- auto layout = xegpu::getDistributeLayoutAttr(op.getOperand(0));
+ auto layout = op.getLayoutAttr();
----------------
tkarna wrote:
getAnchorLayout
https://github.com/llvm/llvm-project/pull/172125
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
