Function func

tests/python/tirx/codegen/test_codegen_hopper.py:66–71

(A: T.Buffer(1))

    # fmt: off
    @T.prim_func
    def func(A: T.Buffer(1)):
        T.device_entry()
        cta_id = T.cta_id([1])
        tid = T.thread_id([128])
        T.ptx.setmaxnreg(inc, 32)
        # fmt: on

    src, mod = _get_source(func)
    assert "setmaxnreg" in src