There was an error while loading. Please reload this page.
2 parents 7c86aee + f158494 commit 6044098Copy full SHA for 6044098
1 file changed
src/qonnx/transformation/infer_datatypes.py
@@ -50,6 +50,10 @@ def infer_mac_result_dtype(idtypes, odtype_orig, possible_negation):
50
ret = DataType["INT32"] if maybe_signed else DataType["UINT32"]
51
elif all([is_scaled_int(x) for x in idtypes]):
52
ret = DataType["SCALEDINT<32>"]
53
+ elif any(["FLOAT" in x.name for x in idtypes]):
54
+ # default to float32 if any inps are float
55
+ # TODO use output container dtype instead?
56
+ ret = DataType["FLOAT32"]
57
return ret
58
59
0 commit comments