blob: 01ef24b94edf840126822f55da93fd9a84b4fc73 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
CUDA_VERSION = "%{cuda_version}"
CUDNN_VERSION = "%{cudnn_version}"
PLATFORM = "%{platform}"
def cuda_sdk_version():
return CUDA_VERSION
def cudnn_sdk_version():
return CUDNN_VERSION
def readlink_command():
if PLATFORM == "Darwin":
return "greadlink"
else:
return "readlink"
|