NVIDIA / nvtx-plugins

Python bindings for NVTX
https://docs.nvidia.com/deeplearning/frameworks/nvtx-plugins/user-guide/docs/en/stable/
Apache License 2.0
66 stars 15 forks source link

Avoid space in NVTXHook session hook's range message #6

Open shawnwang18 opened 4 years ago

shawnwang18 commented 4 years ago

When using nsight-system range capture feature together with nvtx-plugins, range name with blank can no be processed normally by nsight-system, so would better avoid space in range message.

related code: https://github.com/NVIDIA/nvtx-plugins/blob/master/nvtx_plugins/python/nvtx/plugins/tf/estimator.py#L36