Skip to content

Fully fused SwiGLU? #39

@LeoZDong

Description

@LeoZDong

Description

Is there a sample for a fully fused SwiGLU layer? I see that in tile gym there is a fused_swiglu.py example, but that is partially fused using a cuTile matmul and silu_and_mul.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions