guard

Signed-off-by: Masaki Kozuki <[email protected]>

Author: crcrpar

thunder/dynamo/utils.py

@@ -533,7 +533,10 @@ 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 t.dtype == torch.float4_e2m1fn_x2
+        or (
+            not hasattr(torch, "float4_e2m1fn_x2")
+            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):