Skip to content

[CPU][PowerPC] Vectorize top-k and top-p sampling for PowerPC (and s390x) - #2

Open
Akashcodes732 wants to merge 1 commit into
micyan01:arm-neon-topk-toppfrom
Akashcodes732:feat/power-topk-topp
Open

Akashcodes732 wants to merge 1 commit into
micyan01:arm-neon-topk-toppfrom
Akashcodes732:feat/power-topk-topp

Conversation

@Akashcodes732

Copy link
Copy Markdown

PowerPC (VSX) and s390x (VXE) share identical vector intrinsic signatures. Opening separate PRs for each architecture would cause merge conflicts in csrc/cpu/topk_topp_sampler.cpp, so this PR combines the implementation for both.

Signed-off-by: Akash kaothalkar <akash.kaothalkar@ibm.com>
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