saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-03 03:58:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix Numpy array None judgement bug (#13103)

Browse source 
fix https://github.com/microsoft/onnxruntime/issues/13054
This commit is contained in:
Chen Fu 2022-09-26 15:15:32 -07:00 committed by GitHub
parent a83a9ed6b0
commit e9b1bbc6a5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/python/tools/quantization/qdq_loss_debug.py
View file

@ -167,7 +167,7 @@ def _add_pre_post_qdq_pair(
pre_qdq_tensors: Optional[Sequence[numpy.ndarray]],
post_qdq_tensors: Optional[Sequence[numpy.ndarray]],
) -> None:
if post_qdq_tensors and pre_qdq_tensors:
if post_qdq_tensors is not None and pre_qdq_tensors is not None:
qdq_cmp[activation_name] = {}
qdq_cmp[activation_name]["pre_qdq"] = pre_qdq_tensors
qdq_cmp[activation_name]["post_qdq"] = post_qdq_tensors
@ -229,7 +229,7 @@ def create_activation_matching(
for act_name, act_values in qdq_cmp.items():
float_acts = float_activations.get(act_name)
if float_acts:
if float_acts is not None:
act_values["float"] = float_acts
return qdq_cmp

7
onnxruntime/test/python/quantization/test_qdq_loss_debug.py
View file

@ -20,6 +20,7 @@ import onnxruntime
from onnxruntime.quantization import QuantFormat, QuantType, quantize_static
from onnxruntime.quantization.calibrate import CalibrationDataReader
from onnxruntime.quantization.qdq_loss_debug import (
QUANT_INPUT_SUFFIX,
collect_activations,
compute_activation_error,
compute_weight_error,
@ -318,6 +319,12 @@ class TestSaveActivations(unittest.TestCase):
dq_array = matched_weights[weight_name]["dequantized"]
self.assertEqual(float_array.shape, dq_array.shape)
def test_none_test(self):
a = np.array([2, 3, 4])
b = np.array([7, 8, 9])
c = np.array([1, 2, 3])
create_activation_matching({"test" + QUANT_INPUT_SUFFIX: a, "test": c}, {"test": b})
if __name__ == "__main__":
unittest.main()