Skip to content

Commit 55e87c3

Browse files
committed
ggml : fix OpenCL broadcast requirement for ggml_mul (close #4453)
1 parent 873637a commit 55e87c3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml.c

+2-2
Original file line numberDiff line numberDiff line change
@@ -7759,10 +7759,10 @@ static void ggml_compute_forward_mul_f32(
77597759
const int ith = params->ith;
77607760
const int nth = params->nth;
77617761

7762-
// TODO: OpenCL kernel support broadcast
77637762
#ifdef GGML_USE_CLBLAST
77647763
if (src1->backend == GGML_BACKEND_GPU) {
7765-
GGML_ASSERT(ggml_are_same_shape(src0, src1));
7764+
// TODO: OpenCL kernel support full broadcast
7765+
GGML_ASSERT(ggml_can_repeat_rows(src1, src0));
77667766
if (ith == 0) {
77677767
ggml_cl_mul(src0, src1, dst);
77687768
}

0 commit comments

Comments
 (0)