Skip to content

Commit 8da296b

Browse files
committed
avoid `torch.float4_e2m1fn_x2 in _get_min_and_val
Signed-off-by: Masaki Kozuki <[email protected]>
1 parent 280c57e commit 8da296b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

thunder/dynamo/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -533,6 +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 t.dtype == torch.float4_e2m1fn_x2
536537
):
537538
return None, None
538539
if t.dtype in (torch.float8_e4m3fn, torch.float8_e4m3fnuz, torch.float8_e5m2, torch.float8_e5m2fnuz):

0 commit comments

Comments
 (0)