Unit Test: support ncclAvg

Este commit está contenido en:
Wenkai Du
2021-08-25 12:01:49 -07:00
padre 60ca7484c0
commit 1faff323b4
Se han modificado 8 ficheros con 80 adiciones y 36 borrados
+2 -2
Ver fichero
@@ -520,8 +520,8 @@ struct FuncAvg<rccl_bfloat16>: FuncSum<rccl_bfloat16> {
// Obviously, the only invalid combination is both true. An improvement would be
// make this parameterized as a build time setting and passed here through
// preprocessor definitions.
static constexpr bool IsPreOpIdentity = false;
static constexpr bool IsPostOpIdentity = true;
static constexpr bool IsPreOpIdentity = true;
static constexpr bool IsPostOpIdentity = false;
#if __CUDA_ARCH__ >= 800
__device__ FuncAvg(int n) {