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

Function _check

tests/python/relax/test_transform_fuse_tir.py:28–30  ·  view source on GitHub ↗
(mod_before, mod_expected)

Source from the content-addressed store, hash-verified

26
27
28def _check(mod_before, mod_expected):
29 mod_after = relax.transform.FuseTIR()(mod_before)
30 tvm.ir.assert_structural_equal(mod_expected, mod_after)
31
32
33def test_simple():

Callers 15

test_simpleFunction · 0.70
test_conv2d_fuseFunction · 0.70
test_two_subfunctionFunction · 0.70
test_fuse_same_primfuncFunction · 0.70
test_fuse_with_tuple_as_paramFunction · 0.70
test_fuse_with_nested_tuple_as_paramFunction · 0.70
test_fuse_with_call_tir_in_mainFunction · 0.70
test_fuse_with_const_in_argumentFunction · 0.70
test_fuse_tuple_outputFunction · 0.70
test_fuse_with_immediate_tupleFunction · 0.70
test_fuse_return_partial_resultFunction · 0.70
test_multiple_relax_functionsFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…