From 98e18acd752f7d14f48c77e5bac6ca307ea713c1 Mon Sep 17 00:00:00 2001 From: Tal Ben-Nun Date: Wed, 22 Jun 2022 09:57:51 +0200 Subject: [PATCH] Update type_inference.py (#1029) --- dace/codegen/tools/type_inference.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dace/codegen/tools/type_inference.py b/dace/codegen/tools/type_inference.py index dbc0116968..a57becf195 100644 --- 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)