Skip to content

Commit 8ae215d

Browse files
authored
torch backend fix manual seed warning (tinygrad#9292)
1 parent 5aa80cb commit 8ae215d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

extra/torch_backend/backend.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@ class TinyBackend:
1818
def is_initialized(self): return True
1919
def is_available(self): return True
2020
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)
2123
torch.utils.rename_privateuse1_backend("tiny")
2224
torch._register_device_module("tiny", TinyBackend())
2325
torch.utils.generate_methods_for_privateuse1_backend()

0 commit comments

Comments
 (0)