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
19 lines
553 B
Python
Executable File
19 lines
553 B
Python
Executable File
#!/usr/bin/env python3
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
from rpc_test import RpcTest
|
|
from common_distributed import MultiProcessTestCase
|
|
from common_utils import TEST_WITH_ASAN, run_tests
|
|
|
|
import unittest
|
|
|
|
@unittest.skipIf(TEST_WITH_ASAN, "Skip ASAN as torch + multiprocessing spawn have known issues")
|
|
class RpcTestWithSpawn(MultiProcessTestCase, RpcTest):
|
|
|
|
def setUp(self):
|
|
super(RpcTestWithSpawn, self).setUp()
|
|
self._spawn_processes()
|
|
|
|
if __name__ == '__main__':
|
|
run_tests()
|