Fix bug in ggml_cuda_op_scale_tensor (#1031)

Co-authored-by: Iwan Kawrakow <iwan.kawrakow@gmail.com>
This commit is contained in:
Kawrakow
2025-12-03 11:32:19 +01:00
committed by GitHub
parent cf20d0c756
commit 7fbe8d3ac2

View File

@@ -360,9 +360,7 @@ static void ggml_cuda_op_scale_tensor(ggml_backend_cuda_context & ctx, ggml_tens
GGML_ASSERT(src0->type == GGML_TYPE_F32);
GGML_ASSERT( dst->type == GGML_TYPE_F32);
float scale;
memcpy(&scale, dst->src[1]->data, sizeof(float));
GGML_ASSERT(dst->src[1]->type == GGML_TYPE_F32);
scale_f32_cuda_l(src0_d, dst_d, dst->src[1]->data, ggml_nelements(src0), stream);
}