Commit 9b3cb662 authored by zhouxiang's avatar zhouxiang
Browse files

修改前台打印策略

parent 9d75245e
...@@ -51,7 +51,7 @@ cublasMMWrapper::cublasMMWrapper(cublasHandle_t cublas_handle, ...@@ -51,7 +51,7 @@ cublasMMWrapper::cublasMMWrapper(cublasHandle_t cublas_handle,
if (env_n_value_str != nullptr) { if (env_n_value_str != nullptr) {
m_ihgemm_switch_n = std::stoi(env_n_value_str); m_ihgemm_switch_n = std::stoi(env_n_value_str);
} }
if(m_ihgemm_switch == 0) printf("hgemm_switch=%d, hgemm_switch_n_limit=%d\n", m_ihgemm_switch, m_ihgemm_switch_n); if(m_ihgemm_switch != 0) printf("hgemm_switch=%d, hgemm_switch_n_limit=%d\n", m_ihgemm_switch, m_ihgemm_switch_n);
} }
#ifdef SPARSITY_ENABLED #ifdef SPARSITY_ENABLED
......
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