rearranged how the min and max functions are part of msccl (#1025)

* rearranged how the min and max functions are part of msccl

* added more coverage on in place graph tests
Цей коміт міститься в:
akolliasAMD
2023-12-21 08:58:33 -07:00
зафіксовано GitHub
джерело 7e1cbb440d
коміт f4858e14b2
4 змінених файлів з 8 додано та 8 видалено
+2 -2
Переглянути файл
@@ -185,7 +185,7 @@ static void HIPRT_CB mscclSetupProxyCallback(void *args) {
INFO(NCCL_NET,"mscclSetupProxyCallback: proxy args size: %ld\n", params->size());
for (auto &p : *params) {
mscclSetupProxyImpl(p.hostAlgo, p.comm);
}
}
}
ncclResult_t mscclSetupProxy(struct mscclAlgo* hostAlgo, ncclComm_t comm, hipStream_t stream) {
@@ -317,8 +317,8 @@ static ncclResult_t hostToDevRedOp(
#define MSCCL_KERNEL_ENTRY() \
MSCCL_KERNEL_ENTRY_DEVREDOP(Sum, false), \
MSCCL_KERNEL_ENTRY_DEVREDOP(Prod, false), \
MSCCL_KERNEL_ENTRY_DEVREDOP(Min, false), \
MSCCL_KERNEL_ENTRY_DEVREDOP(Max, false), \
MSCCL_KERNEL_ENTRY_DEVREDOP(Min, false), \
MSCCL_KERNEL_ENTRY_DEVREDOP(PreMulSum, false), \
MSCCL_KERNEL_ENTRY_DEVREDOP_NOFLOAT(SumPostDiv, false)