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

set active trace to off no matter cupti function return.

Browse Source 
PiperOrigin-RevId: 263788832
This commit is contained in:
A. Unique TensorFlower 2019-08-16 09:45:54 -07:00 committed by TensorFlower Gardener
parent a913689fdd
commit 79a430dce9
1 changed files with 2 additions and 2 deletions

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

@ -644,6 +644,8 @@ Status CuptiTracer::EnableApiTracing() {
Status CuptiTracer::DisableApiTracing() {
if (!api_tracing_enabled_) return Status::OK();
api_tracing_enabled_ = false;
if (!option_->cbids_selected.empty()) {
for (auto cbid : option_->cbids_selected) {
RETURN_IF_CUPTI_ERROR(cupti_interface_->EnableCallback(
@ -656,8 +658,6 @@ Status CuptiTracer::DisableApiTracing() {
VLOG(1) << "Disable subscriber";
RETURN_IF_CUPTI_ERROR(cupti_interface_->Unsubscribe(subscriber_));
api_tracing_enabled_ = false;
return Status::OK();
}