"megatron/model/vscode:/vscode.git/clone" did not exist on "48c2a144676949420387f4e93317d69d6798a320"
Commit 55d93373 authored by Paul's avatar Paul
Browse files

Add env var for tuning value

parent 687a3310
......@@ -50,6 +50,7 @@ using namespace migraphx::gpu::gen; // NOLINT
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_LOG_CK_GEMM);
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_CK_TUNING);
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_CK_TUNING_VALUE);
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_CK_DEBUG);
// NOLINTNEXTLINE
......@@ -163,7 +164,7 @@ static std::size_t get_tuning_for(const std::vector<shape>& inputs)
if(it == tuning.end())
{
std::cout << "*********** Warning: CK tuning missing for config!" << std::endl;
return 4;
return value_of(MIGRAPHX_CK_TUNING_VALUE{}, 4);
}
return it->second;
}
......
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