This commit is contained in:
Yangqing Jia
2016-01-12 15:59:02 -08:00
parent d08880e61a
commit a05782f025

View File

@ -41,7 +41,6 @@ class CudaRTCFunction {
// Compile the program.
// TODO(Yangqing): how to find the current gpu architecture instead of hard
// coding it?
CAFFE_LOG_ERROR << "NVRTC version: " << major << minor;
const char *nvrtc_opts[] = {"--gpu-architecture=compute_35",
"--use_fast_math"};
nvrtcResult compile_result = nvrtcCompileProgram(