From 98c9980500eaa05dc311a60c4bfa7471e10f405a Mon Sep 17 00:00:00 2001 From: kice Date: Thu, 5 Sep 2019 19:21:54 -0400 Subject: [PATCH] Fix int32 range overflow by using int64 (#3870) --- python/tvm/autotvm/task/space.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/tvm/autotvm/task/space.py b/python/tvm/autotvm/task/space.py index e57f993..8b2dc65 100644 --- a/python/tvm/autotvm/task/space.py +++ b/python/tvm/autotvm/task/space.py @@ -204,8 +204,9 @@ class SplitSpace(TransformSpace): def _generate_space(self, now, tmp_stack): """Generate space by DFS""" if now == self.num_outputs - 1: - if self.product % np.prod(tmp_stack) == 0: - first = int(self.product // int(np.prod(tmp_stack))) + size = np.prod(tmp_stack, dtype=np.int64) + if self.product % size == 0: + first = int(self.product // int(size)) self.entities.append(SplitEntity([first] + tmp_stack[::-1])) else: for factor in self.factors[now]: -- 2.7.4