Add type hint for cuda.set_rng_state (#26200)

yaroslavvb · facebook-github-bot · commit 7f3c42354164 · 2019-09-14T19:29:42.000-07:00
Summary: Fixes #26199 Pull Request resolved: #26200 Differential Revision: D17386885 Pulled By: soumith fbshipit-source-id: 9da03aae29281b2ed691cbfdd7b85fde55e5b7ef
diff --git a/torch/cuda/__init__.pyi b/torch/cuda/__init__.pyi
@@ -40,3 +40,5 @@ def max_memory_cached(device: Optional[_device_t]=...) -> int: ...
 def reset_max_memory_cached(device: Optional[_device_t]=...) -> None: ...
 def cudart() -> ctypes.CDLL: ...
 def find_cuda_windows_lib() -> Optional[ctypes.CDLL]: ...
+def set_rng_state(new_state): ...
+def get_rng_state(): ...
