From f026fa9601e204f89e83aae35c2bf727f40cf2a8 Mon Sep 17 00:00:00 2001
From: Maria Kotsifakou <kotsifa2@illinois.edu>
Date: Wed, 30 Oct 2019 21:02:59 -0500
Subject: [PATCH] Fixed batchnorm bug in wrapper backend
---
.../Transforms/DFG2LLVM_WrapperAPI/DFG2LLVM_WrapperAPI.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/llvm/lib/Transforms/DFG2LLVM_WrapperAPI/DFG2LLVM_WrapperAPI.cpp b/llvm/lib/Transforms/DFG2LLVM_WrapperAPI/DFG2LLVM_WrapperAPI.cpp
index 7ea0c1dce2..c54dd9ef3b 100644
--- a/llvm/lib/Transforms/DFG2LLVM_WrapperAPI/DFG2LLVM_WrapperAPI.cpp
+++ b/llvm/lib/Transforms/DFG2LLVM_WrapperAPI/DFG2LLVM_WrapperAPI.cpp
@@ -857,7 +857,7 @@ errs() << "TensorII: " << *TensorII << "\n";
Args.push_back(TensorII->getOperand(7));
// Create wrapper API runtime function call
- Constant* wrapper_tensorGroupConvolution;
+ Constant* wrapper_tensorGroupConvolution =
M->getOrInsertFunction(StringRef("wrapper_tensorGroupConvolution"),
RtM->getFunction(StringRef("wrapper_tensorGroupConvolution"))->getFunctionType());
CallInst* CI = CallInst::Create(wrapper_tensorGroupConvolution,
@@ -895,9 +895,9 @@ errs() << "TensorII: " << *TensorII << "\n";
Args.push_back(TensorII->getOperand(3));
Args.push_back(TensorII->getOperand(4));
Args.push_back(TensorII->getOperand(5));
-
+
// Create wrapper API runtime function call
- Constant* wrapper_tensorBatchNorm;
+ Constant* wrapper_tensorBatchNorm =
M->getOrInsertFunction(StringRef("wrapper_tensorBatchNorm"),
RtM->getFunction(StringRef("wrapper_tensorBatchNorm"))->getFunctionType());
CallInst* CI = CallInst::Create(wrapper_tensorBatchNorm,
--
GitLab