Add support for CUDA9 half semantics
Esse commit está contido em:
1 pai
ccfc4567dc
commit
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)
|
||||
|
||||
Reference in new issue
Bloquear um usuário