if no gpu, don't create device tracer.

PiperOrigin-RevId: 303812188 Change-Id: Ie153b573bbcc51435783c3b5671e3035350c91ab
2020-03-30 13:36:15 -07:00 · 2020-03-30 13:36:15 -07:00 · 53778b7e92
commit 53778b7e92
parent a853452aab
1 changed files with 1 additions and 1 deletions
--- 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() {