Commit fea5c55d authored by umangyadav's avatar umangyadav
Browse files

paul's comments

parent eb01ef26
...@@ -144,11 +144,12 @@ shape miopen_convolution::find(context& ctx, const shape& output_shape, std::vec ...@@ -144,11 +144,12 @@ shape miopen_convolution::find(context& ctx, const shape& output_shape, std::vec
if(status != miopenStatusSuccess) if(status != miopenStatusSuccess)
MIGRAPHX_THROW("MIOpen Convolution: get solution failed"); MIGRAPHX_THROW("MIOpen Convolution: get solution failed");
solution_id = solutions.front().solution_id;
algo = solutions.front().algorithm;
workspace_size = solutions.front().workspace_size;
return shape{shape::int8_type, {workspace_size}}; const auto& best_solution = solutions.front();
solution_id = best_solution.solution_id;
algo = best_solution.algorithm;
return shape{shape::int8_type, {best_solution.workspace_size}};
} }
void miopen_convolution::finalize(context& ctx, void miopen_convolution::finalize(context& ctx,
......
...@@ -144,11 +144,11 @@ shape miopen_deconvolution::find(context& ctx, const shape& output_shape, std::v ...@@ -144,11 +144,11 @@ shape miopen_deconvolution::find(context& ctx, const shape& output_shape, std::v
if(status != miopenStatusSuccess) if(status != miopenStatusSuccess)
MIGRAPHX_THROW("MIOpen Deconvolution: get solution failed"); MIGRAPHX_THROW("MIOpen Deconvolution: get solution failed");
solution_id = solutions.front().solution_id; const auto& best_solution = solutions.front();
algo = solutions.front().algorithm; solution_id = best_solution.solution_id;
workspace_size = solutions.front().workspace_size; algo = best_solution.algorithm;
return shape{shape::int8_type, {workspace_size}}; return shape{shape::int8_type, {best_solution.workspace_size}};
} }
void miopen_deconvolution::finalize(context& ctx, void miopen_deconvolution::finalize(context& ctx,
......
...@@ -137,12 +137,12 @@ shape miopen_quant_convolution::find(context& ctx, ...@@ -137,12 +137,12 @@ shape miopen_quant_convolution::find(context& ctx,
if(status != miopenStatusSuccess) if(status != miopenStatusSuccess)
MIGRAPHX_THROW("MIOpen Quant Convolution: get solution failed"); MIGRAPHX_THROW("MIOpen Quant Convolution: get solution failed");
solution_id = solutions.front().solution_id; const auto& best_solution = solutions.front();
algo = solutions.front().algorithm; solution_id = best_solution.solution_id;
workspace_size = solutions.front().workspace_size; algo = best_solution.algorithm;
return shape{shape::int8_type, {workspace_size}}; return shape{shape::int8_type, {best_solution.workspace_size}};
} }
void miopen_quant_convolution::finalize(context& ctx, void miopen_quant_convolution::finalize(context& ctx,
......
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