Skip to content

Commit 5097cfb

Browse files
ziyizhang-1Amanieu
authored andcommitted
initial commit to enable amx
AMX Intrinsics: amx-tile: - _tile_loadconfig - _tile_storeconfig - _tile_loadd - _tile_release - _tile_stored - _tile_stream_loadd - _tile_zero amx-int8: - _tile_dpbssd - _tile_dpbsud - _tile_dpbusd - _tile_dpbuud amx-bf16: - _tile_dpbf16ps amx-fp16 - _tile_dpfp16ps amx-complex - _tile_cmmimfp16ps - _tile_cmmrlfp16ps
1 parent fb90dfa commit 5097cfb

File tree

7 files changed

+618
-18
lines changed

7 files changed

+618
-18
lines changed

crates/core_arch/Cargo.toml

+3
Original file line numberDiff line numberDiff line change
@@ -23,3 +23,6 @@ maintenance = { status = "experimental" }
2323
[dev-dependencies]
2424
stdarch-test = { version = "0.*", path = "../stdarch-test" }
2525
std_detect = { version = "0.*", path = "../std_detect" }
26+
27+
[target.'cfg(all(target_arch = "x86_64", target_os = "linux"))'.dev-dependencies]
28+
syscalls = { version = "0.6.18", default-features = false }

crates/core_arch/missing-x86.md

-15
Original file line numberDiff line numberDiff line change
@@ -2,23 +2,19 @@
22
<details><summary>["AMX-BF16"]</summary><p>
33

44
* [ ] [`__tile_dpbf16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_dpbf16ps)
5-
* [ ] [`_tile_dpbf16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_dpbf16ps)
65
</p></details>
76

87

98
<details><summary>["AMX-COMPLEX"]</summary><p>
109

1110
* [ ] [`__tile_cmmimfp16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_cmmimfp16ps)
1211
* [ ] [`__tile_cmmrlfp16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_cmmrlfp16ps)
13-
* [ ] [`_tile_cmmimfp16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_cmmimfp16ps)
14-
* [ ] [`_tile_cmmrlfp16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_cmmrlfp16ps)
1512
</p></details>
1613

1714

1815
<details><summary>["AMX-FP16"]</summary><p>
1916

2017
* [ ] [`__tile_dpfp16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_dpfp16ps)
21-
* [ ] [`_tile_dpfp16ps`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_dpfp16ps)
2218
</p></details>
2319

2420

@@ -28,10 +24,6 @@
2824
* [ ] [`__tile_dpbsud`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_dpbsud)
2925
* [ ] [`__tile_dpbusd`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_dpbusd)
3026
* [ ] [`__tile_dpbuud`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_dpbuud)
31-
* [ ] [`_tile_dpbssd`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_dpbssd)
32-
* [ ] [`_tile_dpbsud`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_dpbsud)
33-
* [ ] [`_tile_dpbusd`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_dpbusd)
34-
* [ ] [`_tile_dpbuud`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_dpbuud)
3527
</p></details>
3628

3729

@@ -41,13 +33,6 @@
4133
* [ ] [`__tile_stored`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_stored)
4234
* [ ] [`__tile_stream_loadd`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_stream_loadd)
4335
* [ ] [`__tile_zero`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=__tile_zero)
44-
* [ ] [`_tile_loadconfig`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_loadconfig)
45-
* [ ] [`_tile_loadd`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_loadd)
46-
* [ ] [`_tile_release`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_release)
47-
* [ ] [`_tile_storeconfig`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_storeconfig)
48-
* [ ] [`_tile_stored`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_stored)
49-
* [ ] [`_tile_stream_loadd`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_stream_loadd)
50-
* [ ] [`_tile_zero`](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_tile_zero)
5136
</p></details>
5237

5338

crates/core_arch/src/lib.rs

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
generic_arg_infer,
3636
asm_experimental_arch,
3737
sha512_sm_x86,
38+
x86_amx_intrinsics,
3839
f16
3940
)]
4041
#![cfg_attr(test, feature(test, abi_vectorcall, stdarch_internal))]

0 commit comments

Comments
 (0)