Explorar o código

ggml : avoid conv 2d kernel round up

Georgi Gerganov %!s(int64=2) %!d(string=hai) anos
pai
achega
c824d2e368
Modificáronse 1 ficheiros con 1 adicións e 2 borrados
  1. 1 2
      ggml.c

+ 1 - 2
ggml.c

@@ -13508,8 +13508,7 @@ static void ggml_compute_forward_conv_2d_sk_p0_f16_f32(
     const int nk1 = ne01;
     const int nk1 = ne01;
 
 
     // size of the convolution row - the kernel size unrolled across all channels
     // size of the convolution row - the kernel size unrolled across all channels
-    // round-up so it is more suitable for SIMD
-    const int ew0 = ggml_up32(nk0*nk1*ne02);
+    const int ew0 = nk0*nk1*ne02;
 
 
     GGML_ASSERT(nb00 == sizeof(ggml_fp16_t));
     GGML_ASSERT(nb00 == sizeof(ggml_fp16_t));
     GGML_ASSERT(nb10 == sizeof(float));
     GGML_ASSERT(nb10 == sizeof(float));