blob: 8a082b966d3387b1c1bf4e932fccc8962b40c9a6 [file] [log] [blame]
#!/usr/bin/env python3
from __future__ import absolute_import, division, print_function, unicode_literals
import torch
# rpc_fork tests use double as the default dtype
torch.set_default_dtype(torch.double)
from rpc_test import RpcTest
from common_distributed import MultiProcessTestCase
from common_utils import run_tests
class RpcTestWithFork(MultiProcessTestCase, RpcTest):
def setUp(self):
super(RpcTestWithFork, self).setUp()
self._fork_processes()
if __name__ == '__main__':
run_tests()