Method eval_expr

python/tvm/script/parser/core/parser.py:501–526

Expression evaluation when parsing. Parameters ---------- node : Union[doc.expr, doc.Expression] The root node of AST tree node of expression to evaluate. extra_vars : Optional[Dict[str, Any]] The optional global value table for expression ev

(
        self,
        node: doc.Expression | doc.expr,
        extra_vars: dict[str, Any] | None = None,
    )

        return _deferred(pop_source)

    def eval_expr(
        self,
        node: doc.Expression | doc.expr,
        extra_vars: dict[str, Any] | None = None,
    ) -> Any:
        """Expression evaluation when parsing.

        Parameters
        ----------
        node : Union[doc.expr, doc.Expression]
            The root node of AST tree node of expression to evaluate.

        extra_vars : Optional[Dict[str, Any]]
            The optional global value table for expression evaluation.

        Returns
        -------
        res : Any
            The evaluation result.
        """
        var_values = self.var_table.get()
        if extra_vars is not None:
            for k, v in extra_vars.items():
                var_values[k] = v
        var_values[ScriptMacro.parser_object_name] = self
        return eval_expr(self, node, var_values)

    def _duplicate_lhs_check(self, target: doc.expr) -> bool | set[str]:
        """Check whether duplicate lhs exists in assignment.