Function test_example_npu_codegen

tests/python/contrib/test_example_npu.py:205–221

Test code generation for the example NPU backend

()

@example_npu_enabled
def test_example_npu_codegen():
    """Test code generation for the example NPU backend"""
    import tvm.relax.backend.contrib.example_npu  # noqa: F401

    mod = MatmulReLU
    patterns = get_patterns_with_prefix("example_npu")

    # Partition and generate code
    partitioned_mod = FuseOpsByPattern(patterns, bind_constants=False, annotate_codegen=True)(mod)
    partitioned_mod = MergeCompositeFunctions()(partitioned_mod)
    partitioned_mod = RunCodegen()(partitioned_mod)

    assert partitioned_mod is not None

    # The module should now contain external function calls
    main_func = partitioned_mod["main"]
    assert main_func is not None


@example_npu_enabled