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

[OPTIMIZER] Added DotOp to the list of expensive ops we don't want to rematerialize. (#718)

Browse Source
This commit is contained in:
Philippe Tillet
2022-09-27 09:05:49 -07:00
committed by GitHub
parent df8d276089
commit 9ddf0921fb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Dialect/TritonGPU/Transforms/Combine.cpp
View File

@@ -120,8 +120,8 @@ public:
auto blacklist = [](Operation *op) { auto blacklist = [](Operation *op) {
if (isa<triton::gpu::ExtractSliceOp, triton::gpu::AllocTensorOp, if (isa<triton::gpu::ExtractSliceOp, triton::gpu::AllocTensorOp,
triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp>( triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp,
op)) triton::DotOp>(op))
return true; return true;
if (isa<scf::YieldOp, scf::ForOp>(op)) if (isa<scf::YieldOp, scf::ForOp>(op))
return true; return true;