github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BACKEND] Merge TT_ElementwisePureExtern and TT_ElementwiseImpureExtern (#2137)

Browse Source 
Use getEffect instead to tell passes whether the op has side effects or
not. This doesn't change functionality otherwise.
This commit is contained in:
Thomas
2023-08-18 13:56:10 -07:00
committed by GitHub
parent bf351b9ba2
commit 23ef2615d2
6 changed files with 34 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/src/triton.cc
View File

@@ -1424,12 +1424,8 @@ void init_triton_ir(py::module &&m) {
const std::string &libPath, const std::string &symbol,
std::vector<mlir::Value> &argList, mlir::Type retType,
bool isPure) -> mlir::Value {
if (isPure)
return self.create<mlir::triton::PureExternElementwiseOp>(
retType, argList, libName, libPath, symbol);
else
return self.create<mlir::triton::ImpureExternElementwiseOp>(
retType, argList, libName, libPath, symbol);
return self.create<mlir::triton::ExternElementwiseOp>(
retType, argList, libName, libPath, symbol, isPure);
})
// Built-in instruction
.def("create_get_program_id",