From 84bbe95b9cf0f6e36b99670fb5a2d9e8ff9083bc Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Thu, 16 Jul 2020 21:46:45 +0000 Subject: [PATCH] fix data type for overflow problems --- src/tir/transforms/lower_tvm_builtin.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tir/transforms/lower_tvm_builtin.cc b/src/tir/transforms/lower_tvm_builtin.cc index c41470725e39c..c8df122d40b5c 100644 --- a/src/tir/transforms/lower_tvm_builtin.cc +++ b/src/tir/transforms/lower_tvm_builtin.cc @@ -212,7 +212,7 @@ class BuiltinLower : public StmtExprMutator { } // call packed. PrimExpr MakeCallPacked(const CallNode* op) { - size_t restore_shape_stack = run_shape_stack_; + int64_t restore_shape_stack = run_shape_stack_; size_t restore_array_stack = run_array_stack_; size_t arg_stack_begin = run_arg_stack_; run_arg_stack_ += op->args.size(); @@ -251,7 +251,7 @@ class BuiltinLower : public StmtExprMutator { } PrimExpr MakeCallTracePacked(const CallNode* op) { - size_t restore_shape_stack = run_shape_stack_; + int64_t restore_shape_stack = run_shape_stack_; size_t restore_array_stack = run_array_stack_; size_t arg_stack_begin = run_arg_stack_; run_arg_stack_ += op->args.size();