Add support for CUDA9 half semantics

This commit is contained in:
Sylvain Jeaugey committed 2017-06-14 11:20:24 -07:00
1 parent ccfc4567dc
commit 29a1a916dc
3 files changed
+28 -24

No files matched your search

+1 -1
View File
@@ -54,7 +54,7 @@ endif
NCCL_MAJOR := 1
NCCL_MINOR := 3
NCCL_PATCH := 4
NCCL_PATCH := 5
CXXFLAGS += -DNCCL_MAJOR=$(NCCL_MAJOR) -DNCCL_MINOR=$(NCCL_MINOR) -DNCCL_PATCH=$(NCCL_PATCH)
CUDA_VERSION ?= $(shell ls $(CUDA_LIB)/libcudart.so.* | head -1 | rev | cut -d "." -f -2 | rev)