未验证 提交 399d3a2d 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #6022 from qingqing01/nvprof

Small fix in profiler.py of fluid.
...@@ -22,21 +22,21 @@ def cuda_profiler(output_file, output_mode=None, config=None): ...@@ -22,21 +22,21 @@ def cuda_profiler(output_file, output_mode=None, config=None):
`output_file` with Key-Value pair format or Comma separated values format. `output_file` with Key-Value pair format or Comma separated values format.
The user can set the output mode by `output_mode` argument and set the The user can set the output mode by `output_mode` argument and set the
counters/options for profiling by `config` argument. The default config counters/options for profiling by `config` argument. The default config
caontains 'gpustarttimestamp', 'gpustarttimestamp', 'gridsize3d', is ['gpustarttimestamp', 'gpustarttimestamp', 'gridsize3d',
'threadblocksize', 'streamid', 'enableonstart 0', 'conckerneltrace'. 'threadblocksize', 'streamid', 'enableonstart 0', 'conckerneltrace'].
Args: Args:
output_file (string) : The output file name, the result will be output_file (string) : The output file name, the result will be
written into this file. written into this file.
output_mode (string) : The output mode has Key-Value pair format and output_mode (string) : The output mode has Key-Value pair format and
Comma separated values format. It should be 'kv' or 'csv'. Comma separated values format. It should be 'kvp' or 'csv'.
config (string) : The profiler options and counters can refer to config (string) : The profiler options and counters can refer to
"Compute Command Line Profiler User Guide". "Compute Command Line Profiler User Guide".
""" """
if output_mode is None: if output_mode is None:
output_mode = 'csv' output_mode = 'csv'
if output_mode not in ['kv', 'csv']: if output_mode not in ['kvp', 'csv']:
raise ValueError("The output mode must be 'key-value' or 'csv'.") raise ValueError("The output mode must be 'kvp' or 'csv'.")
config = NVPROF_CONFIG if config is None else config config = NVPROF_CONFIG if config is None else config
core.nvprof_init(output_file, output_mode, config) core.nvprof_init(output_file, output_mode, config)
# Enables profiler collection by the active CUDA profiling tool. # Enables profiler collection by the active CUDA profiling tool.
......
...@@ -20,7 +20,7 @@ class TestProfiler(unittest.TestCase): ...@@ -20,7 +20,7 @@ class TestProfiler(unittest.TestCase):
with profiler.cuda_profiler("cuda_profiler.txt", 'csv') as nvprof: with profiler.cuda_profiler("cuda_profiler.txt", 'csv') as nvprof:
for i in range(epoc): for i in range(epoc):
input = np.random.random(dshape).astype("float32") input = np.random.random(dshape).astype('float32')
exe.run(fluid.default_main_program(), feed={'data': input}) exe.run(fluid.default_main_program(), feed={'data': input})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册