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
if(status != miopenStatusSuccess)
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,
......
......@@ -144,11 +144,11 @@ 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;
workspace_size = solutions.front().workspace_size;
const auto& best_solution = solutions.front();
solution_id = best_solution.solution_id;
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,
......
......@@ -137,12 +137,12 @@ shape miopen_quant_convolution::find(context& ctx,
if(status != miopenStatusSuccess)
MIGRAPHX_THROW("MIOpen Quant Convolution: get solution failed");
solution_id = solutions.front().solution_id;
const auto& best_solution = solutions.front();
algo = solutions.front().algorithm;
workspace_size = solutions.front().workspace_size;
solution_id = best_solution.solution_id;
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,
......
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