Method main

tests/python/relax/test_vm_builtin_lower.py:32–42

(x: R.Tensor(("m", "n"), "float32"))

    class Before:
        @R.function
        def main(x: R.Tensor(("m", "n"), "float32")) -> R.Tensor:
            R.func_attr({"relax.force_pure": True})
            m, n = T.int64(), T.int64()

            storage = R.memory.alloc_storage(R.shape([m * n * 4]), 0, "global", "uint8")
            alloc = R.memory.alloc_tensor(storage, 0, R.shape([m, n]), "float32")
            _ = R.call_packed(
                "test.op.identity", x, alloc, sinfo_args=(R.Tensor(ndim=2, dtype="float32"))
            )
            gv0 = alloc
            return gv0

    @I.ir_module
    class Expected: