NVIDIA / NVIDIA/cudf

[BUG] NVTX ranges not being closed in Frame._get_columns_by_label?

Open
#13,487 2 comments 0 reactions 0 assignees View on GitHub
0 - Backlog bug Python
Dominant language
C++
Stars
9.8k
Forks
1.1k
Avg merge
3d 6m
Merged PRs (30d)
278

Description

**Describe the bug**
When running the following Python script, it seems that the NVTX ranges aren't being closed. See screenshot below.

```python
import cudf
from cudf.testing._utils import assert_eq

if __name__ == "__main__":
for i in range(3):
gdf = cudf.DataFrame({"a": range(10)})
pdf = gdf.to_pandas()
assert_eq(gdf, pdf)
```

Then execute `nsys profile python nvtx_bug.py`.

![image](https://github.com/rapidsai/cudf/assets/3943761/750120c7-b800-4b8d-9af7-174adc93f7f6)

I didn't see this problem with `cudf.Series`, only `cudf.DataFrame`.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.