Hey, running h100_bench.py raises an error right now, because mha_forward no longer expects l_vec and mha_backward no longer expects d_vec.

I think this issue has been present since 1ac8a92#diff-8d146b717e323dc978e2c78afdc7ac6cf11ca9a29f5c80c7a3e62890e1ae292b.