haosdentmarked this pull request as draft 73 days ago
haosdent
changed the title [WIP][Bugfix] Fix dtype mismatch in RMSNormGated.forward_native() during torch.compile [Bugfix] Fix dtype mismatch in RMSNormGated.forward_native() during torch.compile73 days ago
Login to write a write a comment.
Login via GitHub