bug fix, hang for compiling (#2082)

ef0ec4c6 · Fenglei · Robert Kimball · 4a08feb1 · ef0ec4c6
Commit ef0ec4c6 authored Nov 21, 2018 by Fenglei Committed by Robert Kimball Nov 21, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 5 deletions

gpu_emitter.cpp src/ngraph/runtime/gpu/gpu_emitter.cpp +3 -5

No files found.
--- a/src/ngraph/runtime/gpu/gpu_emitter.cpp
+++ b/src/ngraph/runtime/gpu/gpu_emitter.cpp
@@ -1645,11 +1645,9 @@ tensorflow test failures*/
                                                        args[0].get_shape(),
                                                        sum->get_reduction_axes(),
                                                        CUDNNEmitter::ReductionMode::Reduce);
-
-                writer << "gpu::invoke_primitive(ctx, " << sum_index << ", ";
-                writer << "std::vector<void*>{" << args[0].get_name() << "}.data(), ";
-                writer << "std::vector<void*>{" << out[0].get_name() << "}.data()";
-                writer << ");\n";
+                writer << "void* input[] = {" << node_names(args) << "};\n";
+                writer << "void* output[] = {" << node_names(out) << "};\n";
+                writer << "gpu::invoke_primitive(ctx, " << sum_index << ", input, output);\n";
            }
        }
    }