rapidsai / rmm

RAPIDS Memory Manager
https://docs.rapids.ai/api/rmm/stable/
Apache License 2.0
478 stars 195 forks source link

Add a stack to the statistics resource #1563

Closed madsbk closed 4 months ago

madsbk commented 4 months ago

Description

In order to enable more fine-grained statistics, this PR adds a stack to statistics_resource_adaptor

Closes https://github.com/rapidsai/rmm/issues/1332

Checklist

madsbk commented 4 months ago

@harrism and @wence-, the PR is read for a new round of reviews.

I have added a "profiler" decorator that make it easy to profile specific functions. cudf can use it like this: https://github.com/rapidsai/cudf/pull/15866

wence- commented 4 months ago

/merge