MCPcopy Create free account
hub / github.com/apache/tvm / test_shape_expr

Function test_shape_expr

tests/python/relax/test_ast_printer.py:197–205  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

195
196
197def test_shape_expr() -> None:
198 m = tirx.Var("m", dtype="int32")
199 n = tirx.Var("n", dtype="int32")
200 s = rx.ShapeExpr([m, n])
201 s_str = dump_ast(s)
202 assert s_str.startswith("ShapeExpr(")
203 assert "values=" in s_str
204 assert "PrimExpr(value=`m: int32`)" in s_str
205 assert "PrimExpr(value=`n: int32`)" in s_str
206
207
208def test_func():

Callers

nothing calls this directly

Calls 1

dump_astFunction · 0.90

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…