diff --git a/thunder/dynamo/utils.py b/thunder/dynamo/utils.py index 3c58bc2ded..ae3b0bd6de 100644 --- a/thunder/dynamo/utils.py +++ b/thunder/dynamo/utils.py @@ -533,6 +533,7 @@ def _get_min_and_val(t: torch.Tensor) -> tuple[Number | None, Number | None]: or t.device.type == "meta" or t.numel() == 0 or t.dtype.is_complex + or (hasattr(torch, "float4_e2m1fn_x2") and t.dtype == torch.float4_e2m1fn_x2) ): return None, None if t.dtype in (torch.float8_e4m3fn, torch.float8_e4m3fnuz, torch.float8_e5m2, torch.float8_e5m2fnuz):