• brkirch's avatar
    Add fixes for PyTorch 1.12.1 · e247b740
    brkirch authored
    Fix typo "MasOS" -> "macOS"
    
    If MPS is available and PyTorch is an earlier version than 1.13:
    * Monkey patch torch.Tensor.to to ensure all tensors sent to MPS are contiguous
    * Monkey patch torch.nn.functional.layer_norm to ensure input tensor is contiguous (required for this program to work with MPS on unmodified PyTorch 1.12.1)
    e247b740
devices.py 3.54 KB