Skip to content

Would you consider adopting AMX (Advanced Matrix Extensions) to accelerate inner product computations? #607

@ruclz

Description

@ruclz

Hello, would you consider adopting AMX (Advanced Matrix Extensions) to accelerate inner product computations? It is estimated that for FP32 precision, AMX could offer a 10% performance improvement over AVX-512. If support for BF16 (Brain Floating-Point 16) data is considered, the performance boost could be 1.8 times greater.
If yes, I will prepare a pull request for this optimization. Thanks very much

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions