Commit fde204b0 authored by Amy Zhuang's avatar Amy Zhuang Committed by Scott Cyphers

Fix Convert for boolean output type in CODEGEN. (#2958)

parent f7b13dc4
......@@ -1352,8 +1352,16 @@ namespace ngraph
writer << "#pragma omp parallel for\n";
writer << "for (size_t i = 0; i < " << out[0].get_size() << "; i++)\n";
writer.block_begin();
writer << out[0].get_name() << "[i] = (" << result_element_type.c_type_string()
<< ")(" << args[0].get_name() << "[i]);\n";
writer << out[0].get_name() << "[i] = (";
if (result_element_type == element::boolean)
{
writer << "bool";
}
else
{
writer << result_element_type.c_type_string();
}
writer << ")(" << args[0].get_name() << "[i]);\n";
writer.block_end();
writer.block_end();
}
......
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