Skip to content

GPU Peak performance calculation requires FP8 and GH200 support #353

@le1nux

Description

@le1nux

Currently, the GPU peak performance calculation is rather limited supporting only BF16 and FP16 precision, as well as, A100 and H100 GPUs only.

def _get_theoretical_gpu_peak_performance(wrapped_model: FSDPX, world_size: int) -> Optional[Number]:

We should extend this to support FP8 and GH200.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions