billishyahao 3ac98edcb1 [Feature] add model aware kv ops helper (#16020) ...

Signed-off-by: billishyahao <bill.he@amd.com>

2025-04-15 23:00:43 -07:00

..

device_communicators

[TPU] optimize the all-reduce performance (#15903)

2025-04-03 00:25:14 +00:00

kv_transfer

[Feature] add model aware kv ops helper (#16020)

2025-04-15 23:00:43 -07:00

__init__.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

communication_op.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

parallel_state.py

[MISC] Make GroupCoordinator compatible with out-of-tree devices (#16464)

2025-04-12 09:20:25 +00:00

utils.py

[Misc] Merge the logs of pp layers partitions (#16225)

2025-04-08 00:18:15 -07:00