Enhance nvtx support. (#5636)

2020-05-06 22:54:24 +08:00
parent 67d267f9da
commit eaf2a00b5c
13 changed files with 98 additions and 87 deletions
--- a/src/tree/gpu_hist/gradient_based_sampler.cu
+++ b/src/tree/gpu_hist/gradient_based_sampler.cu
@@ -354,9 +354,9 @@ GradientBasedSampler::GradientBasedSampler(EllpackPageImpl* page,
 // Sample a DMatrix based on the given gradient pairs.
 GradientBasedSample GradientBasedSampler::Sample(common::Span<GradientPair> gpair,
                                                 DMatrix* dmat) {
-  monitor_.StartCuda("Sample");
+  monitor_.Start("Sample");
  GradientBasedSample sample = strategy_->Sample(gpair, dmat);
-  monitor_.StopCuda("Sample");
+  monitor_.Stop("Sample");
  return sample;
 }