[Feature] support shardformer #5916

Closed

Assignees

opened

on Jul 17, 2024

Replace the dtype to fp8 in communication is useful For bandwidth restricted scenario. Please support the fp8 communication in Shardformer(SP and TP)

Metadata

Assignees

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests