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

Function test_op_call

tests/python/tirx/transform/test_stmt_functor.py:950–981  ·  view source on GitHub ↗

Test op call statement

()

Source from the content-addressed store, hash-verified

948
949
950def test_op_call():
951 """Test op call statement"""
952 op_call = create_test_statements()["op_call"]
953 basic_check(
954 op_call,
955 "\n".join(
956 [
957 "Block",
958 "\tTilePrimitiveCall",
959 "\t\tBufferRegion",
960 "\t\t\tIntImm",
961 "\t\t\tIntImm",
962 "\t\tBufferRegion",
963 "\t\t\tIntImm",
964 "\t\t\tIntImm",
965 "\t\tExpr::FloatImm",
966 ]
967 ),
968 "\n".join(
969 [
970 "IntImm",
971 "IntImm",
972 "BufferRegion",
973 "IntImm",
974 "IntImm",
975 "BufferRegion",
976 "Expr::FloatImm",
977 "TilePrimitiveCall",
978 "Block",
979 ]
980 ),
981 )
982
983
984def test_stmt_expr_mutator():

Callers

nothing calls this directly

Calls 3

create_test_statementsFunction · 0.70
basic_checkFunction · 0.70
joinMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…