PR #1488 Do not use torch.long in mps

Do not use torch.long in mps #1488

pcuenca merged 6 commits into main from mps-unet-fix

Do not use torch.long in mps

4af1b0f8

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

anton-l 3 years ago

pcuenca commented on 2022-11-30

Use torch.int instead of float.

fe201e77

Propagate changes.

b90da60d

Do not silently change float -> int.

83879cb9

Propagate changes.

1a5d0155

patrickvonplaten approved these changes on 2022-12-01

anton-l commented on 2022-12-01

Apply suggestions from code review

fdd11e3f

pcuenca merged 3ceaa280 into main 3 years ago

pcuenca deleted the mps-unet-fix branch 3 years ago

Reviewers

patrickvonplaten

anton-l

Assignees

No one assigned

Labels

None yet

Milestone

No milestone