DeepSpeed
b6c47f6b - is_triton_supported added to Accelerator class

Commit

2 years ago

is_triton_supported added to Accelerator class clean-up and formatting

References

#4337 - adds triton flash attention2 kernel

Author

styoun

styoun

Parents

Loading