提交 9964c7f4 authored 作者: Frederic Bastien's avatar Frederic Bastien

Remove useless exception.

上级 530999f7
...@@ -8,7 +8,7 @@ from six.moves import StringIO, xrange ...@@ -8,7 +8,7 @@ from six.moves import StringIO, xrange
from theano.gof.utils import MethodNotDefined from theano.gof.utils import MethodNotDefined
from theano.scalar import Scalar, Composite from theano.scalar import Scalar, Composite
from theano.tensor.elemwise import (Elemwise, DimShuffle, CAReduceDtype) from theano.tensor.elemwise import (Elemwise, DimShuffle, CAReduceDtype)
from theano.scalar.basic_scipy import Erfinv, Erfcinv, GammaLn from theano.scalar.basic_scipy import Erfinv, Erfcinv
from theano.scalar.basic import upgrade_to_float_no_complex, complex_types from theano.scalar.basic import upgrade_to_float_no_complex, complex_types
try: try:
...@@ -2493,13 +2493,6 @@ class GpuCAReduceCuda(GpuKernelBase, HideC, CAReduceDtype): ...@@ -2493,13 +2493,6 @@ class GpuCAReduceCuda(GpuKernelBase, HideC, CAReduceDtype):
return kernels return kernels
class GpuGammaLn(GammaLn):
def c_headers(self):
return ['math_functions.h']
gpu_gammaln = GpuGammaLn(upgrade_to_float_no_complex, name='gpu_gammaln')
class GpuErfinv(Erfinv): class GpuErfinv(Erfinv):
""" """
Inverse error function for GPU. Inverse error function for GPU.
......
...@@ -19,7 +19,7 @@ from theano.ifelse import IfElse ...@@ -19,7 +19,7 @@ from theano.ifelse import IfElse
from theano.misc.ordered_set import OrderedSet from theano.misc.ordered_set import OrderedSet
from theano.scalar.basic import Scalar, Pow, Cast from theano.scalar.basic import Scalar, Pow, Cast
from theano.scalar.basic_scipy import Erfinv, Erfcinv, GammaLn from theano.scalar.basic_scipy import Erfinv, Erfcinv
from theano.scan_module import scan_utils, scan_op, scan_opt from theano.scan_module import scan_utils, scan_op, scan_opt
from theano.tensor.nnet import bn from theano.tensor.nnet import bn
...@@ -61,7 +61,7 @@ from .nnet import (gpu_crossentropy_softmax_1hot_with_bias_dx, ...@@ -61,7 +61,7 @@ from .nnet import (gpu_crossentropy_softmax_1hot_with_bias_dx,
gpu_crossentropy_softmax_argmax_1hot_with_bias, gpu_crossentropy_softmax_argmax_1hot_with_bias,
gpu_softmax_with_bias, gpu_softmax) gpu_softmax_with_bias, gpu_softmax)
from .elemwise import (GpuElemwise, GpuDimShuffle, GpuCAReduceCuda, from .elemwise import (GpuElemwise, GpuDimShuffle, GpuCAReduceCuda,
GpuCAReduceCPY, gpu_erfinv, gpu_erfcinv, gpu_gammaln, GpuCAReduceCPY, gpu_erfinv, gpu_erfcinv,
max_inputs_to_GpuElemwise) max_inputs_to_GpuElemwise)
from .subtensor import (GpuIncSubtensor, GpuSubtensor, from .subtensor import (GpuIncSubtensor, GpuSubtensor,
GpuAdvancedSubtensor, GpuAdvancedSubtensor,
...@@ -712,8 +712,7 @@ def local_gpua_elemwise(op, context_name, inputs, outputs): ...@@ -712,8 +712,7 @@ def local_gpua_elemwise(op, context_name, inputs, outputs):
elif kind.startswith(b'cuda'): elif kind.startswith(b'cuda'):
have_cuda = True have_cuda = True
convert = {Erfinv: gpu_erfinv, convert = {Erfinv: gpu_erfinv,
Erfcinv: gpu_erfcinv, Erfcinv: gpu_erfcinv}
GammaLn: gpu_gammaln}
if scal_op.__class__ in convert: if scal_op.__class__ in convert:
scal_op = convert[scal_op.__class__] scal_op = convert[scal_op.__class__]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论