benchmark

0a4eeedb - Support `triton.language.dtype` with `torch.compile` (#121690)

Support `triton.language.dtype` with `torch.compile` (#121690) Summary: Putting this PR as an RFC since I have resorted to some horrible hacks in order to make this work. ``` (Pdb) p triton.language.float32 triton.language.fp32 (Pdb) p str(triton.language.float32) 'fp32' (Pdb) p repr(triton.language.float32) 'triton.language.fp32' ``` This means that we need to "rewrite" them for fx graph and inductor execution. This PR allows Mamba2 to work with `torch.compile`. X-link: https://github.com/pytorch/pytorch/pull/121690 Approved by: https://github.com/Skylion007 Reviewed By: osalpekar Differential Revision: D54913846 Pulled By: oulgen fbshipit-source-id: 0a85be88e60149acc4c9d799c24598d62dcf7bbf