Check for single output

8734b6ad · Paul · bdcd2fc6 · 8734b6ad
Commit 8734b6ad authored Oct 09, 2018 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/targets/gpu/fuse_ops.cpp src/targets/gpu/fuse_ops.cpp +2 -2

No files found.
--- a/src/targets/gpu/fuse_ops.cpp
+++ b/src/targets/gpu/fuse_ops.cpp
@@ -252,8 +252,8 @@ struct miopen_conv_bias_relu
 template <class... Ms>
 auto conv_bias(Ms... ms)
 {
-    return match::name("gpu::add")(match::either_arg(0, 1)(match::arg(0)(bias_shape()).bind("bias"),
+    return match::name("gpu::add")(match::either_arg(0, 1)(match::arg(0)(bias_shape(match::output())).bind("bias"),
-                                                           fusable_conv().bind("conv")),
+                                                           fusable_conv(match::output()).bind("conv")), match::output(),
                                   ms...);
 }