| 691 | }; |
| 692 | |
| 693 | Status DeviceTracer::CollectData(RunMetadata* run_metadata) { |
| 694 | mutex_lock l(mu_); |
| 695 | if (enabled_) { |
| 696 | return errors::FailedPrecondition("DeviceTracer is still enabled."); |
| 697 | } |
| 698 | |
| 699 | StepStatsCollector step_stats_collector(run_metadata->mutable_step_stats()); |
| 700 | TF_RETURN_IF_ERROR( |
| 701 | CudaEventCollector::Collect(recorder_.get(), &step_stats_collector)); |
| 702 | step_stats_collector.Finalize(); |
| 703 | return Status::OK(); |
| 704 | } |
| 705 | } // namespace |
| 706 | |
| 707 | // Not in anonymous namespace for testing purposes. |