Fix rope_norm_fast_cuda

This commit is contained in:
Iwan Kawrakow
2025-11-12 17:00:13 +02:00
parent 8a8de91a42
commit 2aee6a0d94

View File

@@ -895,7 +895,7 @@ void ggml_cuda_op_rope_fast(ggml_backend_cuda_context & ctx, ggml_tensor * dst)
} else {
//printf("Using norm\n");
rope_norm_fast_cuda(
(const float *)src0->data, (const float *)src1->data, (float *)dst->data, ne00, ne01, s01, s02, n_dims, nr, stream);
(const float *)src0->data, (const float *)src1->data, (float *)dst->data, ne00, ne01, ne02, s01, s02, n_dims, stream);
}
}