Runs on all bench

li3zhen1 · Mar 19, 2024 · d14fac3 · d14fac3
1 parent ca07b8a
commit d14fac3
Show file tree

Hide file tree

Showing 5 changed files with 171 additions and 137 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/benchmark/micro_benchmark/test/test3.cpp b/benchmark/micro_benchmark/test/test3.cpp
@@ -0,0 +1,11 @@
+static int a[200] = {0};
+static int b[20][12] = {0};
+
+void tt(int t[20][20]) { t[1][1] = 1; }
+
+int main() {
+  int local1d[200] = {0};
+  local1d[42] = 77;
+  a[1] = 1;
+  b[1][1] = 1;
+}
diff --git a/src/BoundCheckInsertion.cpp b/src/BoundCheckInsertion.cpp
@@ -54,16 +54,15 @@ PreservedAnalyses BoundCheckInsertion::run(Function &F,
         Value *Bound =
             cast<ValueAsMetadata>(MN->getOperand(1).get())->getValue();
 
+        // llvm::errs() << "Unknown GEP type: ";
+        // GEP->print(llvm::errs());
         Value *subscript = nullptr;
         if (GEP->getSourceElementType()->isArrayTy()) {
-          subscript = GEP->getOperand(2);
+          subscript = GEP->getOperand(GEP->getNumIndices());
         } else if (GEP->getSourceElementType()->isIntegerTy()) {
           subscript = GEP->getOperand(1);
         } else {
-
-          // GEP->print(errs());
-          subscript = GEP->getOperand(1);
-          // throw std::runtime_error("Unknown GEP type.");
+          subscript = GEP->getOperand(GEP->getNumIndices());
         }
 
         createCheckBoundCall(&I, Bound, subscript);
@@ -102,6 +101,17 @@ PreservedAnalyses BoundCheckInsertion::run(Function &F,
       }
     }
   }
+
+
+  for (auto &BB : F) {
+    // llvm::errs() << "BB: ";
+    BB.printAsOperand(llvm::errs());
+    llvm::errs() << "\n";
+    for (auto &Inst : BB) {
+      Inst.print(llvm::errs());
+      llvm::errs() << "\n";
+    }
+  }
   return PreservedAnalyses::none();
 }