提交 63ef2ead authored 作者: Frederic Bastien's avatar Frederic Bastien

flake8

上级 9067f773
...@@ -404,6 +404,7 @@ class GpuAdvancedSubtensor1(HideC, tensor.AdvancedSubtensor1): ...@@ -404,6 +404,7 @@ class GpuAdvancedSubtensor1(HideC, tensor.AdvancedSubtensor1):
AdvancedSubrensor1 on the GPU. AdvancedSubrensor1 on the GPU.
""" """
_f16_ok = True _f16_ok = True
def make_node(self, x, ilist): def make_node(self, x, ilist):
ctx_name = infer_context_name(x, ilist) ctx_name = infer_context_name(x, ilist)
x_ = as_gpuarray_variable(x, ctx_name) x_ = as_gpuarray_variable(x, ctx_name)
......
...@@ -7,7 +7,7 @@ import numpy as np ...@@ -7,7 +7,7 @@ import numpy as np
import theano import theano
from theano import tensor from theano import tensor
from theano.tests import unittest_tools as utt from theano.tests import unittest_tools as utt
from theano.tensor.blas import gemv_inplace, gemm_inplace, _dot22, batched_dot from theano.tensor.blas import gemm, gemv_inplace, gemm_inplace, _dot22, batched_dot
from theano.tensor.tests.test_blas import TestGer, BaseGemv from theano.tensor.tests.test_blas import TestGer, BaseGemv
from .. import gpuarray_shared_constructor from .. import gpuarray_shared_constructor
......
...@@ -68,7 +68,7 @@ class G_subtensorF16(test_subtensor.T_subtensor): ...@@ -68,7 +68,7 @@ class G_subtensorF16(test_subtensor.T_subtensor):
dimshuffle=GpuDimShuffle, dimshuffle=GpuDimShuffle,
mode=mode_with_gpu, mode=mode_with_gpu,
# avoid errors with limited devices # avoid errors with limited devices
dtype='float16', #use floatX? dtype='float16', # use floatX?
ignore_topo=(HostFromGpu, GpuFromHost, ignore_topo=(HostFromGpu, GpuFromHost,
DeepCopyOp, GpuContiguous)) DeepCopyOp, GpuContiguous))
# GPU opt can't run in fast_compile only. # GPU opt can't run in fast_compile only.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论