From 340009d6a49061e5ad0f3845f92583bff5fad7fe Mon Sep 17 00:00:00 2001 From: Yuanle Liu Date: Fri, 21 Oct 2022 13:03:41 +0800 Subject: [PATCH] fix nvprof_nvtx_push interface bug (#47232) * fix nvprof_nvtx_push interface bug --- paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h | 2 +- paddle/fluid/pybind/pybind.cc | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h b/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h index 193e08bdde5..555a83a0210 100644 --- a/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h +++ b/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h @@ -42,7 +42,7 @@ enum class NvtxRangeColor : uint32_t { }; void CudaNvtxRangePush(const std::string& name, - const NvtxRangeColor color = NvtxRangeColor::Blue); + const NvtxRangeColor color = NvtxRangeColor::Green); void CudaNvtxRangePop(); #endif diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc index c37839a7c65..c5898b97624 100644 --- a/paddle/fluid/pybind/pybind.cc +++ b/paddle/fluid/pybind/pybind.cc @@ -2063,7 +2063,9 @@ All parameter, weight, gradient are variables in Paddle. m.def("nvprof_init", platform::CudaProfilerInit); m.def("nvprof_start", platform::CudaProfilerStart); m.def("nvprof_stop", platform::CudaProfilerStop); - m.def("nvprof_nvtx_push", platform::CudaNvtxRangePush); + m.def("nvprof_nvtx_push", [](const std::string &name) { + platform::CudaNvtxRangePush(name, platform::NvtxRangeColor::Green); + }); m.def("nvprof_nvtx_pop", platform::CudaNvtxRangePop); m.def("nvprof_enable_record_event", platform::NvprofEnableRecordEvent); m.def("nvprof_disable_record_event", platform::NvprofDisableRecordEvent); -- GitLab