Check error code for find

7e9699ce · Paul · 82af58fe · 7e9699ce
Commit 7e9699ce authored Dec 12, 2018 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

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

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

    int algo_count = 1;
    miopenConvAlgoPerf_t perf;
-    miopenFindConvolutionForwardAlgorithm(ctx.get_stream().get_miopen(),
+    auto status = miopenFindConvolutionForwardAlgorithm(ctx.get_stream().get_miopen(),
                                          x_desc.get(),
                                          x.implicit(),
                                          w_desc.get(),
@@ -78,6 +78,8 @@ shape miopen_convolution::compile(context& ctx,
                                          workspace.implicit(),
                                          workspace_size,
                                          false);
+    if (status != miopenStatusSuccess)
+      MIGRAPHX_THROW("Find convolution failed");
    algo = perf.fwd_algo;
    return shape{shape::int8_type, {perf.memory}};
 }