Skip to content

Commit

Permalink
fix fuse linear (#6567)
Browse files Browse the repository at this point in the history
  • Loading branch information
DesmonDay authored Aug 1, 2023
1 parent 018755f commit 6e77082
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion llm/ernie-3.5-se/modeling.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ def parallel_matmul(
if bias is not None:
logits += bias
else:
if fuse_linear:
if not fuse_linear:
logits = F.linear(input_parallel, y, bias)
else:
logits = paddle.incubate.nn.functional.fused_linear(input_parallel, y, bias) # hack for 逐位对齐
Expand Down

0 comments on commit 6e77082

Please sign in to comment.