Skip to content

[Feature]: AutoDeploy: use torch's built-in rms norm as reference #9526

@lucaslie

Description

@lucaslie

🚀 The feature, motivation and pitch

see #9504 (comment)

Consider using torch.nn.functional's rms norm instead of eager rms norm our torch_rmsnorm custom op as reference op for rms norm

Will require updates to the pattern matching infra as well and the use of rms norm throughout the code base

Alternatives

No response

Additional context

No response

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and checked the documentation and examples for answers to frequently asked questions.

Metadata

Metadata

Assignees

Labels

AutoDeploy<NV> AutoDeploy Backendfeature requestNew feature or request. This includes new model, dtype, functionality support

Type

No type

Projects

Status

Backlog

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions