Add case for divzero in insert_generic_instructions

08ffe31b · Ted Themistokleous · 3a84c350 · 08ffe31b
Commit 08ffe31b authored Jul 27, 2022 by Ted Themistokleous
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

src/module.cpp src/module.cpp +9 -2

No files found.
--- a/src/module.cpp
+++ b/src/module.cpp
@@ -247,8 +247,15 @@ insert_generic_instructions(module& m,
                mod_outputs = copy_inputs;
                break;
            }
+            else if(sins->name() == "@divzero")
-            copy_ins = m.insert_instruction(ins, sins->get_operator(), copy_inputs, mod_args);
+            {
+                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);
+            }
        }
        map_ins[sins] = copy_ins;
    }