Просмотр исходного кода

ggml : avoid conv 2d kernel round up

Georgi Gerganov 2 лет назад
Родитель
Сommit
c824d2e368
1 измененных файлов с 1 добавлено и 2 удалено
  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;
 
     // 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(nb10 == sizeof(float));