Function _calc

tests/python/tvmscript/test_tvmscript_parser_evaluator.py:27–35

(expr, extra_vars=None)

def _calc(expr, extra_vars=None):
    if extra_vars is None:
        extra_vars = {}
    source = Source(expr)
    mod_ast = source.as_ast()
    mod_body_ast = mod_ast.body
    expr_stmt_ast = mod_body_ast[0]
    expr_ast = expr_stmt_ast.value
    return ExprEvaluator.eval(None, extra_vars, expr_ast)


def test_evaluator_basic():