diff options
| author | gchanan <gregchanan@gmail.com> | 2016-12-29 22:23:45 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2016-12-29 22:23:45 +0300 |
| commit | 1ac06689dba1a4a672ed1fb3c3117000a46d7af5 (patch) | |
| tree | 5edb7b33c8ef506cbe488fe5d83de6349e42aaeb /CMakeLists.txt | |
| parent | 6b763fd55f9919ec2f1ccf58c962213e6fb755ea (diff) | |
Add THHalfTensor support to cutorch (#655)
* Add THHalfTensor support to cutorch.
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 96224c4..9d1d0a0 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -11,7 +11,8 @@ IF (NOT WIN32) SET(CMAKE_C_FLAGS "-std=c99 -Werror=implicit-function-declaration ${CMAKE_C_FLAGS}") ENDIF (NOT WIN32) IF(CUDA_HAS_FP16 OR NOT ${CUDA_VERSION} LESS 7.5) - SET(CMAKE_C_FLAGS "-DCUDA_HAS_FP16=1 ${CMAKE_C_FLAGS}") + ADD_DEFINITIONS(-DTH_GENERIC_USE_HALF=1) + ADD_DEFINITIONS(-DCUDA_HAS_FP16=1) ENDIF() INCLUDE_DIRECTORIES(${CUDA_INCLUDE_DIRS}) |