Commit 0723126a authored by Chris Sullivan's avatar Chris Sullivan Committed by Robert Kimball

Update gpu_emitter.cpp (#1628)

bug fix for nvgpu goe
parent 4782e060
......@@ -1381,7 +1381,7 @@ namespace ngraph
auto get_tuple_element = static_cast<const ngraph::op::GetOutputElement*>(node);
writer.block_begin();
writer << "runtime::gpu::cuda_memcpyDtH(" << out[0].get_name() << ", "
writer << "runtime::gpu::cuda_memcpyDtD(" << out[0].get_name() << ", "
<< args[get_tuple_element->get_n()].get_name() << ", "
<< out[0].get_size() * out[0].get_element_type().size() << ");\n";
writer.block_end();
......
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