Sfoglia il codice sorgente

cuda : fix LLAMA_CUDA_F16 build (#6197)

slaren 1 anno fa
parent
commit
03a8f8fafe
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      ggml-cuda.cu

+ 1 - 1
ggml-cuda.cu

@@ -9453,7 +9453,7 @@ static void ggml_cuda_op_dequantize_mul_mat_vec(
 
     // on some GPUs it is faster to convert src1 to half and to use half precision intrinsics
 #ifdef GGML_CUDA_F16
-    cuda_pool_alloc<half> src1_dfloat_a;
+    ggml_cuda_pool_alloc<half> src1_dfloat_a(ctx.pool());
     half * src1_dfloat = nullptr; // dfloat == half
 
     bool src1_convert_f16 =