diff options
Diffstat (limited to 'tensorflow/core/platform/default/gpu_tracer.cc')
-rw-r--r-- | tensorflow/core/platform/default/gpu_tracer.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/platform/default/gpu_tracer.cc b/tensorflow/core/platform/default/gpu_tracer.cc index 50c27b3cf6..3f85546127 100644 --- a/tensorflow/core/platform/default/gpu_tracer.cc +++ b/tensorflow/core/platform/default/gpu_tracer.cc @@ -579,8 +579,8 @@ Status GPUTracerImpl::Collect(StepStatsCollector *collector) { // TODO(pbar) Handle device IDs and prefix properly. const string prefix = ""; const int id = 0; - const string stream_device = strings::StrCat(prefix, "/gpu:", id, "/stream:"); - const string memcpy_device = strings::StrCat(prefix, "/gpu:", id, "/memcpy"); + const string stream_device = strings::StrCat(prefix, "/device:GPU:", id, "/stream:"); + const string memcpy_device = strings::StrCat(prefix, "/device:GPU:", id, "/memcpy"); mutex_lock l2(trace_mu_); for (const auto &rec : kernel_records_) { |