Limit hip gemm pointwise fusion to hip_gemm

ROCm · Dec 10, 2024 · 0a711d4 · 0a711d4
1 parent 64fe0c5
commit 0a711d4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/targets/gpu/fuse_ops.cpp b/src/targets/gpu/fuse_ops.cpp
@@ -715,6 +715,9 @@ struct find_hipblas_gemm_pointwise : gemm_pointwise
 
         auto gemm_op = any_cast<hipblaslt_op>(gemm_ins->get_operator()).op;
 
+        if(gemm_op.name() != "gpu::hip_gemm")
+            return;
+
         auto gemm = any_cast<hip_gemm<op::dot>>(gemm_op);
 
         // Already fused gemm