提交 32d3ce77 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix GpuSumcum opt in the new back-end.

上级 d20ad2a1
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import os import os
from theano import Apply from theano import Apply, Op
from theano.tensor.extra_ops import CumsumOp from theano.tensor.extra_ops import CumsumOp
try: try:
...@@ -13,7 +13,7 @@ from .basic_ops import (as_gpuarray_variable, GpuKernelBase, Kernel, ...@@ -13,7 +13,7 @@ from .basic_ops import (as_gpuarray_variable, GpuKernelBase, Kernel,
from .opt import register_opt as register_gpu_opt, op_lifter from .opt import register_opt as register_gpu_opt, op_lifter
class GpuCumsum(GpuKernelBase): class GpuCumsum(GpuKernelBase, Op):
""" """
Parameters Parameters
---------- ----------
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论