gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[BACKEND] Add generic support of convert_layout from distributed to shared (#1025)

Browse Source
This commit is contained in:
goostavz
2022-12-31 03:29:58 +08:00
committed by GitHub
parent 194ba103b1
commit 0e8590f1c9
5 changed files with 68 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/Conversion/TritonGPUToLLVM/ConvertLayoutOpToLLVM.h
View File

@@ -11,10 +11,12 @@ using ::mlir::triton::gpu::DotOperandEncodingAttr;
bool isMmaToDotShortcut(MmaEncodingAttr &mmaLayout,
DotOperandEncodingAttr &dotOperandLayout);
void storeBlockedToShared(Value src, Value llSrc, ArrayRef<Value> srcStrides,
ArrayRef<Value> srcIndices, Value dst, Value smemBase,
Type elemPtrTy, Location loc,
ConversionPatternRewriter &rewriter);
void storeDistributedToShared(Value src, Value llSrc,
ArrayRef<Value> srcStrides,
ArrayRef<SmallVector<Value>> srcIndices,
Value dst, Value smemBase, Type elemPtrTy,
Location loc,
ConversionPatternRewriter &rewriter);
void populateConvertLayoutOpToLLVMPatterns(
mlir::LLVMTypeConverter &typeConverter, RewritePatternSet &patterns,