From ebde3b1a0a6ffeb8c8898e69c240a7895b1f01fd Mon Sep 17 00:00:00 2001
From: Xin Pan <panxin.grad@gmail.com>
Date: Tue, 13 Mar 2018 23:01:44 -0700
Subject: [PATCH] Reproduce profiler failure on multi-gpu.
---
.../fluid/tests/unittests/test_parallel_op.py | 20 ++++++++++++++-----
1 file changed, 15 insertions(+), 5 deletions(-)
diff --git a/python/paddle/fluid/tests/unittests/test_parallel_op.py b/python/paddle/fluid/tests/unittests/test_parallel_op.py
index 1a7551c57b..79bea148f9 100644
--- a/python/paddle/fluid/tests/unittests/test_parallel_op.py
+++ b/python/paddle/fluid/tests/unittests/test_parallel_op.py
@@ -15,6 +15,7 @@
import unittest
import paddle.fluid as fluid
+import paddle.fluid.profiler as profiler
import numpy
@@ -60,20 +61,23 @@ class BaseParallelForTest(unittest.TestCase):
feed=feed,
fetch=fetch,
place=gpu,
- use_parallel=False)
+ use_parallel=False,
+ use_gpu=True)
result_gpu_parallel = self._run_test_impl_(
callback=callback,
feed=feed,
fetch=fetch,
place=gpu,
- use_parallel=True)
+ use_parallel=True,
+ use_gpu=True)
result_gpu_nccl = self._run_test_impl_(
callback=callback,
feed=feed,
fetch=fetch,
place=gpu,
use_parallel=True,
- use_nccl=True)
+ use_nccl=True,
+ use_gpu=True)
self._assert_same_(fetch, result_cpu, result_cpu_parallel,
result_gpu, result_gpu_parallel, result_gpu_nccl)
else:
@@ -85,7 +89,8 @@ class BaseParallelForTest(unittest.TestCase):
fetch,
place,
use_parallel=False,
- use_nccl=False):
+ use_nccl=False,
+ use_gpu=False):
"""
Run a single test, returns the fetch values
Args:
@@ -132,7 +137,12 @@ class BaseParallelForTest(unittest.TestCase):
exe = fluid.Executor(place)
exe.run(startup)
- return exe.run(main, feed=feed, fetch_list=fetch)
+ if use_gpu:
+ profile_type = 'GPU'
+ else:
+ profile_type = 'CPU'
+ with profiler.profiler(profile_type, 'total', '/tmp/profiler'):
+ return exe.run(main, feed=feed, fetch_list=fetch)
def _assert_same_(self, fetch, *args):
"""