Open
Description
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 themImplement the associated intrinsics in stdarchStabilization 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 themovrs
target feature #138003 - Add runtime detection for new AMX variants and MOVRS stdarch#1763
Unresolved Questions
Metadata
Metadata
Assignees
Labels
Type
Projects
Milestone
Relationships
Development
No branches or pull requests
Activity
x86_amx_intrinsics
feature flag #126639AMX
target-features rust-lang/stdarch#1587rustbot commentedon Jun 24, 2024
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.
Rollup merge of rust-lang#126639 - sayantn:amx, r=Amanieu
Rollup merge of rust-lang#126639 - sayantn:amx, r=Amanieu
Rollup merge of rust-lang#126639 - sayantn:amx, r=Amanieu
Rollup merge of rust-lang#126639 - sayantn:amx, r=Amanieu
Rollup merge of rust-lang#126639 - sayantn:amx, r=Amanieu
Unrolled build for rust-lang#126639
workingjubilee commentedon Oct 23, 2024
We will need to understand the ABI impacts of AMX, if any.
sayantn commentedon Oct 27, 2024
Afaik, AMX doesn't change any ABI. Although we would need to change some things in inline asm to support
tmm
registers (it is currently clobber-only, when we introduce the__tile1024i
type in stdarch, we can use that in inline asm, but sadly it would require compiler support to add the__tile1024i
type)amx
target features and themovrs
target feature #138003Rollup merge of rust-lang#138003 - sayantn:new-amx, r=Amanieu
Unrolled build for rust-lang#138003