From e47d75aff9261de84fad0b2b5b78f2c350606f71 Mon Sep 17 00:00:00 2001 From: jhlee525 Date: Mon, 11 Dec 2023 13:49:46 +0900 Subject: [PATCH 1/2] remove deprecated np.int in slice --- python/tvm/relay/frontend/pytorch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index eadd0a3c464f..d5307a153e51 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -446,10 +446,10 @@ def slice(self, inputs, input_types): stride = inputs[4] target_begin, is_begin_const = try_infer_value( - inputs[2], lambda ret: ret.astype(np.int).item(0) + inputs[2], lambda ret: ret.astype(int).item(0) ) target_end, is_end_const = try_infer_value( - inputs[3], lambda ret: ret.astype(np.int).item(0) + inputs[3], lambda ret: ret.astype(int).item(0) ) # A fast path when slicing is nop. From 22fcd2e6eb04c95d77ebf17f83ff71b628fd8f92 Mon Sep 17 00:00:00 2001 From: jhlee525 Date: Wed, 13 Dec 2023 14:08:58 +0900 Subject: [PATCH 2/2] lint --- python/tvm/relay/frontend/pytorch.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index d5307a153e51..b6be05a3d69e 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -448,9 +448,7 @@ def slice(self, inputs, input_types): target_begin, is_begin_const = try_infer_value( inputs[2], lambda ret: ret.astype(int).item(0) ) - target_end, is_end_const = try_infer_value( - inputs[3], lambda ret: ret.astype(int).item(0) - ) + target_end, is_end_const = try_infer_value(inputs[3], lambda ret: ret.astype(int).item(0)) # A fast path when slicing is nop. if (