blob: 43521540a2b0cc37929a607c0a0dbf65e9e549e4 [file] [log] [blame]
#!/usr/bin/env python3
from __future__ import absolute_import, division, print_function, unicode_literals
from dist_optimizer_test import DistOptimizerTest
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 DistOptimizerTestWithSpawn(MultiProcessTestCase, DistOptimizerTest):
def setUp(self):
super(DistOptimizerTestWithSpawn, self).setUp()
self._spawn_processes()
if __name__ == '__main__':
run_tests()