Commit 41f0ccf0 authored by fenglei.tian's avatar fenglei.tian

fix bug for cumemdopyDtD usage in gpu_external_function.cpp

parent 7042165a
...@@ -571,7 +571,7 @@ using namespace std; ...@@ -571,7 +571,7 @@ using namespace std;
writer << "runtime::gpu::cuda_memcpyDtD(reinterpret_cast<" writer << "runtime::gpu::cuda_memcpyDtD(reinterpret_cast<"
<< et.c_type_string() << "*>(outputs[" << output_index << "]), " << et.c_type_string() << "*>(outputs[" << output_index << "]), "
<< m_variable_name_map[ptv->get_tensor().get_name()] << ", " << m_variable_name_map[ptv->get_tensor().get_name()] << ", "
<< ptv->get_tensor().size() << ",1);\n"; << ptv->get_tensor().size() << ");\n";
break; break;
} }
} }
...@@ -721,7 +721,7 @@ void runtime::gpu::GPU_ExternalFunction::handle_output_alias( ...@@ -721,7 +721,7 @@ void runtime::gpu::GPU_ExternalFunction::handle_output_alias(
{ {
writer << "runtime::gpu::cuda_memcpyDtD(static_cast<void*>(outputs[" writer << "runtime::gpu::cuda_memcpyDtD(static_cast<void*>(outputs["
<< outputs[i] << "]), static_cast<void*>(outputs[" << outputs[0] << outputs[i] << "]), static_cast<void*>(outputs[" << outputs[0]
<< "]), " << otv->get_tensor().size() << ",1);\n"; << "]), " << otv->get_tensor().size() << ");\n";
} }
writer.indent--; writer.indent--;
writer << "}\n"; writer << "}\n";
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment