Merge pull request #16088 from cuba3/dev_cuba3
Maintaining Project Compatibility for Python 3.9 Users Without Upgrade Requirements.
This commit is contained in:
commit
edebe4d4de
|
@ -20,7 +20,6 @@ def get_param(model) -> torch.nn.Parameter:
|
||||||
|
|
||||||
def float64(t: torch.Tensor):
|
def float64(t: torch.Tensor):
|
||||||
"""return torch.float64 if device is not mps or xpu, else return torch.float32"""
|
"""return torch.float64 if device is not mps or xpu, else return torch.float32"""
|
||||||
match t.device.type:
|
if t.device.type in ['mps', 'xpu']:
|
||||||
case 'mps', 'xpu':
|
return torch.float32
|
||||||
return torch.float32
|
|
||||||
return torch.float64
|
return torch.float64
|
||||||
|
|
Loading…
Reference in New Issue