code: def load_triton_flash_attn():
      global triton_flash_attn
      try:
          import triton
      except ImportError:
          raise ImportError(
              "Please install triton 2.0+ or `pip install deepspeed[sd]`"
          )
  
      if pkg_version.parse(triton.__version__) < pkg_version.parse(
          "2.0"
      ):
          raise ImportError(
              "Please install triton 2.0+ or `pip install deepspeed[sd]`"
          )
  
      from .triton_ops import triton_flash_attn
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/ops/transformer/inference/diffusers_attention.py
imports: import math
  import torch
  from torch.autograd import Function
  from ...None import op_builder
  import torch.nn
  from packaging import version
  from deepspeed.utils.logging import log_dist
label_desc: Use image node_id 1 for calling a global function with example usage: load_triton_flash_attn() without return types