Method eval

python/tvm/script/parser/core/evaluator.py:101–132

Expression evaluation for TVMScript parser. Parameters ---------- parser : Parser The parser bound with the evaluator. value_table : Dict[str, Any] The value table for expression evaluation. node : doc.AST The root node o

(parser: "Parser", value_table: dict[str, Any], node: doc.AST)

    @staticmethod
    def eval(parser: "Parser", value_table: dict[str, Any], node: doc.AST) -> Any:
        """Expression evaluation for TVMScript parser.

        Parameters
        ----------
        parser : Parser
            The parser bound with the evaluator.

        value_table : Dict[str, Any]
            The value table for expression evaluation.

        node : doc.AST
            The root node of AST tree node of expression to evaluate.

        Returns
        -------
        res : Any
            The evaluation result.
        """
        self = ExprEvaluator(parser, value_table)
        result = self._visit(node)  # pylint: disable=protected-access
        if isinstance(result, doc.Name):
            if result.id in self.value_table:
                return self.value_table[result.id]
            else:
                builtin = _get_builtin_or_none(result.id)
                if builtin:
                    return builtin
                raise ParserError(result, f"Undefined variable: {result.id}")
        if isinstance(result, doc.Constant):
            return result.value
        raise TypeError(f"Unexpected result type: {type(result)}")

    def _add_intermediate_result(self, value: Any) -> doc.Name:
        """Add intermediate result during evaluation into value table.