Commit f9edd578 authored by brkirch's avatar brkirch
Browse files

Remove MPS fix no longer needed for PyTorch 

The torch.narrow fix was required for nightly PyTorch builds for a while to prevent a hard crash, but newer nightly builds don't have this issue.
parent 02b8b957

modules/devices.py

+0 −3
Original line number Diff line number Diff line
@@ -201,6 +201,3 @@ if has_mps(): 
        cumsum_needs_bool_fix = not torch.BoolTensor([True,True]).to(device=torch.device("mps"), dtype=torch.int64).equal(torch.BoolTensor([True,False]).to(torch.device("mps")).cumsum(0))

        torch.cumsum = lambda input, *args, **kwargs: ( cumsum_fix(input, orig_cumsum, *args, **kwargs) )

        torch.Tensor.cumsum = lambda self, *args, **kwargs: ( cumsum_fix(self, orig_Tensor_cumsum, *args, **kwargs) )
 
        orig_narrow = torch.narrow

        torch.narrow = lambda *args, **kwargs: ( orig_narrow(*args, **kwargs).clone() )

CRA Git | Maintained and supported by SUSTech CRA and CCSE