Skip to content

MPS AMP Support #14358

@justusschock

Description

@justusschock

🚀 Feature

After pytorch/pytorch#78168 was apparently resolved on pytorch master/nightlies we should re-evaluate fp16/amp support for mps devices.

cc @Borda @carmocca @justusschock @awaelchli @akihironitta @rohitgr7

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

Relationships

None yet

Development

No branches or pull requests

Issue actions