Nick Hill 416f9cdede [Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433) ...

Signed-off-by: Nick Hill <nickhill123@gmail.com>

2026-05-05 02:43:25 +00:00

..

seqwise

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00

tokwise

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00

__init__.py

[Model] Reorganize pooling layers (#31973)

2026-01-09 11:02:14 +00:00

abstract.py

[Model] Reorganize pooling layers (#31973)

2026-01-09 11:02:14 +00:00

activations.py

[Model] Deprecate the score task (this will not affect users). (#37537)

2026-03-20 08:07:56 +00:00

common.py

[Perf] Remove redundant device copies for CPU-only pooling token IDs, 48.9% E2E throughput improvement (#38139)

2026-03-29 18:12:50 +00:00

special.py

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00