diff --git a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py index bc72d32b6ae06b04b7102476771537cdf5ab51c0..3a4d096667ece74f2f5c4ebef96f54f1e974af85 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py @@ -197,7 +197,7 @@ class MindsporeDataProcessor(BaseDataProcessor): if self.config.summary_mode == Const.MD5 and not self.config.async_dump: tensor = convert_bf16_to_fp32(tensor) # 拷贝并搬到 CPU - tensor_bytes = tensor.asnumpy().tobytes() + tensor_bytes = tensor.asnumpy() future = self._crc_executor.submit( MindsporeDataProcessor.compute_crc32_bytes, diff --git a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py index c3d6ebe5ea3d37c3680dc2c00483dbc91c4a48a6..cfa974b9aa82c03a2f65e6c43a385fbd9cd16e2f 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/pytorch_processor.py @@ -303,7 +303,7 @@ class PytorchDataProcessor(BaseDataProcessor): # 拷贝并搬到 CPU if tensor.dtype == torch.bfloat16: tensor = tensor.float() - tensor_bytes = tensor.cpu().detach().numpy().tobytes() + tensor_bytes = tensor.cpu().detach().numpy() future = self._crc_executor.submit( PytorchDataProcessor.compute_crc32_bytes,