Skip to content

Commit 81d53ba

Browse files
committed
Specify scales in float32 always
Signed-off-by: Ali Boubezari <[email protected]>
1 parent b9a8e3d commit 81d53ba

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/onnx/autocast/precisionconverter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
OP_TYPES_NOT_SUPPORTED_IN_LOW_PRECISION.extend(["Scan", "If", "Loop", "LSTM"])
5454

5555
# Mapping of op types to indices of inputs that should not be converted to low precision.
56-
SKIP_LOW_PRECISION_MAPPING_FP16 = {"Resize": {1}}
56+
SKIP_LOW_PRECISION_MAPPING_FP16 = {"Resize": {2}}
5757
SKIP_LOW_PRECISION_MAPPING_BF16 = {"Resize": {1, 2}}
5858

5959

0 commit comments

Comments
 (0)