parent: 85eafd | patch | commitdiff
AlexeyAB
2018-02-25 cad4d1618fee74471d335314cb77070fee951a42
refs
author AlexeyAB <alexeyab84@gmail.com>
Sunday, February 25, 2018 13:29 +0000
committer AlexeyAB <alexeyab84@gmail.com>
Sunday, February 25, 2018 13:29 +0000
commitcad4d1618fee74471d335314cb77070fee951a42
tree 05a40bb5c7f5ad92bda6e11d2796691d2c37613d tree | zip | gz | tar | bzip2 | xz
parent 85eafd3d590c449b35627af3e0f2a007c5a27d15 view | diff 
Added support for Tensor Cores CC >= 7.0 (V100). For FP16/32 (mixed precision) define CUDNN_HALF should be used.
8 files modified
141 ■■■■■ changed files
build/darknet/darknet.vcxproj 2 ●●●●● diff | view | raw | blame | history
src/convolutional_kernels.cu 122 ●●●●● diff | view | raw | blame | history
src/convolutional_layer.c 8 ●●●●● diff | view | raw | blame | history
src/convolutional_layer.h 1 ●●●●● diff | view | raw | blame | history
src/layer.c 2 ●●●●● diff | view | raw | blame | history
src/layer.h 1 ●●●●● diff | view | raw | blame | history
src/network.c 3 ●●●●● diff | view | raw | blame | history
src/network_kernels.cu 2 ●●●●● diff | view | raw | blame | history