Skip to content

Commit

Permalink
Remove flaky test (#6120)
Browse files Browse the repository at this point in the history
  • Loading branch information
jonb377 authored Dec 13, 2023
1 parent f87e971 commit 41d0fe5
Showing 1 changed file with 0 additions and 38 deletions.
38 changes: 0 additions & 38 deletions test/test_profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,44 +93,6 @@ def train_worker():
self._check_trace_namespace_exists(path)
self._check_metrics_warnings_exist(self.fname)

def test_trace_detached(self):

port = xu.get_free_tcp_ports()[0]
training_started = multiprocessing.Event()
logdir = tempfile.mkdtemp()

def train_worker():
flags = args_parse.parse_common_options(
datadir='/tmp/mnist-data',
batch_size=16,
momentum=0.5,
lr=0.01,
num_epochs=10)
flags.fake_data = True
flags.profiler_port = port

# Set programmatic capture options
flags.profile_step = 10
flags.profile_epoch = 1
flags.profile_logdir = logdir
flags.profile_duration_ms = 5000

test_profile_mp_mnist.train_mnist(
flags,
training_started=training_started,
dynamic_graph=True,
fetch_often=True)

p = multiprocessing.Process(target=train_worker, daemon=True)
p.start()
training_started.wait(60)
# Delay to allow the profile to capture
time.sleep(10)
p.terminate()
path = self._check_xspace_pb_exist(logdir)
self._check_trace_namespace_exists(path)
self._check_metrics_warnings_exist(self.fname)


if __name__ == '__main__':
logging.getLogger().setLevel(logging.INFO)
Expand Down

0 comments on commit 41d0fe5

Please sign in to comment.