"include/ck/utility/magic_division.hpp" did not exist on "19c5d6e651d00d15b3909bf1ba44bf59df7f29cf"
Commit 11f33609 authored by turneram's avatar turneram
Browse files

Fix scale and bias ops

parent e47c0140
......@@ -31,9 +31,9 @@ struct parse_layernorm : op_parser<parse_layernorm>
make_op("layernorm", {{"epsilon", epsilon}, {"axis", axis}}), args.front());
if(args.size() >= 2)
layernorm = info.add_instruction(make_op("mul"), layernorm, args.at(1));
layernorm = info.add_broadcastable_binary_op("mul", layernorm, args.at(1));
if(args.size() == 3)
layernorm = info.add_instruction(make_op("add"), layernorm, args.at(2));
layernorm = info.add_broadcastable_binary_op("add", layernorm, args.at(2));
return layernorm;
}
......
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