Add support for CUDA9 half semantics

Αυτή η υποβολή περιλαμβάνεται σε:
Sylvain Jeaugey committed 2017-06-14 11:20:24 -07:00
1 γονέας ccfc4567dc
υποβολή 29a1a916dc
3 files changed
+28 -24

No files matched your search

+1 -1
Προβολή Αρχείου
@@ -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)