3 people 30870b4f66 [torch.compile] Dynamic fp8 + rms_norm fusion (#10906) ...

Signed-off-by: luka <[email protected]>
Co-authored-by: Varun Sundar Rabindranath <[email protected]>

2024-12-13 03:19:23 +00:00

..

fused_layernorm_dynamic_per_token_quant.cu

[torch.compile] Dynamic fp8 + rms_norm fusion (#10906)

2024-12-13 03:19:23 +00:00

layernorm_utils.cuh

[torch.compile] Dynamic fp8 + rms_norm fusion (#10906)

2024-12-13 03:19:23 +00:00

quant_conversions.cuh

[torch.compile] Dynamic fp8 + rms_norm fusion (#10906)

2024-12-13 03:19:23 +00:00