Add bfloat16 support in RCCL

Preprocessor symbol RCCL_BFLOAT16 is used as feature indicator
Este commit está contenido en:
Wenkai Du
2019-11-15 10:39:48 -08:00
padre 58a6e535f6
commit 5e109ed400
Se han modificado 10 ficheros con 370 adiciones y 9 borrados
+1
Ver fichero
@@ -9,6 +9,7 @@
#define NCCL_DEVICE_H_
#include "nccl.h"
#include "rccl_bfloat16.h"
#include <stdint.h>
// Convert volatile access to atomic