experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cupti tracer should check cupti interface is nullptr or not.

Browse Source 
PiperOrigin-RevId: 264419630
This commit is contained in:
A. Unique TensorFlower 2019-08-20 10:56:38 -07:00 committed by TensorFlower Gardener
parent a3fb4cd4ea
commit 9c6b1d6898
2 changed files with 5 additions and 4 deletions
tensorflow/core/profiler/internal/gpu
cupti_tracer.cccupti_tracer.h

5
tensorflow/core/profiler/internal/gpu/cupti_tracer.cc
View File

@ -590,7 +590,7 @@ int CuptiTracer::NumGpus() {
if (cuDeviceGetCount(&gpu_count) != CUDA_SUCCESS) {
return 0;
}
LOG(INFO) << "xprof found " << gpu_count << " GPUs";
LOG(INFO) << "Profiler found " << gpu_count << " GPUs";
return gpu_count;
}();
return num_gpus;
@ -709,7 +709,8 @@ Status CuptiTracer::DisableActivityTracing() {
uint64 CuptiTracer::GetTimestamp() {
uint64_t tsc;
if (cupti_interface_->GetTimestamp(&tsc) == CUPTI_SUCCESS) {
if (cupti_interface_ &&
cupti_interface_->GetTimestamp(&tsc) == CUPTI_SUCCESS) {
return tsc;
}
// Return 0 on error. If an activity timestamp is 0, the activity will be

4
tensorflow/core/profiler/internal/gpu/cupti_tracer.h
View File

@ -219,8 +219,8 @@ class CuptiTracer {
void ConfigureActivityUnifiedMemoryCounter(bool enable);
absl::optional<CuptiTracerOptions> option_;
CuptiInterface* cupti_interface_;
CuptiTraceCollector* collector_;
CuptiInterface* cupti_interface_ = nullptr;
CuptiTraceCollector* collector_ = nullptr;
absl::optional<AnnotationMap> annotation_map_;
bool api_tracing_enabled_ = false;