This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new e51f5bb [TOPI] Bugfix for topi.prod (#8416)
e51f5bb is described below
commit e51f5bb383baa0fbc2f12621101a9c361072cdb2
Author: Y <c...@live.com>
AuthorDate: Sat Jul 10 01:49:09 2021 +0800
[TOPI] Bugfix for topi.prod (#8416)
---
python/tvm/topi/scan.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/python/tvm/topi/scan.py b/python/tvm/topi/scan.py
index f579673..32a7e29 100644
--- a/python/tvm/topi/scan.py
+++ b/python/tvm/topi/scan.py
@@ -23,7 +23,7 @@ import tvm
from ..te import extern
from ..tir import decl_buffer, generic, ir_builder
from .math import cast
-from .utils import get_const_int, prod
+from . import utils
def scanop(
@@ -93,11 +93,11 @@ def scanop(
if axis is None:
axis = 0
- cumsum_axis_len = prod(data.shape)
+ cumsum_axis_len = utils.prod(data.shape)
shape = (cumsum_axis_len,)
else:
if not isinstance(axis, int):
- axis = get_const_int(axis)
+ axis = utils.get_const_int(axis)
shape = data.shape
cumsum_axis_len = shape[axis]
