We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5aa80cb commit 8ae215dCopy full SHA for 8ae215d
extra/torch_backend/backend.py
@@ -18,6 +18,8 @@ class TinyBackend:
18
def is_initialized(self): return True
19
def is_available(self): return True
20
def current_device(self): return 0
21
+ def _is_in_bad_fork(self): return False
22
+ def manual_seed_all(self, seed: int): Tensor.manual_seed(seed)
23
torch.utils.rename_privateuse1_backend("tiny")
24
torch._register_device_module("tiny", TinyBackend())
25
torch.utils.generate_methods_for_privateuse1_backend()
0 commit comments