Skip to content

Commit 5c5a438

Browse files
committed
fix cond
Signed-off-by: Masaki Kozuki <[email protected]>
1 parent 24729b6 commit 5c5a438

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

thunder/dynamo/utils.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -533,10 +533,7 @@ def _get_min_and_val(t: torch.Tensor) -> tuple[Number | None, Number | None]:
533533
or t.device.type == "meta"
534534
or t.numel() == 0
535535
or t.dtype.is_complex
536-
or (
537-
not hasattr(torch, "float4_e2m1fn_x2")
538-
or (hasattr(torch, "float4_e2m1fn_x2") and t.dtype == torch.float4_e2m1fn_x2)
539-
)
536+
or (hasattr(torch, "float4_e2m1fn_x2") and t.dtype == torch.float4_e2m1fn_x2)
540537
):
541538
return None, None
542539
if t.dtype in (torch.float8_e4m3fn, torch.float8_e4m3fnuz, torch.float8_e5m2, torch.float8_e5m2fnuz):

0 commit comments

Comments
 (0)