Skip to content

[Metal][Performance] Add radix select kernel for partition and argpartition - #4559

Closed
mateuuszzzzz wants to merge 4 commits into
ml-explore:mainfrom
mateuuszzzzz:add-metal-radix-partition
Closed

[Metal][Performance] Add radix select kernel for partition and argpartition#4559
mateuuszzzzz wants to merge 4 commits into
ml-explore:mainfrom
mateuuszzzzz:add-metal-radix-partition

Relax radix dispatch for contiguous inputs with stride-1 partition axes

a7edc13
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar