xora/utils/dist_util.py

from enum import Enum

def execute_graph() -> None:
    if _acceleration_type == AccelerationType.TPU:
        xm.mark_step()