Add bf16 support in rccl-tests

This commit is contained in:
Wenkai Du
2019-11-15 13:46:03 -08:00
bovenliggende 32399955af
commit 1cda2f52b6
4 gewijzigde bestanden met toevoegingen van 283 en 3 verwijderingen
+1 -2
Bestand weergeven
@@ -15,11 +15,10 @@ HIPCC = $(ROCM_HOME)/hip/bin/hipcc
CXX = $(HIPCC)
HIPCUFLAGS :=
HIPCUFLAGS := -std=c++14
HIPCUFLAGS += -I$(ROCM_HOME)/include
HIPCUFLAGS += -I$(ROCM_HOME)/include/rccl
HIPCUFLAGS += -I$(ROCM_HOME)/hip/include/hip
HIPCUFLAGS += -I$(ROCM_HOME)/hiprand/include
LDFLAGS := -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt
HIPLDFLAGS := $(CUSTOM_RCCL_LIB) -L$(ROCM_HOME)/lib -lhsa-runtime64 -lrt