From 799da4757c38bd5189c6ad3b8017f4b9972e2801 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Sun, 14 Jul 2019 18:03:36 -0500
Subject: [PATCH] Fixing BatchNorm HPVM Frontend translation bug
---
llvm/projects/keras/frontend/hpvm_dfg_translator.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/llvm/projects/keras/frontend/hpvm_dfg_translator.py b/llvm/projects/keras/frontend/hpvm_dfg_translator.py
index e66ed64dff..066d5d23b9 100644
--- a/llvm/projects/keras/frontend/hpvm_dfg_translator.py
+++ b/llvm/projects/keras/frontend/hpvm_dfg_translator.py
@@ -365,7 +365,7 @@ class HPVMTranslator:
inst_str += " void *r = __visc__tensor_batchnorm(t1, t2, t3, t4, t5, "
inst_str += str(cur_node.epsilon) + "); \n"
- footer_str = self.genNodeFooter(5)
+ footer_str = self.genNodeFooter(2)
inst_str += footer_str
self.node_str += inst_str
--
GitLab