"git@developer.sourcefind.cn:change/sglang.git" did not exist on "b91cb67e7ae55b57b81e6645b4773f15987f7f97"
Commit 1e623983 authored by Davis King's avatar Davis King
Browse files

Fixed cmake script so nvcc doesn't freak out.

parent 9e2f26fc
......@@ -412,7 +412,10 @@ if (NOT TARGET dlib)
if (CUDA_FOUND)
set(CUDA_HOST_COMPILATION_CPP ON)
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_30;-std=c++11")
# Note that we add __STRICT_ANSI__ to avoid freaking out nvcc with gcc specific
# magic in the standard C++ header files (since nvcc uses gcc headers on
# linux).
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_30;-std=c++11;-D__STRICT_ANSI__")
message(STATUS "Looking for cuDNN install...")
# Look for cudnn, we will look in the same place as other CUDA
......
......@@ -2,7 +2,7 @@
#include <stdlib.h>
#include <stdio.h>
//#include "cuda_dlib.h"
#include "cuda_dlib.h"
#define CHECK(call) \
{ \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment