| 397 | } |
| 398 | |
| 399 | Status DeviceTracer::Stop() { |
| 400 | VLOG(1) << "DeviceTracer::Stop"; |
| 401 | mutex_lock l(mu_); |
| 402 | if (!enabled_) { |
| 403 | return Status::OK(); |
| 404 | } |
| 405 | cupti_hook_.reset(); |
| 406 | tracing::ScopedAnnotation::Enable(false); |
| 407 | |
| 408 | enabled_ = false; |
| 409 | return Status::OK(); |
| 410 | } |
| 411 | |
| 412 | class CudaEventCollector { |
| 413 | struct DeviceInfo { |