提交 4236fc0e authored 作者: Olivier Delalleau's avatar Olivier Delalleau

PEP8

上级 52b98b42
import atexit, logging, os, stat, sys, shutil import atexit, logging, os, shutil, stat, sys
from theano.compile import optdb from theano.compile import optdb
from theano.gof.cmodule import get_lib_extension from theano.gof.cmodule import get_lib_extension
from theano.configparser import config, AddConfigVar, StrParam from theano.configparser import config, AddConfigVar, StrParam
...@@ -122,8 +122,8 @@ if cuda_available: ...@@ -122,8 +122,8 @@ if cuda_available:
try: try:
open(libcuda_ndarray_so).close() open(libcuda_ndarray_so).close()
except IOError: except IOError:
if sys.platform=="win32": if sys.platform == "win32":
# The Python os module does not support symlinks on win32. # The Python `os` module does not support symlinks on win32.
shutil.copyfile(cuda_ndarray_so, libcuda_ndarray_so) shutil.copyfile(cuda_ndarray_so, libcuda_ndarray_so)
else: else:
os.symlink(cuda_ndarray_so, libcuda_ndarray_so) os.symlink(cuda_ndarray_so, libcuda_ndarray_so)
......
...@@ -164,12 +164,12 @@ def nvcc_module_compile_str( ...@@ -164,12 +164,12 @@ def nvcc_module_compile_str(
if config.nvcc.compiler_bindir: if config.nvcc.compiler_bindir:
cmd.extend(['--compiler-bindir', config.nvcc.compiler_bindir]) cmd.extend(['--compiler-bindir', config.nvcc.compiler_bindir])
if sys.platform=='win32': if sys.platform == 'win32':
# add flags for Microsoft compiler to create .pdb files # add flags for Microsoft compiler to create .pdb files
preargs2.append('/Zi') preargs2.append('/Zi')
cmd.extend(['-Xlinker', '/DEBUG']) cmd.extend(['-Xlinker', '/DEBUG'])
if sys.platform!='win32': if sys.platform != 'win32':
if local_bitwidth() == 64: if local_bitwidth() == 64:
cmd.append('-m64') cmd.append('-m64')
preargs2.append('-m64') preargs2.append('-m64')
...@@ -185,7 +185,7 @@ def nvcc_module_compile_str( ...@@ -185,7 +185,7 @@ def nvcc_module_compile_str(
if sys.platform != 'darwin': if sys.platform != 'darwin':
# the 64bit CUDA libs are in the same files as are named by the function above # the 64bit CUDA libs are in the same files as are named by the function above
rpaths.append(os.path.join(config.cuda.root,'lib64')) rpaths.append(os.path.join(config.cuda.root,'lib64'))
if sys.platform!="win32": if sys.platform != 'win32':
# the -rpath option is not understood by the Microsoft linker # the -rpath option is not understood by the Microsoft linker
for rpath in rpaths: for rpath in rpaths:
cmd.extend(['-Xlinker',','.join(['-rpath',rpath])]) cmd.extend(['-Xlinker',','.join(['-rpath',rpath])])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论