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

Function _check_inference

tests/python/relax/test_op_ccl.py:34–36  ·  view source on GitHub ↗
(bb: relax.BlockBuilder, call: relax.Call, expected_sinfo: relax.StructInfo)

Source from the content-addressed store, hash-verified

32
33
34def _check_inference(bb: relax.BlockBuilder, call: relax.Call, expected_sinfo: relax.StructInfo):
35 ret = bb.normalize(call)
36 tvm.ir.assert_structural_equal(ret.struct_info, expected_sinfo)
37
38
39def test_allreduce_infer_struct_info():

Callers 15

test_allreduce_infer_struct_infoFunction · 0.70
test_allreduce_infer_struct_info_shape_symbolicFunction · 0.70
test_allreduce_infer_struct_info_shape_varFunction · 0.70
test_allreduce_infer_struct_info_more_input_dtypeFunction · 0.70
test_allgather_infer_struct_infoFunction · 0.70
test_allgather_infer_struct_info_shape_symbolicFunction · 0.70
test_allgather_infer_struct_info_shape_varFunction · 0.70
test_allgather_infer_struct_info_more_input_dtypeFunction · 0.70
test_broadcast_from_worker0_infer_struct_infoFunction · 0.70
test_broadcast_from_worker0_infer_struct_info_shape_symbolicFunction · 0.70
test_broadcast_from_worker0_infer_struct_info_shape_varFunction · 0.70
test_broadcast_from_worker0_infer_struct_info_more_input_dtypeFunction · 0.70

Calls 1

normalizeMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…