mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
987e37b9c2
Summary: According to https://github.com/pytorch/pytorch/issues/27285 , seems we do not intend to use shebang as an indication of Python version, thus we enable EXE001 flake8 check. For violations, we either remove shebang from non-executable Python scripts or grant them executable permission. Pull Request resolved: https://github.com/pytorch/pytorch/pull/27560 Differential Revision: D17831782 Pulled By: ezyang fbshipit-source-id: 6282fd3617b25676a6d959af0d318faf05c09b26
22 lines
523 B
Python
Executable File
22 lines
523 B
Python
Executable File
#!/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()
|