"test/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "bd12f16bfcac426714de9a5d5f93c727de92da9f"
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