Commit 119a6b86 authored by Umang Yadav's avatar Umang Yadav
Browse files

Try removing fusing converts

parent 40e7698a
...@@ -344,6 +344,10 @@ bool is_pointwise_op_supported_by_mlir(const instruction& i) ...@@ -344,6 +344,10 @@ bool is_pointwise_op_supported_by_mlir(const instruction& i)
// supported. // supported.
if(is_float and name == "convert") if(is_float and name == "convert")
{ {
if(result_type == shape::fp8e4m3fnuz_type)
{
return false;
} // else
return std::all_of(i.inputs().begin(), i.inputs().end(), [](const auto& arg) { return std::all_of(i.inputs().begin(), i.inputs().end(), [](const auto& arg) {
return contains({type_t::float_type, type_t::half_type}, arg->get_shape().type()); return contains({type_t::float_type, type_t::half_type}, arg->get_shape().type());
}); });
......
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