mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
Revert "[MPS] Add Python Module Bindings for the MPS backend (#94417)"
This reverts commit beb4f5bf396ec2d53defa73c81aac48c38360544.
Reverted https://github.com/pytorch/pytorch/pull/94417 on behalf of https://github.com/huydhn due to Sorry for reverting your PR, but it seems to break MacOS test in trunk bae397ec63
This commit is contained in:
@ -39,9 +39,6 @@ def manual_seed(seed) -> torch._C.Generator:
|
||||
if not torch.cuda._is_in_bad_fork():
|
||||
torch.cuda.manual_seed_all(seed)
|
||||
|
||||
import torch.mps
|
||||
torch.mps.manual_seed(seed)
|
||||
|
||||
return default_generator.manual_seed(seed)
|
||||
|
||||
|
||||
@ -55,9 +52,6 @@ def seed() -> int:
|
||||
if not torch.cuda._is_in_bad_fork():
|
||||
torch.cuda.manual_seed_all(seed)
|
||||
|
||||
import torch.mps
|
||||
torch.mps.manual_seed(seed)
|
||||
|
||||
return seed
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user