PR #961 Experimental: allow fp16 in `mps`

Experimental: allow fp16 in `mps` #961

pcuenca merged 11 commits into main from mps-allow-fp16

Docs: refer to pre-RC version of PyTorch 1.13.0.

cd221bce

Remove temporary workaround for unavailable op.

c3960edf

Update comment to make it less ambiguous.

8674cf76

Remove use of contiguous in mps.

762650fa

Merge remote-tracking branch 'origin/main' into m1-torch-1.13

f55fcf52

Merge remote-tracking branch 'origin/main' into m1-torch-1.13

de5258f4

Merge remote-tracking branch 'origin/main' into m1-torch-1.13

78dbfeac

Special case: use einsum for much better performance in mps

ef41ddd5

Update mps docs.

8d47740c

MPS: make pipeline work in half precision.

2f8fe8e3

Merge remote-tracking branch 'origin/main' into mps-allow-fp16

ff9553c8

patrickvonplaten approved these changes on 2022-10-26

pcuenca merged 95414bd6 into main 3 years ago

pcuenca deleted the mps-allow-fp16 branch 3 years ago

Reviewers

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone