Source code for deepfold.modules.inductor

import torch

_ENABLED = False



[docs]
def enable() -> None:
    global _ENABLED
    _ENABLED = True




[docs]
def disable():
    global _ENABLED
    _ENABLED = False




[docs]
def is_enabled() -> bool:
    return _ENABLED




[docs]
def is_enabled_and_autograd_off() -> bool:
    return _ENABLED and not torch.is_grad_enabled()