Skip to content

GitLab

Commit 96c856c5 authored by Paul's avatar Paul
Browse files

Formatting

parent 1e969c74
Hide whitespace changes
Inline Side-by-side
Showing with 14 additions and 14 deletions
src/targets/gpu/convolution.cpp
View file @ 96c856c5
...@@ -22,18 +22,18 @@ argument miopen_convolution::compute(context& ctx, ...@@ -22,18 +22,18 @@ argument miopen_convolution::compute(context& ctx,
float alpha = 1; float alpha = 1;
float beta = 0; float beta = 0;
auto status = miopenConvolutionForward(ctx.get_stream().get_miopen(), auto status = miopenConvolutionForward(ctx.get_stream().get_miopen(),
&alpha, &alpha,
x_desc.get(), x_desc.get(),
args[0].implicit(), args[0].implicit(),
w_desc.get(), w_desc.get(),
args[1].implicit(), args[1].implicit(),
cd.get(), cd.get(),
algo, algo,
&beta, &beta,
y_desc.get(), y_desc.get(),
args[3].implicit(), args[3].implicit(),
args[2].implicit(), args[2].implicit(),
args[2].get_shape().bytes()); args[2].get_shape().bytes());
if(status != miopenStatusSuccess) if(status != miopenStatusSuccess)
MIGRAPHX_THROW("Running convolution failed"); MIGRAPHX_THROW("Running convolution failed");
return args[3]; return args[3];
...@@ -93,8 +93,8 @@ void miopen_convolution::finalize(context& ctx, ...@@ -93,8 +93,8 @@ void miopen_convolution::finalize(context& ctx,
return; return;
// Check that workspace hasn't changed // Check that workspace hasn't changed
auto size = inputs.at(2).bytes(); auto size = inputs.at(2).bytes();
auto ws = compile(ctx, output_shape, std::move(inputs)); auto ws = compile(ctx, output_shape, std::move(inputs));
if (ws.bytes() > size) if(ws.bytes() > size)
MIGRAPHX_THROW("Workspace has changed during finalization."); MIGRAPHX_THROW("Workspace has changed during finalization.");
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment