[BugFix] Propagate mask correctly in allocated_fused_rms_norm_qkv kernel

[aws-qieqingy]

Issue #, if available:

N/A

Description of changes:

The mask of in the allocated_fused_rmsnorm_qkv kernel is not propagated correctly, and is causing failure in internal testing.

Testing:

Please see detailed unit test requirements in the CONTRIBUTING.md

• The change is covered by numeric check using nki.baremetal
• The change is covered by performance benchmark test using nki.benchmark
• The change is covered by end-to-end integration test

Pull Request Checklist

• I have filled in all the required field in the template
• I have tested locally that all the tests pass
• By submitting this pull request, I confirm that my contribution is made under the terms of the MIT-0 license.

[aws-zhehongb]

LGTM