Skip to content

[Feat][Kernel] Add TP Fused linear logp Triton - #208

Open
KJLdefeated wants to merge 4 commits into
mainfrom
feat/linear-logp-cuda-backward
Open

[Feat][Kernel] Add TP Fused linear logp Triton#208
KJLdefeated wants to merge 4 commits into
mainfrom
feat/linear-logp-cuda-backward

fix Reject TP vocab shards with local_vocab_size <= 0 in _validate_tp…

3551dcb
Select commit
Loading
Failed to load commit list.