Skip to content

Add the remaining AMX intrinsics#2091

Draft
sayantn wants to merge 2 commits intorust-lang:mainfrom
sayantn:amx-more
Draft

Add the remaining AMX intrinsics#2091
sayantn wants to merge 2 commits intorust-lang:mainfrom
sayantn:amx-more

Conversation

@sayantn
Copy link
Copy Markdown
Contributor

@sayantn sayantn commented Apr 21, 2026

As rust-lang/rust#155264 is now merged, we can finally implement the __tile1024i AMX intrinsics.

Currently draft because I am still unsure of what should be the API of __tile1024i -- the current design is very cumbersome to use. To load a tile from memory, the code looks like

let mut tile = MaybeUninit::uninit();

let tile_ptr = tile.as_mut_ptr();

(*tile_ptr).rows = rows;
(*tile_ptr).cols = cols;
__tile_loadd(tile_ptr, mem_addr, stride);

let tile = tile.assume_init();

So it might be useful to provide at least 1 function that just sets the rows and cols fields and returns the MaybeUninit.

cc @Amanieu @folkertdev

@sayantn sayantn force-pushed the amx-more branch 2 times, most recently from 5ee6ebc to 699f1b2 Compare April 22, 2026 00:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant