Function _is_tvmscript_class_annotator

python/tvm/script/parser/core/utils.py:188–195

Checks if the line contains a TVMScript annotator for a class These match either `@I.ir_module` or `@R.rewriter`, or their imported names `@ir_module` or `@rewriter`.

(line: str)

    """

    def _is_tvmscript_class_annotator(line: str) -> bool:
        """Checks if the line contains a TVMScript annotator for a class

        These match either `@I.ir_module` or `@R.rewriter`, or their
        imported names `@ir_module` or `@rewriter`.
        """

        return line.startswith("@") and ("ir_module" in line or "rewriter" in line)

    if len(frames) > 2:
        frame_info = frames[2]