def METHOD_NAME(f, *args) -> float:
    """
    Assumes all inputs are fp32
    """
    metrics.reset()
    torch._dynamo.optimize(count_bytes_inductor)(f)(*args)
    print(metrics.node_runtimes)
    ret = 0.0
    for pair in metrics.node_runtimes:
        ret += pair[1]
    return ret