[tvm] branch main updated: [TOPI] Bugfix for topi.prod (#8416)

[wuwei]

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]