Commit 08ffe31b authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Add case for divzero in insert_generic_instructions

parent 3a84c350
...@@ -247,9 +247,16 @@ insert_generic_instructions(module& m, ...@@ -247,9 +247,16 @@ insert_generic_instructions(module& m,
mod_outputs = copy_inputs; mod_outputs = copy_inputs;
break; break;
} }
else if(sins->name() == "@divzero")
{
shape s = sins->get_shape();
copy_ins = m.add_divzero(inputs, {s});
}
else
{
copy_ins = m.insert_instruction(ins, sins->get_operator(), copy_inputs, mod_args); copy_ins = m.insert_instruction(ins, sins->get_operator(), copy_inputs, mod_args);
} }
}
map_ins[sins] = copy_ins; map_ins[sins] = copy_ins;
} }
if(mod_outputs.empty() and instructions.begin() != instructions.end()) if(mod_outputs.empty() and instructions.begin() != instructions.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