Unverified Commit 1339ba35 authored by Paul Fultz II's avatar Paul Fultz II Committed by GitHub
Browse files

Fix dangling reference with gemm add fusion (#1233)

parent 0cc6304d
...@@ -965,7 +965,7 @@ struct find_gemm_pointwise ...@@ -965,7 +965,7 @@ struct find_gemm_pointwise
inputs.pop_back(); inputs.pop_back();
inputs.push_back(c_ins); inputs.push_back(c_ins);
inputs.push_back(gemm_ins->inputs().back()); inputs.push_back(ins->inputs().back());
gemm.beta = 1; gemm.beta = 1;
m.replace_instruction(ins, gemm, inputs); m.replace_instruction(ins, gemm, inputs);
......
#include "verify_program.hpp"
#include <migraphx/program.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/make_op.hpp>
#include <migraphx/apply_alpha_beta.hpp>
struct gemm_add : verify_program<gemm_add>
{
migraphx::program create_program() const
{
migraphx::program p;
auto* mm = p.get_main_module();
migraphx::shape m1_shape{migraphx::shape::float_type, {1, 2, 3}};
migraphx::shape m2_shape{migraphx::shape::float_type, {1, 3, 4}};
migraphx::shape m3_shape{migraphx::shape::float_type, {1, 2, 4}};
auto l1 = mm->add_parameter("1", m1_shape);
auto l2 = mm->add_parameter("2", m2_shape);
auto l3 = mm->add_parameter("3", m3_shape);
auto dot = mm->add_instruction(migraphx::make_op("dot"), l1, l2);
mm->add_instruction(migraphx::make_op("add"), dot, l3);
return p;
}
};
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment