Dont generate output

00106f54 · Paul · 94461807 · 00106f54
Commit 00106f54 authored Oct 28, 2018 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/convolution.cpp src/targets/gpu/convolution.cpp +1 -1

No files found.
--- a/src/targets/gpu/convolution.cpp
+++ b/src/targets/gpu/convolution.cpp
@@ -57,7 +57,7 @@ shape miopen_convolution::compile(context& ctx,

    auto x         = to_gpu(generate_argument(inputs[0]->get_shape()));
    auto w         = to_gpu(generate_argument(inputs[1]->get_shape()));
-    auto y         = to_gpu(generate_argument(output_shape));
+    auto y         = allocate_gpu(output_shape);
    auto workspace = allocate_gpu(workspace_shape);

    int algo_count = 1;