DeepSpeed
4864991f - Allow triton==3.0.x for fp_quantizer (#6447)

Commit
1 year ago
Allow triton==3.0.x for fp_quantizer (#6447) Tested with triton==3.0.x and the kernel tests pass so adding as an allowed version. Triton 2.3.x is not well supported on arm64. Triton 3.0.0 is supported on arm64 and it appears the fp8 kernel works fine with triton==3.0.0 so this simplifies usage on arm hosts (GH200).
Author
Parents
Loading