Method setUpClass

profiler/test/test_profiler_e2e.py:48–69

(cls)

class TestCustomOps(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        model = Module()
        inputs = (torch.ones(2, 2, dtype=torch.float),)

        # The serialized program file. This must live longer than cls.module,
        # because the C++ pybindings will have a pointer to it. But none of the
        # tests should need to touch it.
        cls.__buffer: bytes = (
            to_edge(export(model, inputs, strict=True)).to_executorch().buffer
        )

        cls.module = _load_for_executorch_from_buffer(cls.__buffer)

        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
        cls.inputs_flattened, _ = tree_flatten(inputs)
        cls.module.run_method("forward", tuple(cls.inputs_flattened))
        prof_dump = _dump_profile_results()
        assert (
            len(prof_dump) > 0
        ), "prof_dump is empty; may need to build with `-c executorch.prof_enabled=true`"
        cls.prof_results, cls.mem_results = deserialize_profile_results(prof_dump)
        cls.expect_ops = ["native_call_add.out", "native_call_mul.out"]

    def test_profiler_new_block(self) -> None:
        block_names = ["block_1", "block_2"]