"...composable_kernel_rocm.git" did not exist on "b74918bc6a82ab8f5636286b378e41b1e19a084c"
Commit 50141f2a authored by Paul's avatar Paul
Browse files

Fix size

parent f977f3d3
......@@ -1208,7 +1208,7 @@ struct find_concat_pointwise
inputs.insert(inputs.end(), ins->inputs().begin() + 1, ins->inputs().end());
auto op = concat->get_operator();
op.from_value({{"additional_args", ins->inputs().size()}, {"ignore_modules", true}});
op.from_value({{"additional_args", ins->inputs().size() - 1}, {"ignore_modules", true}});
m.replace_instruction(ins, op, inputs, {pm});
}
......
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