Skip to content

Commit 04abf71

Browse files
committed
revert rms_norm logic
1 parent 220db37 commit 04abf71

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
@@ -2492,7 +2492,7 @@ static bool ggml_opencl_can_fuse(const struct ggml_cgraph * cgraph, int node_idx
24922492

24932493
// if rms_norm is the B operand, then we don't handle broadcast
24942494
if (rms_norm == mul->src[1] &&
2495-
!ggml_are_same_shape(mul->src[0], mul->src[1])) {
2495+
!ggml_are_same_shape(mul->src[0], rms_norm->src[1])) {
24962496
return false;
24972497
}
24982498

0 commit comments

Comments
 (0)