Merge remote-tracking branch 'origin/master' into phil/fused-attention-perf-fixup
This commit is contained in:
@@ -257,6 +257,11 @@ SmallVector<unsigned> getOrder(const Attribute &layout) {
|
||||
}
|
||||
};
|
||||
|
||||
bool isaDistributedLayout(const Attribute &layout) {
|
||||
return layout.isa<BlockedEncodingAttr>() || layout.isa<MmaEncodingAttr>() ||
|
||||
layout.isa<SliceEncodingAttr>();
|
||||
}
|
||||
|
||||
} // namespace gpu
|
||||
} // namespace triton
|
||||
} // namespace mlir
|
||||
|
Reference in New Issue
Block a user