From 111fa339f641e45128342bb8b7143776f77a3b54 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 30 Mar 2020 13:36:15 -0700 Subject: [PATCH] if no gpu, don't create device tracer. PiperOrigin-RevId: 303812188 Change-Id: Ie153b573bbcc51435783c3b5671e3035350c91ab --- tensorflow/core/profiler/internal/gpu/cupti_tracer.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/core/profiler/internal/gpu/cupti_tracer.cc b/tensorflow/core/profiler/internal/gpu/cupti_tracer.cc index 3ce6678de01..e001d831b65 100644 --- a/tensorflow/core/profiler/internal/gpu/cupti_tracer.cc +++ b/tensorflow/core/profiler/internal/gpu/cupti_tracer.cc @@ -1348,7 +1348,7 @@ absl::string_view AnnotationMap::LookUp(uint32 device_id, } bool CuptiTracer::IsAvailable() const { - return !activity_tracing_enabled_ && !api_tracing_enabled_; + return NumGpus() && !activity_tracing_enabled_ && !api_tracing_enabled_; } int CuptiTracer::NumGpus() {