Automatically scale down final ouptut.

Previously if there are mul like ops, the final output would be of
scale = 2 * scaling_factor. Now we introduce a scale down (if required)
so that the scale of the model output is = scaling_factor.

Athos/SeeDot/AST/IRBuilderAST.py

+'''
+
+Authors: Pratik Bhatu.
+
+Copyright:
+Copyright (c) 2020 Microsoft Research
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+'''
+
+import AST.AST as AST
+from AST.ASTVisitor import ASTVisitor
+
+class IRBuilderAST(ASTVisitor):
+	typeInfo = {}
+	def visit(self, node, args=None):
+		ret = super().visit( node, args)
+		if type(ret) is tuple:
+			if ret[1].idf not in self.typeInfo:
+				self.typeInfo[ret[1].idf] = node.type
+		return ret
\ No newline at end of file

Athos/SeeDot/Compiler.py

@@ -29,6 +29,7 @@ import Util
 import IR.IR as IR
 import AST.AST as AST
 from Writer import Writer
+import Type as Type
 from Type import InferType
 import IR.IRUtil as IRUtil
 from AST.PrintAST  import PrintAST
@@ -77,6 +78,34 @@ class Compiler:
 		prog.cmd_l.append(IR.FuncCall('EndComputation', []))
 		return (prog, expr)
 
+	def fixOuputScale(self, res:(IR.Prog, IR.Expr), compiler:IRBuilderCSF):
+		prog = res[0]
+		expr = res[1]
+		output_scale = compiler.scaleFacMapping[expr.idf]
+		if output_scale == Util.Config.consSF:
+			return (prog, expr)
+		elif output_scale > Util.Config.consSF:
+			scale_down =  output_scale - Util.Config.consSF
+			type = compiler.typeInfo[expr.idf]
+			if Type.isInt(type):
+				output_shape = []
+			if Type.isTensor(type):
+				output_shape = type.shape
+
+			argsDict = OrderedDict()
+			funcName = "ScaleDown"
+			for ii, curDimSize in enumerate(output_shape):
+				argsDict[IR.Int(curDimSize, 32)] = "size_" + str(ii)
+			funcName = funcName + str(len(output_shape))
+			argsDict[expr] = "expr"
+			argsDict[IR.Int(scale_down,32)] = "consSF"
+			funcCall = IR.FuncCall(funcName, argsDict)
+			new_prog = IR.Prog([funcCall])
+			prog = IRUtil.prog_merge(prog, new_prog)
+			return (prog, expr)
+		else:
+			assert False, "Scale up shouldnt be required of final output. We lost precision somewhere"
+
 	def run(self):
 		with open(Util.Config.astFile, 'rb') as ff:
 			ast = pickle.load(ff)
@@ -104,6 +133,7 @@ class Compiler:
 		IRUtil.init()
 		compiler = IRBuilderCSF()
 		res = compiler.visit(ast)
+		res = self.fixOuputScale(res, compiler);
 
 		Util.write_debug_info(compiler.name_mapping) 
 

Athos/SeeDot/IR/IRBuilderCSF.py

@@ -33,8 +33,9 @@ import Type as Type
 import AST.AST as AST
 import IR.IRUtil as IRUtil
 from AST.ASTVisitor import ASTVisitor
+from AST.IRBuilderAST import IRBuilderAST
 
-class IRBuilderCSF(ASTVisitor):
+class IRBuilderCSF(IRBuilderAST):
 	varNameDelim = ''
 	def __init__(self, intPartBitwidth=-1):
 		# For tracking temp variables