Skip to content

Custom topk-logprobs kernel + Remove a redundant Device to Host logits copy.#1

Open
pramodith wants to merge 9 commits into
mainfrom
pramodith/optimize_topk
Open

Custom topk-logprobs kernel + Remove a redundant Device to Host logits copy.#1
pramodith wants to merge 9 commits into
mainfrom
pramodith/optimize_topk

Merge remote-tracking branch 'origin/main' into pramodith/optimize_topk

46b4210
Select commit
Loading
Failed to load commit list.