Răsfoiți Sursa

metal : avoid uint (#11019)

Georgi Gerganov 1 an în urmă
părinte
comite
e7da954ecc
1 a modificat fișierele cu 2 adăugiri și 2 ștergeri
  1. 2 2
      ggml/src/ggml-metal/ggml-metal.m

+ 2 - 2
ggml/src/ggml-metal/ggml-metal.m

@@ -2067,8 +2067,8 @@ static void ggml_metal_encode_node(
                 GGML_ASSERT(ne12 % ne02 == 0);
                 GGML_ASSERT(ne13 % ne03 == 0);
 
-                const uint r2 = ne12/ne02;
-                const uint r3 = ne13/ne03;
+                const uint32_t r2 = ne12/ne02;
+                const uint32_t r3 = ne13/ne03;
 
                 // find the break-even point where the matrix-matrix kernel becomes more efficient compared
                 // to the matrix-vector kernel