Fix nil span check

DataDog · Oct 16, 2024 · 119e009 · 119e009
1 parent 3f37339
commit 119e009
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pkg/gpu/stream.go b/pkg/gpu/stream.go
@@ -227,6 +227,7 @@ func (sh *StreamHandler) getCurrentData(now uint64) *model.StreamData {
 	span := sh.getCurrentKernelSpan(now)
 	if span != nil {
 		data.Spans = append(data.Spans, span)
+		data.Allocations = append(data.Allocations, getAssociatedAllocations(span)...)
 	}
 
 	for _, alloc := range sh.memAllocEvents {
@@ -238,8 +239,6 @@ func (sh *StreamHandler) getCurrentData(now uint64) *model.StreamData {
 		})
 	}
 
-	data.Allocations = append(data.Allocations, getAssociatedAllocations(data.Spans[0])...)
-
 	return data
 }