"...resnet50_tensorflow.git" did not exist on "2d2f79eb96e5bee47992232581a685ce0a1a1c0a"
Commit 2bba1c7c authored by Paul's avatar Paul
Browse files

Check type for fp32

parent 24ea1c41
...@@ -60,6 +60,11 @@ struct find_conv_pointwise ...@@ -60,6 +60,11 @@ struct find_conv_pointwise
i.name()); i.name());
})) }))
return; return;
// Only fuse with fp32 for now
if (std::any_of(ins->inputs().begin(), ins->inputs().end(), [&](auto i) {
return i->get_shape().type() != shape::type_t::float_type;
}))
return;
std::sort(names.begin(), names.end()); std::sort(names.begin(), names.end());
module_ref mm = mpm.create_module("mlir_" + pm->name()); module_ref mm = mpm.create_module("mlir_" + pm->name());
mm->set_bypass(); mm->set_bypass();
......
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