Skip to content

[XPU] Llama XPU's swiglu uses phi's swiglu#9414

Merged
wawltor merged 1 commit into
PaddlePaddle:developfrom
dynamicheart:xpu_llama_swiglu
Nov 13, 2024
Merged

[XPU] Llama XPU's swiglu uses phi's swiglu#9414
wawltor merged 1 commit into
PaddlePaddle:developfrom
dynamicheart:xpu_llama_swiglu

Conversation

@dynamicheart
Copy link
Copy Markdown
Contributor

@dynamicheart dynamicheart commented Nov 12, 2024

PR types

Function optimization

PR changes

Models

Description

Llama swiglu XPU调用逻辑使用Paddle phi原生的swiglu算子

Llama2 13B tp2 20steps回归测试

调用paddle_xpu_nn.xpu_swiglu的Loss
image

调用paddle.incubate.nn.functional.swiglu的Loss
image

@paddle-bot
Copy link
Copy Markdown

paddle-bot Bot commented Nov 12, 2024

Thanks for your contribution!

@paddle-bot paddle-bot Bot added the XPU label Nov 12, 2024
@dynamicheart dynamicheart changed the title [XPU] Llama XPU's swiglu uses Paddle's native swiglu [XPU] Llama XPU's swiglu uses phi's swiglu Nov 12, 2024
Copy link
Copy Markdown
Contributor

@wawltor wawltor left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@wawltor wawltor merged commit 2838e80 into PaddlePaddle:develop Nov 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants