This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 976a95172879ec708eff7fb56696bacffd99e7f2
Author: Wuwei Lin <wu...@apache.org>
AuthorDate: Mon Dec 11 20:09:16 2023 +0000
fix
---
python/tvm/relax/transform/legalize_ops/index.py | 18 +++++++++---------
1 file changed, 9 insertions(+), 9 deletions(-)
diff --git a/python/tvm/relax/transform/legalize_ops/index.py
b/python/tvm/relax/transform/legalize_ops/index.py
index 13228c4805..4472e7c9ce 100644
--- a/python/tvm/relax/transform/legalize_ops/index.py
+++ b/python/tvm/relax/transform/legalize_ops/index.py
@@ -37,15 +37,15 @@ def _take(bb: BlockBuilder, call: Call) -> Expr:
@register_legalize("relax.strided_slice")
def _strided_slice(bb: BlockBuilder, call: Call) -> Expr:
- if not all(
- isinstance(call.args[0].struct_info.shape.values[i.value], tir.IntImm)
- for i in call.attrs.axes
- ):
- logging.info(
- "Cases where an axis with symbolic length is sliced are not able "
- "to be legalized through TOPI"
- )
- return call
+ # if not all(
+ # isinstance(call.args[0].struct_info.shape.values[i.value],
tir.IntImm)
+ # for i in call.attrs.axes
+ # ):
+ # logging.info(
+ # "Cases where an axis with symbolic length is sliced are not able
"
+ # "to be legalized through TOPI"
+ # )
+ # return call
strides = (
[tir.IntImm("int64", 1)] * len(call.attrs.axes)
