diff --git a/libkineto/src/CuptiActivityApi.cpp b/libkineto/src/CuptiActivityApi.cpp index 58df6a9ee..098c2e48c 100644 --- a/libkineto/src/CuptiActivityApi.cpp +++ b/libkineto/src/CuptiActivityApi.cpp @@ -164,11 +164,12 @@ void CuptiActivityApi::bufferRequested( size_t* maxNumRecords) { std::lock_guard guard(mutex_); LOG(VERBOSE) << "CUPTI buffer requested"; - if (allocatedGpuTraceBuffers_.size() >= maxGpuBufferCount_) { - stopCollection = true; + if (allocatedGpuTraceBuffers_.size() >= 1) { LOG(WARNING) << "Exceeded max GPU buffer count (" - << allocatedGpuTraceBuffers_.size() - << " >= " << maxGpuBufferCount_ << ") - terminating tracing"; + << allocatedGpuTraceBuffers_.size() << " >= " << 1 + << ") - denying request "; + *buffer = NULL; + return; } auto buf = std::make_unique(kBufSize);