Skip to content

GitLab

Unverified Commit f9c06bb0 authored by Shangyan Zhou's avatar Shangyan Zhou Committed by GitHub
Browse files

Use TMA to optimize combine forwarder. (#320) 



* Remove an outdated todo

* Increase the number of combine forward warps.

* forwarder use TMA.

* Small fix

* Code lint

---------
Co-authored-by: default avatarChenggang Zhao <chenggangz@deepseek.com>
parent e6012370
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 142 additions and 61 deletions
csrc/kernels/internode.cu
View file @ f9c06bb0
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment