提交 41e7e36a authored 作者: Iban Harlouchet's avatar Iban Harlouchet 提交者: Iban Harlouchet

flake8 for theano/misc/pycuda_utils.py

上级 3c346fa5
import numpy
import pycuda.gpuarray import pycuda.gpuarray
from theano.sandbox import cuda from theano.sandbox import cuda
if cuda.cuda_available == False: if cuda.cuda_available is False:
raise ImportError('Optional theano package cuda disabled') raise ImportError('Optional theano package cuda disabled')
...@@ -29,7 +28,7 @@ def to_gpuarray(x, copyif=False): ...@@ -29,7 +28,7 @@ def to_gpuarray(x, copyif=False):
# Check if it is c contiguous # Check if it is c contiguous
size = 1 size = 1
c_contiguous = True c_contiguous = True
for i in range(x.ndim-1, -1, -1): for i in range(x.ndim - 1, -1, -1):
if x.shape[i] == 1: if x.shape[i] == 1:
continue continue
if x._strides[i] != size: if x._strides[i] != size:
...@@ -59,7 +58,7 @@ def to_cudandarray(x): ...@@ -59,7 +58,7 @@ def to_cudandarray(x):
else: else:
strides = [1] strides = [1]
for i in x.shape[::-1][:-1]: for i in x.shape[::-1][:-1]:
strides.append(strides[-1]*i) strides.append(strides[-1] * i)
strides = tuple(strides[::-1]) strides = tuple(strides[::-1])
ptr = int(x.gpudata) # in pycuda trunk, y.ptr also works, which is a little cleaner ptr = int(x.gpudata) # in pycuda trunk, y.ptr also works, which is a little cleaner
z = cuda.from_gpu_pointer(ptr, x.shape, strides, x) z = cuda.from_gpu_pointer(ptr, x.shape, strides, x)
......
...@@ -184,7 +184,6 @@ whitelist_flake8 = [ ...@@ -184,7 +184,6 @@ whitelist_flake8 = [
"scan_module/scan_opt.py", "scan_module/scan_opt.py",
"scan_module/tests/test_scan.py", "scan_module/tests/test_scan.py",
"scan_module/tests/test_scan_opt.py", "scan_module/tests/test_scan_opt.py",
"misc/pycuda_utils.py",
"misc/pycuda_example.py", "misc/pycuda_example.py",
"misc/ordered_set.py", "misc/ordered_set.py",
"misc/strutil.py", "misc/strutil.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论