Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[compiler] Support RMSNorm Op and Fuse #13964

Open
seockho-kim opened this issue Sep 9, 2024 · 3 comments
Open

[compiler] Support RMSNorm Op and Fuse #13964

seockho-kim opened this issue Sep 9, 2024 · 3 comments
Assignees
Labels
DRAFT A draft issue or PR for sharing one's current working status and discussion.

Comments

@seockho-kim
Copy link
Contributor

seockho-kim commented Sep 9, 2024

What

To support LLM,

  • Introduce RmsNorm Op and extend circle format.
  • Support RmsNorm Fuse
@seockho-kim seockho-kim self-assigned this Sep 9, 2024
@seockho-kim seockho-kim changed the title Implement RMSNorm fuse tool Support RMSNorm Op and Fuse Sep 9, 2024
@seockho-kim
Copy link
Contributor Author

seockho-kim commented Sep 9, 2024

RMS Norm

image

https://github.com/bzhangGo/rmsnorm

@seockho-kim
Copy link
Contributor Author

Pattern of RMS Norm (need to fuse)

image

@seockho-kim seockho-kim changed the title Support RMSNorm Op and Fuse [compiler] Support RMSNorm Op and Fuse Sep 9, 2024
@seockho-kim seockho-kim added the DRAFT A draft issue or PR for sharing one's current working status and discussion. label Sep 9, 2024
@seockho-kim
Copy link
Contributor Author

seockho-kim commented Sep 10, 2024

To Do in draft

  • Add RmsNorm op to circle schema
  • Add RmsNorm to luci modules including lang, import, export, interpreter, service, logex, partition
  • Implement RmsNorm Fuse
  • support rmsnorm in circle2circle
  • prepare test recipe
  • Add rmsnorm to circlechef
  • rmsnorm in circledump

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
DRAFT A draft issue or PR for sharing one's current working status and discussion.
Projects
Status: In Progress
Development

No branches or pull requests

1 participant