Commit 8276bc83 authored by Khalique's avatar Khalique
Browse files

formatting

parent e17fb984
...@@ -55,7 +55,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op) ...@@ -55,7 +55,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op)
{ {
auto c = make_obj<convolution_descriptor>(&miopenCreateConvolutionDescriptor); auto c = make_obj<convolution_descriptor>(&miopenCreateConvolutionDescriptor);
miopenConvolutionMode_t c_mode = miopenConvolution; miopenConvolutionMode_t c_mode = miopenConvolution;
if (op.group > 1) if(op.group > 1)
c_mode = miopenGroupConv; c_mode = miopenGroupConv;
miopenInitConvolutionDescriptor(c.get(), miopenInitConvolutionDescriptor(c.get(),
c_mode, c_mode,
...@@ -65,7 +65,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op) ...@@ -65,7 +65,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op)
op.stride[1], op.stride[1],
op.dilation[0], op.dilation[0],
op.dilation[1]); op.dilation[1]);
if (op.group > 1) if(op.group > 1)
miopenSetConvolutionGroupCount(c.get(), op.group); miopenSetConvolutionGroupCount(c.get(), op.group);
return c; return c;
} }
......
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