Commit efac0323 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

remove unnecessary code

parent 8a7e325c
...@@ -186,9 +186,9 @@ auto layernorm_fusion(hipStream_t stream, ...@@ -186,9 +186,9 @@ auto layernorm_fusion(hipStream_t stream,
return [=](auto input, auto output) { return [=](auto input, auto output) {
auto relements = arg1.get_shape().lens().back(); auto relements = arg1.get_shape().lens().back();
auto nelements = result.get_shape().elements() / relements; auto nelements = result.get_shape().elements() / relements;
auto output_shape = result.get_shape(); // auto output_shape = result.get_shape();
auto reduce_output_lens(output_shape.lens()); // auto reduce_output_lens(output_shape.lens());
reduce_output_lens.back() = 1; // reduce_output_lens.back() = 1;
if((relements % 4) == 0) if((relements % 4) == 0)
layernorm_vec_impl<4>( layernorm_vec_impl<4>(
......
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