提交 54f89c9a authored 作者: nouiz's avatar nouiz

Merge pull request #1305 from lamblin/fix_nvcc_arch

Fix insertion of -arch=... when compiling with nvcc
......@@ -159,9 +159,9 @@ class NVCC_compiler(object):
move_shared_float32_to_gpu=False,
enable_cuda=False)
n = theano.sandbox.cuda.use.device_number
p = theano.sandbox.cuda.device_properties(n)
flags.append('-arch=sm_' + str(p['major']) +
str(p['minor']))
p = theano.sandbox.cuda.device_properties(n)
flags.append('-arch=sm_' + str(p['major']) +
str(p['minor']))
return flags
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论