Skip to content

GitLab

".github/git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "652c273f0ea9006856973b960e6feea85d138d3c"
Commit d1ab802c authored by Paul's avatar Paul
Browse files

Add env var

parent 19bf60bd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/targets/gpu/jit/ck_gemm.cpp
View file @ d1ab802c
...@@ -50,6 +50,7 @@ using namespace migraphx::gpu::gen; // NOLINT ...@@ -50,6 +50,7 @@ using namespace migraphx::gpu::gen; // NOLINT
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_LOG_CK_GEMM); MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_LOG_CK_GEMM);
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_CK_TUNING); MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_CK_TUNING);
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_CK_DEBUG);
// NOLINTNEXTLINE // NOLINTNEXTLINE
static const char* const ck_gemm_kernel = R"__migraphx__( static const char* const ck_gemm_kernel = R"__migraphx__(
...@@ -184,7 +185,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler> ...@@ -184,7 +185,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
options.kernel_name = v.get("kernel", "ck_gemm_kernel"); options.kernel_name = v.get("kernel", "ck_gemm_kernel");
options.virtual_inputs = inputs; options.virtual_inputs = inputs;
if(v.get("check", false)) if(v.get("check", false) or enabled(MIGRAPHX_CK_DEBUG{}))
options.params += " -DMIGRAPHX_CK_CHECK=1"; options.params += " -DMIGRAPHX_CK_CHECK=1";
auto src = interpolate_string(ck_gemm_kernel, auto src = interpolate_string(ck_gemm_kernel,
......
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