opencl: fix local size for norm (#27339)

This commit is contained in:
lhez 2026-08-20 10:52:07 -07:00 committed by GitHub
parent 521a64cd01
commit 6b4fa88a6c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -12830,7 +12830,10 @@ static void ggml_cl_norm(ggml_backend_t backend, const ggml_tensor * src0, const
GGML_TENSOR_LOCALS(int, ne0, src0, ne);
GGML_TENSOR_LOCALS(cl_ulong, nb0, src0, nb);
const int nth = MIN(64, ne00);
int nth = 1;
while (nth < ne00 && nth < 64) {
nth *= 2;
}
cl_kernel kernel = backend_ctx->kernel_norm;