Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHieu Hoang <hieuhoang@gmail.com>2017-12-01 02:49:22 +0300
committerHieu Hoang <hieuhoang@gmail.com>2017-12-01 02:49:22 +0300
commit126e65ecf3e727e553cd01da94dea3498fb6dac7 (patch)
tree9206cf3e7472847387ea8cbe244fa57901921fa0
parent2f1e3db1db5d9fba5fad392b85fbbe2774becfb0 (diff)
set sm_60 for half type
-rw-r--r--CMakeLists.txt2
1 files changed, 1 insertions, 1 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 765e2731..47858a87 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -38,7 +38,7 @@ endif(FPGA)
if(CUDA)
find_package(CUDA)
if(CUDA_FOUND)
- LIST(APPEND CUDA_NVCC_FLAGS --default-stream per-thread; -std=c++11; -O3; -arch=sm_37; -gencode=arch=compute_37,code=sm_37; -gencode=arch=compute_53,code=sm_53; -gencode=arch=compute_62,code=sm_62; -lineinfo; --use_fast_math; )
+ LIST(APPEND CUDA_NVCC_FLAGS --default-stream per-thread; -std=c++11; -O3; -arch=sm_60; -lineinfo; --use_fast_math; )
# add_definitions(-DCUDA_API_PER_THREAD_DEFAULT_STREAM)
add_definitions(-DCUDA)
SET(CUDA_PROPAGATE_HOST_FLAGS OFF)