diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py index 72f3bca37453cbdec02cc858db5c4427a6a27d72..4250464b6833898c46aacebad584db19cd1094d2 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py @@ -23,7 +23,7 @@ import stat import subprocess import sys import time -import hashlib +import zlib import json from datetime import datetime, timezone from functools import wraps @@ -756,9 +756,9 @@ def check_file_valid(file_path): def get_md5_for_tensor(x): - tensor_bytes = x.cpu().detach().float().numpy().tobytes() - md5_hash = hashlib.md5(tensor_bytes) - return md5_hash.hexdigest() + tensor_bytes = x.cpu().detach().numpy().tobytes() + crc_hash = zlib.crc32(tensor_bytes) + return crc_hash def check_path_before_create(path): diff --git a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_utils.py b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_utils.py index 7a0a34a0c81863806daf9d9b2dc23e3e9af0a1f8..18ef57b9c1219ea591c4e473577f08ecf5bebd18 100644 --- a/debug/accuracy_tools/ptdbg_ascend/test/ut/test_utils.py +++ b/debug/accuracy_tools/ptdbg_ascend/test/ut/test_utils.py @@ -41,4 +41,4 @@ class TestUtilsMethods(unittest.TestCase): data = [[1, 2], [3, 4]] x_data = torch.tensor(data) md5_value = get_md5_for_tensor(x_data) - self.assertEqual(md5_value, "b4ad64bc24f643726bddf5606af738da") + self.assertEqual(md5_value, 2624136704)