Kernels
activation / tests /test_rms_norm_sequence_parallel.py

Commit History

feat: support sequence parallel with fused_add_rms_norm
151bb5a

wyldecat commited on

feat: support sequence parallel with rms_norm
06d6367

wyldecat commited on