Update type_inference.py (#1029)

spcl · Jun 22, 2022 · 98e18ac · 98e18ac
1 parent f05391f
commit 98e18ac
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dace/codegen/tools/type_inference.py b/dace/codegen/tools/type_inference.py
@@ -436,7 +436,9 @@ def _Call(t, symbols, inferred_symbols):
     if name in ('abs', 'log'):
         return arg_types[0]
     if name in ('min', 'max'): # binary math operations that do not exist in the math module
-        return dtypes.result_type_of(arg_types[0], *arg_types)     
+        return dtypes.result_type_of(arg_types[0], *arg_types)
+    if name in ('round', ):
+        return dtypes.typeclass(int)   
 
     # dtypes (dace.int32, np.float64) can be used as functions
     inf_type = _infer_dtype(t)