From 9ddf0921fb752f3f428c8e550bd75d5c0bec8904 Mon Sep 17 00:00:00 2001
From: Philippe Tillet <phil@openai.com>
Date: Tue, 27 Sep 2022 09:05:49 -0700
Subject: [PATCH] [OPTIMIZER] Added `DotOp` to the list of expensive ops we
 don't want to rematerialize. (#718)

---
 lib/Dialect/TritonGPU/Transforms/Combine.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/Dialect/TritonGPU/Transforms/Combine.cpp b/lib/Dialect/TritonGPU/Transforms/Combine.cpp
index d879f451e..a354993a8 100644
--- a/lib/Dialect/TritonGPU/Transforms/Combine.cpp
+++ b/lib/Dialect/TritonGPU/Transforms/Combine.cpp
@@ -120,8 +120,8 @@ public:
 
     auto blacklist = [](Operation *op) {
       if (isa<triton::gpu::ExtractSliceOp, triton::gpu::AllocTensorOp,
-              triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp>(
-              op))
+              triton::gpu::InsertSliceAsyncOp, triton::LoadOp, triton::StoreOp,
+              triton::DotOp>(op))
         return true;
       if (isa<scf::YieldOp, scf::ForOp>(op))
         return true;