未验证 提交 a4b0e4a1 编写于 作者: Y Yibing Liu 提交者: GitHub

Merge pull request #9051 from panyx0718/profiler

Add a test to ensure profiler works on multi-gpu
......@@ -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):
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册