|
|
|
@@ -121,8 +121,8 @@ void E2eDumpUtil::DumpOutput(const session::KernelGraph *graph, const std::strin |
|
|
|
auto type = AnfAlgo::GetOutputInferDataType(node, j); |
|
|
|
std::string file_path = dump_path + '/' + kernel_name + '_' + "output_" + std::to_string(j); |
|
|
|
if (IsDeviceTargetGPU()) { |
|
|
|
DumpGPUMemToFile(file_path, node->fullname_with_scope(), NOT_NULL(addr), trans_flag, int_shapes, type, j, |
|
|
|
debugger); |
|
|
|
DumpGPUMemToFile(file_path, node->fullname_with_scope() + "_output", NOT_NULL(addr), trans_flag, int_shapes, |
|
|
|
type, j, debugger); |
|
|
|
} else { |
|
|
|
DumpMemToFile(file_path, NOT_NULL(addr), trans_flag, int_shapes, type); |
|
|
|
} |
|
|
|
@@ -160,8 +160,8 @@ void E2eDumpUtil::DumpInput(const session::KernelGraph *graph, const std::string |
|
|
|
auto type = AnfAlgo::GetOutputInferDataType(input, index); |
|
|
|
std::string file_path = dump_path + '/' + kernel_name + '_' + "input_" + std::to_string(j); |
|
|
|
if (IsDeviceTargetGPU()) { |
|
|
|
DumpGPUMemToFile(file_path, node->fullname_with_scope(), NOT_NULL(addr), trans_flag, int_shapes, type, j, |
|
|
|
debugger); |
|
|
|
DumpGPUMemToFile(file_path, node->fullname_with_scope() + "_input", NOT_NULL(addr), trans_flag, int_shapes, |
|
|
|
type, j, debugger); |
|
|
|
} else { |
|
|
|
DumpMemToFile(file_path, NOT_NULL(addr), trans_flag, int_shapes, type); |
|
|
|
} |
|
|
|
@@ -249,7 +249,7 @@ void E2eDumpUtil::DumpSingleAnfnode(const AnfNodePtr &anf_node, const size_t out |
|
|
|
|
|
|
|
std::string file_path = dump_path + '/' + dump_name + '_' + "output_0"; |
|
|
|
if (IsDeviceTargetGPU()) { |
|
|
|
DumpGPUMemToFile(file_path, node_name, NOT_NULL(addr), trans_flag, int_shapes, type, 0, debugger); |
|
|
|
DumpGPUMemToFile(file_path, node_name + "_output", NOT_NULL(addr), trans_flag, int_shapes, type, 0, debugger); |
|
|
|
} else { |
|
|
|
DumpMemToFile(file_path, NOT_NULL(addr), trans_flag, int_shapes, type); |
|
|
|
} |
|
|
|
|