Skip to content

[TorchAO] Float8 training support on XPU #2325

@liangan1

Description

@liangan1

🚀 The feature, motivation and pitch

The scaled_mm(pytorch/pytorch#165978) and inference(pytorch/ao#3332) is WIP, You can enable the realted training UTs with emulate path and comments your PR in pytorch/ao#2917 to help us to track the total status.
Reference: https://github.com/pytorch/ao/tree/main/torchao/float8

Alternatives

No response

Additional context

No response

Metadata

Metadata

Assignees

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions