-
Notifications
You must be signed in to change notification settings - Fork 61
Milestone
Description
🐛 Describe the bug
import torch
value = torch.load("clamped_value.pt").to(0)
print(value.to(torch.float8_e4m3fn).to(torch.float16).max())XPU output:
tensor(nan, device='xpu:0', dtype=torch.float16)
A100 output:
tensor(288., device='cuda:0', dtype=torch.float16)
Versions
[pip3] torch==2.10.0.dev20251026+xpu
[pip3] torchao==0.15.0.dev20251027+xpu
[pip3] torchaudio==2.10.0.dev20251026+xpu
[pip3] torchdata==0.12.0.dev20250220
[pip3] torchvision==0.25.0.dev20251026+xpu
[pip3] triton==3.5.0
Metadata
Metadata
Assignees
Labels
No labels