Skip to content

[FEA] will cuteDSL support INT8 GEMM and FP8 GEMM on Ada/Ampere platform #3081

@houqi

Description

@houqi

Which component requires the feature?

CuTe DSL

Feature Request

I wish I could use CUTLASS CuteDSL to support INT8 GEMM and FP8 GEMM now. it seems that for now only FP16/BF16 mma is supported: https://github.com/NVIDIA/cutlass/blob/main/python/CuTeDSL/cutlass/cute/nvgpu/warp/mma.py#L58

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions