Fiy type check in find_pow2_div

ROCm · Mar 1, 2024 · e72105b · e72105b
1 parent 8973085
commit e72105b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/rewrite_low_precision.cpp b/src/rewrite_low_precision.cpp
@@ -54,8 +54,7 @@ struct find_pow2_div
         auto n   = r.instructions["n"];
         auto x   = r.instructions["x"];
 
-        // sqrt(n) with integral numbers can cause huge accuracy loss due to float->int rounding
-        if(n->get_shape().type() != migraphx::shape::half_type)
+        if(x->get_shape().type() != migraphx::shape::half_type)
             return;
         auto n_sqrt        = m.insert_instruction(ins, make_op("sqrt"), n);
         auto x_div_n_rsqrt = m.insert_instruction(ins, make_op("div"), {x, n_sqrt});