diff --git a/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py b/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py index 2d836479177be2c6b149bba0af74d772f8e0950d..738e36afc8130711ab4709ee6907680f985d7230 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/json_writer.py @@ -157,8 +157,6 @@ class DataWriter: logger.warning("Some bad data in async dump") else: tensor_stat_index, tensor_stat_data = tensor_stat[0], tensor_stat[1] - if hasattr(tensor_stat_data, "device") and tensor_stat_data.device != Const.CPU_LOWERCASE: - tensor_stat_data = tensor_stat_data.cpu() for index, stat in zip(tensor_stat_index, tensor_stat_data): data.update({index: stat.item()}) del data["tensor_stat"]