Skip to content

Commit ca1ded3

Browse files
committed
revert rms_norm logic
1 parent 3bad778 commit ca1ded3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-opencl/ggml-opencl.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2374,7 +2374,7 @@ static bool ggml_opencl_can_fuse(const struct ggml_cgraph * cgraph, int node_idx
23742374

23752375
// if rms_norm is the B operand, then we don't handle broadcast
23762376
if (rms_norm == mul->src[1] &&
2377-
!ggml_are_same_shape(mul->src[0], mul->src[1])) {
2377+
!ggml_are_same_shape(mul->src[0], rms_norm->src[1])) {
23782378
return false;
23792379
}
23802380

0 commit comments

Comments
 (0)