add print test_dist UT elapsed time (#26397)

* add print test_dist UT elapsed time

* UT elapsed time optimize
revert-24895-update_cub
tangwei12 5 years ago committed by GitHub
parent 9bcfc1a804
commit 39bd2fb5e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -57,7 +57,7 @@ class FleetDistRunnerBase(object):
if args.role.upper() == "PSERVER":
role = role_maker.UserDefinedRoleMaker(
is_collective=False,
init_gloo=True,
init_gloo=False,
path=args.gloo_path,
current_id=args.current_id,
role=role_maker.Role.SERVER,
@ -66,7 +66,7 @@ class FleetDistRunnerBase(object):
else:
role = role_maker.UserDefinedRoleMaker(
is_collective=False,
init_gloo=True,
init_gloo=False,
path=args.gloo_path,
current_id=args.current_id,
role=role_maker.Role.WORKER,
@ -160,7 +160,13 @@ class TestFleetBase(unittest.TestCase):
def _setup_config(self):
raise NotImplementedError("tests should have _setup_config implemented")
def tearDown(self):
t = time.time() - self.startTime
print('%s: %.3f' % (self.__class__.__name__, t))
def setUp(self):
self.startTime = time.time()
self._mode = "sync"
self._reader = "pyreader"
self._trainers = 2

Loading…
Cancel
Save