Add support for CUDA9 half semantics

此提交包含在:
Sylvain Jeaugey committed 2017-06-14 11:20:24 -07:00
1 父節點 ccfc4567dc
當前提交 29a1a916dc
3 file 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)