Skip to content

Commit ec83b41

Browse files
committed
Add method for get_total_time
1 parent 9e5bf65 commit ec83b41

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/Recipes/cuda_api_gpu_sum.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,5 +124,11 @@ function get_total_time(summary, category::Category)
124124
total_time = sum(item.total_time for item in _iter(filtered_summary)) # Works for `DataFrame`
125125
return total_time
126126
end
127+
function get_total_time(summary, categories::AbstractArray{Category})
128+
return sum(categories) do category
129+
filtered_summary = filter(item -> item.category == category, summary)
130+
sum(item.total_time for item in _iter(filtered_summary))
131+
end
132+
end
127133
_iter(array::AbstractArray) = array
128134
_iter(dataframe) = eachrow(dataframe)

0 commit comments

Comments
 (0)