Commit eb01ef26 authored by umangyadav's avatar umangyadav
Browse files

formatting

parent f02c2856
......@@ -144,8 +144,8 @@ shape miopen_convolution::find(context& ctx, const shape& output_shape, std::vec
if(status != miopenStatusSuccess)
MIGRAPHX_THROW("MIOpen Convolution: get solution failed");
solution_id = solutions.front().solution_id;
algo = solutions.front().algorithm;
solution_id = solutions.front().solution_id;
algo = solutions.front().algorithm;
workspace_size = solutions.front().workspace_size;
return shape{shape::int8_type, {workspace_size}};
......
......@@ -144,8 +144,8 @@ shape miopen_deconvolution::find(context& ctx, const shape& output_shape, std::v
if(status != miopenStatusSuccess)
MIGRAPHX_THROW("MIOpen Deconvolution: get solution failed");
solution_id = solutions.front().solution_id;
algo = solutions.front().algorithm;
solution_id = solutions.front().solution_id;
algo = solutions.front().algorithm;
workspace_size = solutions.front().workspace_size;
return shape{shape::int8_type, {workspace_size}};
......
......@@ -220,8 +220,8 @@ MIGRAPHX_PRED_MATCHER(fusable_conv, instruction_ref ins)
auto op = conv.op;
// Dont fuse winograd for non-3x3s since there is no fused windograd for those configs
if(conv.algo == miopenConvolutionAlgoWinograd and wei.lens()[2] != 3 and
wei.lens()[3] != 3 and contains({{1, 1}}, op.stride))
if(conv.algo == miopenConvolutionAlgoWinograd and wei.lens()[2] != 3 and wei.lens()[3] != 3 and
contains({{1, 1}}, op.stride))
return false;
return contains({{0, 0, 0, 0}, {1, 1, 1, 1}, {2, 2, 2, 2}}, op.padding) and
contains({{0, 0}, {1, 1}}, op.stride) and contains({{1, 1}}, op.dilation);
......
......@@ -139,7 +139,7 @@ shape miopen_quant_convolution::find(context& ctx,
solution_id = solutions.front().solution_id;
algo = solutions.front().algorithm;
algo = solutions.front().algorithm;
workspace_size = solutions.front().workspace_size;
return shape{shape::int8_type, {workspace_size}};
......
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