From 97fb6ce33b1ecda6916837e8587e6ca66a65fa67 Mon Sep 17 00:00:00 2001
From: Bhatu <prbhatu@microsoft.com>
Date: Wed, 23 Dec 2020 10:16:51 +0530
Subject: [PATCH] Fix typo for Conv2D truncation

---
 Athos/SeeDot/IR/IRBuilderCSF.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Athos/SeeDot/IR/IRBuilderCSF.py b/Athos/SeeDot/IR/IRBuilderCSF.py
index fb0ce8f..de0cf2f 100644
--- a/Athos/SeeDot/IR/IRBuilderCSF.py
+++ b/Athos/SeeDot/IR/IRBuilderCSF.py
@@ -890,7 +890,7 @@ class IRBuilderCSF(IRBuilderAST):
 			expr2_sf = self.scaleFacMapping[expr_2.idf]
 			if (expr1_sf > self.scaleFac):
 				progExtraBefore = self.addTruncateFunctionCall(node.expr1, "Conv", expr_1, expr1_sf-self.scaleFac)
-				self.scaleFacMapping[expr1.idf] = self.scaleFac
+				self.scaleFacMapping[expr_1.idf] = self.scaleFac
 			if (not inputs_same) and (expr2_sf > self.scaleFac):
 				progExtraBefore = IRUtil.prog_merge(progExtraBefore, self.addTruncateFunctionCall(node.expr2, "Conv", expr_2, expr2_sf-self.scaleFac))
 				self.scaleFacMapping[expr_2.idf] = self.scaleFac
-- 
GitLab