Merge remote-tracking branch 'origin/master' into phil/fused-attention-perf-fixup

This commit is contained in:
Phil Tillet
2022-12-30 11:53:49 -08:00
21 changed files with 765 additions and 337 deletions

View File

@@ -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