Tracking Issue for AMX target features and intrinsics

[sayantn]

This issue tracks the integration of Intel AMX instruction family to Rust.

The feature gate for the issue is #![feature(x86_amx_intrinsics)].

Steps

• Add the target-feature attributes and runtime detection for them
• Implement the associated intrinsics in stdarch
• Stabilization PR

Public API

The following target features and all their associated intrinsics and types (not complete, will be extended in future)

• amx-tile
• amx-int8
• amx-bf16
• amx-fp16
• amx-complex
• amx-avx512
• amx-fp8
• amx-movrs
• amx-tf32
• amx-transpose

Implementation History

• Add AMX target-features and x86_amx_intrinsics feature flag #126639
• Runtime detection for AMX target-features stdarch#1587
• initial commit to enable amx stdarch#1618
• Add the new amx target features and the movrs target feature #138003
• Add runtime detection for new AMX variants and MOVRS stdarch#1763
• cleanup: upstream dropped amx-transpose functionality #148338
• Add intrinsics for the new AMX target features stdarch#1846
• Add the remaining AMX intrinsics stdarch#2091
• cleanup: upstream dropped AMX-TF32 stdarch#2184
• cleanup: upstream dropped AMX-TF32 #158910

Unresolved Questions

• How should we handle matrix ABIs? #133144

[rustbot]

Error: Label B-unstable can only be set by Rust team members

Please file an issue on GitHub at triagebot if there's a problem with this bot, or reach out on #t-infra on Zulip.

[workingjubilee]

We will need to understand the ABI impacts of AMX, if any.