Skip to content

Commit ce063a9

Browse files
committed
[InfCost] add allow_scaledint_dtypes=True earlier
1 parent c34bd9a commit ce063a9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/qonnx/util/inference_cost.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def inference_cost(
114114
qnt_node.domain = "qonnx.custom_op.general"
115115
model = model.transform(InferShapes())
116116
model = model.transform(GiveUniqueParameterTensors())
117-
model = model.transform(InferDataTypes())
117+
model = model.transform(InferDataTypes(allow_scaledint_dtypes=True))
118118
model = model.transform(FoldConstants(exclude_op_types=[]))
119119
model = model.transform(RemoveUnusedTensors())
120120
model = model.transform(RemoveStaticGraphInputs())

0 commit comments

Comments
 (0)