|
|
|
@@ -107,9 +107,9 @@ void DataDumper::SetOpMappingInfo(NotNull<aicpu::dump::OpMappingInfo *> dump_inf |
|
|
|
MS_EXCEPTION_IF_NULL(currnet_epoch_tensor->device_address()); |
|
|
|
MS_EXCEPTION_IF_NULL(steps_per_epoch_tensor->device_address()); |
|
|
|
|
|
|
|
void *current_step = current_step_tensor->device_address()->ptr_; |
|
|
|
void *current_epoch = currnet_epoch_tensor->device_address()->ptr_; |
|
|
|
void *steps_per_epoch = steps_per_epoch_tensor->device_address()->ptr_; |
|
|
|
void *current_step = current_step_tensor->device_address()->GetMutablePtr(); |
|
|
|
void *current_epoch = currnet_epoch_tensor->device_address()->GetMutablePtr(); |
|
|
|
void *steps_per_epoch = steps_per_epoch_tensor->device_address()->GetMutablePtr(); |
|
|
|
|
|
|
|
if (current_epoch != nullptr && current_step != nullptr && steps_per_epoch != nullptr) { |
|
|
|
dump_info->set_step_id_addr(reinterpret_cast<uint64_t>(current_epoch)); |
|
|
|
|