提交 3c346fa5 authored 作者: Iban Harlouchet's avatar Iban Harlouchet 提交者: Iban Harlouchet

flake8 for theano/misc/cudamat_utils.py

上级 f3028c43
...@@ -13,7 +13,7 @@ try: ...@@ -13,7 +13,7 @@ try:
cudamat_available = True cudamat_available = True
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as 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')
def cudandarray_to_cudamat(x, copyif=False): def cudandarray_to_cudamat(x, copyif=False):
...@@ -43,7 +43,7 @@ try: ...@@ -43,7 +43,7 @@ try:
# 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:
...@@ -73,11 +73,10 @@ try: ...@@ -73,11 +73,10 @@ try:
cm_mat.data_device = ctypes.cast(x.gpudata, ctypes.POINTER(ctypes.c_float)) cm_mat.data_device = ctypes.cast(x.gpudata, ctypes.POINTER(ctypes.c_float))
px = cudamat.CUDAMatrix(cm_mat) px = cudamat.CUDAMatrix(cm_mat)
px._base = x # x won't be __del__'ed as long as px is around. px._base = x # x won't be __del__'ed as long as px is around.
px.mat_on_host = False # let cudamat know that we don't have a numpy # let cudamat know that we don't have a numpy array attached.
# array attached. px.mat_on_host = False
return px return px
def cudamat_to_cudandarray(x): def cudamat_to_cudandarray(x):
...@@ -86,12 +85,12 @@ try: ...@@ -86,12 +85,12 @@ try:
if not isinstance(x, cudamat.CUDAMatrix): if not isinstance(x, cudamat.CUDAMatrix):
raise ValueError("We can transfer only cudamat.CUDAMatrix to CudaNdarray") raise ValueError("We can transfer only cudamat.CUDAMatrix to CudaNdarray")
# elif x.dtype != "float32": # elif x.dtype != "float32":
# raise ValueError("CudaNdarray support only float32") # raise ValueError("CudaNdarray support only float32")
# We don't need this, because cudamat is always float32. # We don't need this, because cudamat is always float32.
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])
import ctypes import ctypes
......
...@@ -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/cudamat_utils.py",
"misc/pycuda_utils.py", "misc/pycuda_utils.py",
"misc/pycuda_example.py", "misc/pycuda_example.py",
"misc/ordered_set.py", "misc/ordered_set.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论