code: def disable() -> None:
      global _TRI_MHA_ENABLED
      _TRI_MHA_ENABLED = False
  
file_path: apex/apex/contrib/openfold_triton/mha.py
imports: import math
  from typing import Optional
  import torch
  import triton
  from einops import rearrange
  from apex.contrib.openfold_triton._mha_kernel import _attention_core, _bwd_kernel, _bwd_preprocess
label_desc: Use image node_id 3 for calling a global function with example usage: disable() without return types