biondizzle 65763a200c Fix NaN check: wrap in @torch.compiler.disable to prevent Dynamo graph break ...

The inline os.environ gate doesn't work — Dynamo still sees the
data-dependent branching (torch.isnan().any()) and crashes with
'Unsupported: Data-dependent branching'. Extracting into a
@torch.compiler.disable decorated function makes Dynamo skip it.

2026-05-18 11:33:29 +00:00

..

patches

Fix NaN check: wrap in @torch.compiler.disable to prevent Dynamo graph break

2026-05-18 11:33:29 +00:00

nvfp4_cutedsl.py

HOTFIX: remove NaN checks from run() — torch.isnan().any() does CPU-GPU sync, breaks cudagraph

2026-05-17 22:28:32 +00:00