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

Fix

Browse Source
This commit is contained in:
Jokeren
2022-12-06 13:24:50 -08:00
parent 74f3d7a80f
commit bab7338965
2 changed files with 10 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.cpp
View File

@@ -4105,10 +4105,12 @@ struct InsertSliceAsyncOpConversion
}
v = bitcast(v, IntegerType::get(getContext(), byteWidth));
// Write shared memory if predicate is true
auto *valOperand = ptxBuilder.newOperand(v, "r");
auto &st = *ptxBuilder.create<PTXInstr>("st");
PTXBuilder ptxStoreBuilder;
auto *valOperand = ptxStoreBuilder.newOperand(v, "r");
auto &st = *ptxStoreBuilder.create<PTXInstr>("st");
st.shared().o("b" + std::to_string(bitWidth));
st(dstOperand, valOperand).predicate(pred);
ptxStoreBuilder.launch(rewriter, loc, void_ty(getContext()));
}
}
copyAsyncOp(dstOperand, srcOperand, copySize, srcSize);