This commit is contained in:
Felix Marty 2024-01-25 18:14:35 +00:00 committed by OlivierDehaene
parent 359dd46474
commit 059a2d9fa6

View File

@ -135,7 +135,7 @@ void gemm_half_q_half_cuda
if (!temp_dq) temp_dq = b->temp_dq; if (!temp_dq) temp_dq = b->temp_dq;
b->reconstruct(temp_dq); b->reconstruct(temp_dq);
//cublasSetMathMode(cublas_handle, CUBLAS_TENSOR_OP_MATH); //cublasSetMathMode(cublas_handle, CUBLAS_TENSOR_OP_MATH);
const half alpha = __float2half(1.0f); const half alpha = __float2half(1.0f);