Skip to content

Migrate CollectTimeIterator to use new NvtxIdWithMetrics class#12900

Merged
zpuller merged 11 commits into
NVIDIA:branch-25.12from
zpuller:nvtx_doc
Oct 13, 2025
Merged

Migrate CollectTimeIterator to use new NvtxIdWithMetrics class#12900
zpuller merged 11 commits into
NVIDIA:branch-25.12from
zpuller:nvtx_doc

Conversation

@zpuller

@zpuller zpuller commented Jun 9, 2025

Copy link
Copy Markdown
Collaborator

We are in the process of migrating all of our nvtx ranges to use the new documented Nvtx api. Many of the existing ranges are already wrapped in other classes such as the NvtxWithMetrics class. This pr establishes a replacement for NvtxWithMetrics called NvtxIdWithMetrics which provides the same wrapping functionality, but using the documented Nvtx class as input. Then, the PR begins migrating some of the many uses of this wrapper, namely the ranges that go through the CollectTimeIterator

zpuller added 4 commits June 9, 2025 10:13
Signed-off-by: Zach Puller <zpuller@nvidia.com>
Signed-off-by: Zach Puller <zpuller@nvidia.com>
Signed-off-by: Zach Puller <zpuller@nvidia.com>
Signed-off-by: Zach Puller <zpuller@nvidia.com>
Signed-off-by: Zach Puller <zpuller@nvidia.com>
@zpuller zpuller requested a review from gerashegalov June 10, 2025 16:35
@zpuller zpuller requested a review from abellina June 24, 2025 17:25
@sameerz sameerz added the task Work required that improves the product but is not user facing label Jul 28, 2025
@zpuller zpuller changed the base branch from branch-25.08 to branch-25.10 July 29, 2025 17:41
Signed-off-by: Zach Puller <zpuller@nvidia.com>
@zpuller zpuller changed the base branch from branch-25.10 to branch-25.12 September 29, 2025 16:35
Signed-off-by: Zach Puller <zpuller@nvidia.com>
@zpuller

zpuller commented Oct 13, 2025

Copy link
Copy Markdown
Collaborator Author

build

Comment thread sql-plugin/src/main/scala/com/nvidia/spark/rapids/NvtxRangeWithDoc.scala Outdated
Comment thread sql-plugin/src/main/scala/com/nvidia/spark/rapids/NvtxRangeWithDoc.scala Outdated
Signed-off-by: Zach Puller <zpuller@nvidia.com>
@zpuller zpuller requested a review from abellina October 13, 2025 18:19
Comment thread sql-plugin/src/main/scala/com/nvidia/spark/rapids/NvtxRangeWithDoc.scala Outdated
Comment thread sql-plugin/src/main/scala/com/nvidia/spark/rapids/NvtxRangeWithDoc.scala Outdated
Signed-off-by: Zach Puller <zpuller@nvidia.com>
@zpuller zpuller requested a review from abellina October 13, 2025 18:32
@zpuller

zpuller commented Oct 13, 2025

Copy link
Copy Markdown
Collaborator Author

build

@zpuller zpuller merged commit c4367bd into NVIDIA:branch-25.12 Oct 13, 2025
59 of 60 checks passed
@zpuller zpuller deleted the nvtx_doc branch October 13, 2025 21:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

task Work required that improves the product but is not user facing

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants