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 50a543d69260b4ee331a9da3bc09ba8ff62f80ad..b1d7550a40254da8098b0e40d5941fc5d58591d9 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 @@ -274,6 +274,10 @@ class PytorchDataProcessor(BaseDataProcessor): def _analyze_and_save_tensor(self, tensor, suffix): dump_data_name, file_path = self.get_save_file_path(suffix) single_arg = PytorchDataProcessor._analyze_tensor(self, tensor, suffix) + if tensor.storage().data_ptr() == 0: + logger.debug("Collecting tensor data_ptr is 0, Skip.") + return single_arg + single_arg.update({"data_name": dump_data_name}) if self.config.async_dump: self._async_dump_cache[file_path] = tensor.clone().detach()