remove unnecessary data copy related to rocblas api call

ae59a3b1 · Shucai Xiao · 9f06859b · ae59a3b1
Commit ae59a3b1 authored Mar 09, 2022 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 7 deletions

src/targets/gpu/fuse_ops.cpp src/targets/gpu/fuse_ops.cpp +7 -7

No files found.
--- a/src/targets/gpu/fuse_ops.cpp
+++ b/src/targets/gpu/fuse_ops.cpp
@@ -916,15 +916,15 @@ struct find_gemm_add
        auto inputs = gemm_ins->inputs();
        inputs.pop_back();

-        auto copy_ins = c_ins;
+        // auto copy_ins = c_ins;

        // Insert copy
-        if(ins == p.end() or c_ins->outputs().size() > 1 or c_ins->inputs().empty())
-        {
-            copy_ins = p.insert_instruction(ins, hip_copy{}, c_ins, ins->inputs().back());
-        }
-        inputs.push_back(copy_ins);
-        inputs.push_back(copy_ins);
+        // if(ins == p.end() or c_ins->outputs().size() > 1 or c_ins->inputs().empty())
+        // {
+        //     copy_ins = p.insert_instruction(ins, hip_copy{}, c_ins, ins->inputs().back());
+        // }
+        inputs.push_back(c_ins);
+        inputs.push_back(gemm_ins->inputs().back());

        gemm.beta = 1;
        p.replace_instruction(ins, gemm, inputs);