indent

843e83a2 · fenglei.tian · ee463b66 · 843e83a2
Commit 843e83a2 authored Mar 06, 2018 by fenglei.tian
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

gpu_external_function.cpp src/ngraph/runtime/gpu/gpu_external_function.cpp +2 -0

No files found.
--- a/src/ngraph/runtime/gpu/gpu_external_function.cpp
+++ b/src/ngraph/runtime/gpu/gpu_external_function.cpp
@@ -499,9 +499,11 @@ using namespace std;
                        node->get_outputs()[0].get_tensor_view();
                    writer << "if(" << tv->get_tensor().get_name() << " == NULL)\n";
                    writer << "{\n";
+                    writer.indent++;
                    writer << "runtime::gpu::cuda_memcpyHtD(" << tv->get_tensor().get_name() << ", "
                           << tv->get_tensor().get_name() << "_cpu, " << tv->get_tensor().size()
                           << ");\n";
+                    writer.indent--;
                    writer << "}\n";
                }
            }