Fault injection minor fix (#21)

* Minor fix in fault injection for newer ML models
DependableSystemsLab · Jul 23, 2022 · aaf9a53 · aaf9a53
1 parent c3ce575
commit aaf9a53
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/llvm_passes/CustomTensorOperatorInstSelector.cpp b/llvm_passes/CustomTensorOperatorInstSelector.cpp
@@ -88,7 +88,10 @@ class CustomTensorOperatorInstSelector : public HardwareFIInstSelector {
                 {"matmul", 119251066446157},
                 {"add", 6579265},
                 {"avgpool", 30521821365761601},
-                {"softmax", 33884119937478483}
+                {"softmax", 33884119937478483},
+                {"loop", 1886351180},
+                {"nonmaxs", 23494782373228366},
+                {"unsqueeze", 28540527736745557}
             };
 
             if (ONNXOperatorId.find(opname) == ONNXOperatorId.end())

diff --git a/runtime_lib/FaultInjectionLib.c b/runtime_lib/FaultInjectionLib.c
@@ -175,8 +175,9 @@ void initInjections() {
 
 bool preFunc(long llfi_index, unsigned opcode, unsigned my_reg_index, 
              unsigned total_reg_target_num) {
-  assert(opcodecyclearray[opcode] >= 0 && 
-          "opcode does not exist, need to update instructions.def");
+  if (opcodecyclearray[opcode] < 0 &&
+          "opcode does not exist, need to update instructions.def")
+     return false;
 
    if (! fiFlag) return false;
    if (my_reg_index == 0)