biondizzle b8df4a8cc5 Fix NaN check: use os.environ gate instead of is_current_stream_capturing ...

torch.cuda.is_current_stream_capturing() returns bool, which breaks
Dynamo FX tracing (non-Tensor output). Switch to env var gate:
CLAWMINE_NAN_CHECK=1 enables NaN/Inf detection.

Dynamo evaluates os.environ at trace time — if the env var is not set,
the entire NaN check block is compiled away. Set it before first
inference to get NaN detection during prefill only.

2026-05-18 02:20:14 +00:00

..

patches

Fix NaN check: use os.environ gate instead of is_current_stream_capturing

2026-05-18 02:20:14 +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