[ggml] fix index for ne03 value in ggml_cl_mul_f32 (ggerganov#2088)

YellowRoseCx · Jul 3, 2023 · 14a2cc7 · 14a2cc7
1 parent 1cf14cc
commit 14a2cc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ggml-opencl.cpp b/ggml-opencl.cpp
@@ -1376,7 +1376,7 @@ static void ggml_cl_mul_f32(const ggml_tensor * src0, const ggml_tensor * src1,
     const int64_t ne00 = src0->ne[0];
     const int64_t ne01 = src0->ne[1];
     const int64_t ne02 = src0->ne[2];
-    const int64_t ne03 = src0->ne[2];
+    const int64_t ne03 = src0->ne[3];
     const int64_t ne0 = ne00 * ne01 * ne02 * ne03;
     const int64_t ne10 = src1->ne[0];
     const int64_t ne11 = src1->ne[1];