DeepSpeed
301bc288 - import triton files when triton is supported and installed (#6989)

Comment changes are shownComment changes are hidden
Commit
157 days ago
import triton files when triton is supported and installed (#6989) Those files have code that gets run when importing them, so in systems that doesn't support triton but have triton installed this causes issues. In general, I think it is better to import triton when it is installed and supported. Signed-off-by: Omar Elayan <oelayan@habana.ai>
Author
Parents
  • deepspeed/model_implementations/transformers
    • File
      ds_transformer.py