Use getCalledFunction (rust-lang#991)

* use getCalledFunction
EnzymeAD · Feb 12, 2023 · 23d6ecc · 23d6ecc
1 parent ce11d52
commit 23d6ecc
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/enzyme/Enzyme/TraceGenerator.h b/enzyme/Enzyme/TraceGenerator.h
@@ -123,15 +123,12 @@ class TraceGenerator final : public llvm::InstVisitor<TraceGenerator> {
       if (tutils->hasDynamicTraceInterface())
         args.push_back(tutils->getDynamicTraceInterface());
 
-      Function *samplefn = Logic.CreateTrace(
-          call.getCalledFunction(), tutils->generativeFunctions, tutils->mode,
-          tutils->hasDynamicTraceInterface());
+      Function *called = getFunctionFromCall(&call);
+      assert(called);
 
-#if LLVM_VERSION_MAJOR >= 11
-      Value *called = call.getCalledOperand();
-#else
-      Value *called = call.getCalledValue();
-#endif
+      Function *samplefn =
+          Logic.CreateTrace(called, tutils->generativeFunctions, tutils->mode,
+                            tutils->hasDynamicTraceInterface());
 
       Instruction *tracecall;
       switch (mode) {