fix: Fix fuse addmm pass

Signed-off-by: Dheeraj Peri <[email protected]>
p517332051 · Dec 2, 2021 · 58e9ea0 · 58e9ea0
1 parent 4d2cb14
commit 58e9ea0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/lowering/passes/fuse_addmm_branches.cpp b/core/lowering/passes/fuse_addmm_branches.cpp
@@ -49,7 +49,7 @@ struct AddMMBranchFusion {
     if ((*arm1_start)->kind().toQualString() == std::string("aten::addmm") &&
         (*(++arm1_start))->kind() == prim::Return &&
         (*arm2_start)->kind().toQualString() == std::string("aten::matmul") &&
-        (*(++arm2_start))->kind().toQualString() != std::string("aten::add") &&
+        (*(++arm2_start))->kind().toQualString() == std::string("aten::add") &&
         (*(++arm2_start))->kind() == prim::Return) {
       // Make sure that block0 is solely just the aten::addmm op and block1 is matmul + add
       return true;