Wen-Heng (Jack) Chung baadda4bd8 Relax workgroup barrier implementation for MSCCL send/recv ops. (#997) ...

* Trim logic.

* Revert "Trim logic."

This reverts commit 8f2dba6c764108acf2bf5428366b9f41d4d206b9.

* Introduce MSCCL template parameters to send / recv.

* Address review feedbacks.

2023-12-08 17:46:53 -06:00

..

device

Relax workgroup barrier implementation for MSCCL send/recv ops. (#997)

2023-12-08 17:46:53 -06:00

all_gather.cc

Merge remote-tracking branch 'nccl/master' into HEAD

2023-02-04 01:44:43 +00:00

all_reduce.cc

Merge remote-tracking branch 'nccl/master' into HEAD

2023-02-04 01:44:43 +00:00

all_to_all.cc

Add MSCCL Support (#658)

2022-12-12 15:51:04 -08:00

all_to_allv.cc

Add MSCCL Support (#658)

2022-12-12 15:51:04 -08:00

broadcast.cc

Merge remote-tracking branch 'nccl/master' into HEAD

2023-02-04 01:44:43 +00:00

gather.cc

Add MSCCL Support (#658)

2022-12-12 15:51:04 -08:00

msccl.cc

Move MSCCL algorithm loading to initialization to workaround HIP graph conflict (#982)

2023-11-30 09:47:20 -08:00

reduce_scatter.cc

Merge remote-tracking branch 'nccl/master' into HEAD

2023-02-04 01:44:43 +00:00

reduce.cc

Merge remote-tracking branch 'nccl/master' into HEAD

2023-02-04 01:44:43 +00:00

scatter.cc

Add MSCCL Support (#658)

2022-12-12 15:51:04 -08:00

sendrecv.cc

Merge remote-tracking branch 'nccl/master' into HEAD

2023-02-04 01:44:43 +00:00