From 7934bf5469c6cb229d587d4b88d46ce70efae828 Mon Sep 17 00:00:00 2001 From: wrongtest Date: Thu, 30 Mar 2023 15:24:06 +0800 Subject: [PATCH] fix name of iter var type 4 --- python/tvm/tir/expr.py | 2 +- tests/python/unittest/test_tvmscript_ir_builder_tir.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/tir/expr.py b/python/tvm/tir/expr.py index 52153fd41d63..f93e39ee0fbd 100644 --- a/python/tvm/tir/expr.py +++ b/python/tvm/tir/expr.py @@ -417,7 +417,7 @@ class IterVar(Object, ExprOp, Scriptable): ThreadIndex = 1 CommReduce = 2 Ordered = 3 - DimInfo = 4 + Opaque = 4 Unrolled = 5 Vectorized = 6 Parallelized = 7 diff --git a/tests/python/unittest/test_tvmscript_ir_builder_tir.py b/tests/python/unittest/test_tvmscript_ir_builder_tir.py index 5599d2f7c69a..70180ecbb05f 100644 --- a/tests/python/unittest/test_tvmscript_ir_builder_tir.py +++ b/tests/python/unittest/test_tvmscript_ir_builder_tir.py @@ -193,7 +193,7 @@ def test_ir_builder_tir_axis(): tir.IterVar((0, 8), tir.Var("", "int32"), iter_type=tir.IterVar.DataPar), tir.IterVar((0, 16), tir.Var("", "int32"), iter_type=tir.IterVar.CommReduce), tir.IterVar((0, 32), tir.Var("", "int32"), iter_type=tir.IterVar.Ordered), - tir.IterVar((0, 64), tir.Var("", "int32"), iter_type=tir.IterVar.DimInfo), + tir.IterVar((0, 64), tir.Var("", "int32"), iter_type=tir.IterVar.Opaque), ], reads=[], writes=[],