Kernels
activation / tests /test_rms_norm_sequence_parallel.py

Commit History

  1. feat: support sequence parallel with fused_add_rms_norm

    151bb5a

  1. feat: support sequence parallel with rms_norm

    06d6367