Nick Hill 6c18a54648 [Perf] Avoid h2d copies from non-pinned CPU tensors (#54299) ...

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

2026-08-29 12:06:30 +08:00

..

__init__.py

[Misc] Aligning tokwise pooler heads for consistency (#43041)

2026-05-19 06:16:42 +00:00

heads.py

Added extra_repr() to pooler classes to improve debuggability (#44805)

2026-06-08 03:19:31 +00:00

methods.py

[Perf] Avoid h2d copies from non-pinned CPU tensors (#54299)

2026-08-29 12:06:30 +08:00

poolers.py

Added extra_repr() to pooler classes to improve debuggability (#44805)

2026-06-08 03:19:31 +00:00