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

flake8

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