提交 5f08e4d7 authored 作者: notoraptor's avatar notoraptor

Assert axis is not None for GpuCumOp.

上级 273912a6
...@@ -28,7 +28,8 @@ class GpuCumOp(GpuKernelBase, Op): ...@@ -28,7 +28,8 @@ class GpuCumOp(GpuKernelBase, Op):
context=gpu_context_type) context=gpu_context_type)
def __init__(self, axis, mode='add'): def __init__(self, axis, mode='add'):
self.axis = int(axis) if axis is not None else 0 assert axis is not None
self.axis = int(axis)
self.mode = mode self.mode = mode
def __eq__(self, other): def __eq__(self, other):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论