|
9 | 9 | #include "CuptiActivityProfiler.h"
|
10 | 10 | #include <fmt/format.h>
|
11 | 11 | #include <fmt/ranges.h>
|
12 |
| -#include <time.h> |
13 | 12 | #include <atomic>
|
14 | 13 | #include <cstdint>
|
15 | 14 | #include <functional>
|
16 |
| -#include <iomanip> |
17 | 15 | #include <limits>
|
18 | 16 | #include <optional>
|
19 | 17 | #include <string>
|
@@ -504,22 +502,23 @@ static GenericTraceActivity createUserGpuSpan(
|
504 | 502 | }
|
505 | 503 |
|
506 | 504 | void CuptiActivityProfiler::GpuUserEventMap::insertOrExtendEvent(
|
507 |
| - const ITraceActivity& userActivity, |
508 |
| - const ITraceActivity& gpuActivity) { |
509 |
| - StreamKey key(gpuActivity.deviceId(), gpuActivity.resourceId()); |
| 505 | + const ITraceActivity& cpuTraceActivity, |
| 506 | + const ITraceActivity& gpuTraceActivity) { |
| 507 | + StreamKey key(gpuTraceActivity.deviceId(), gpuTraceActivity.resourceId()); |
510 | 508 | CorrelationSpanMap& correlationSpanMap = streamSpanMap_[key];
|
511 |
| - auto it = correlationSpanMap.find(userActivity.correlationId()); |
| 509 | + auto it = correlationSpanMap.find(cpuTraceActivity.correlationId()); |
512 | 510 | if (it == correlationSpanMap.end()) {
|
513 | 511 | auto it_success = correlationSpanMap.insert(
|
514 |
| - {userActivity.correlationId(), |
515 |
| - createUserGpuSpan(userActivity, gpuActivity)}); |
| 512 | + {cpuTraceActivity.correlationId(), |
| 513 | + createUserGpuSpan(cpuTraceActivity, gpuTraceActivity)}); |
516 | 514 | it = it_success.first;
|
517 | 515 | }
|
518 | 516 | GenericTraceActivity& span = it->second;
|
519 |
| - if (gpuActivity.timestamp() < span.startTime || span.startTime == 0) { |
520 |
| - span.startTime = gpuActivity.timestamp(); |
| 517 | + if (gpuTraceActivity.timestamp() < span.startTime || span.startTime == 0) { |
| 518 | + span.startTime = gpuTraceActivity.timestamp(); |
521 | 519 | }
|
522 |
| - int64_t gpu_activity_end = gpuActivity.timestamp() + gpuActivity.duration(); |
| 520 | + int64_t gpu_activity_end = |
| 521 | + gpuTraceActivity.timestamp() + gpuTraceActivity.duration(); |
523 | 522 | if (gpu_activity_end > span.endTime) {
|
524 | 523 | span.endTime = gpu_activity_end;
|
525 | 524 | }
|
|
0 commit comments