Skip to content

Add Helper to Create NumPy Files#1645

Merged
rapids-bot[bot] merged 8 commits intorapidsai:mainfrom
julianmi:ace-improve-numpy-headers
Jan 15, 2026
Merged

Add Helper to Create NumPy Files#1645
rapids-bot[bot] merged 8 commits intorapidsai:mainfrom
julianmi:ace-improve-numpy-headers

Conversation

@julianmi
Copy link
Copy Markdown
Contributor

This PR introduces create_numpy_file in the file_io.hpp header. It creates NumPy files with pre-allocated space and writes the appropriate header based on the specified shape and data type. This removes duplicated code in CAGRA ACE in cagra_build.cuh. It addresses one follow-up item of #1486.

CC @tfeher

- Introduces `create_numpy_file` in the file_io.hpp header.
- The function creates numpy files with pre-allocated space and writes the appropriate header based on the specified shape and data type.
- Removes duplicated code.
- Partly addresses rapidsai#1486.
@cjnolet cjnolet moved this from Todo to In Progress in Unstructured Data Processing Jan 5, 2026
@cjnolet cjnolet added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Jan 5, 2026
Copy link
Copy Markdown
Contributor

@tfeher tfeher left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks Julian for simplifying the code. LGTM.

@tfeher
Copy link
Copy Markdown
Contributor

tfeher commented Jan 15, 2026

/merge

@rapids-bot rapids-bot Bot merged commit 1a0a1d4 into rapidsai:main Jan 15, 2026
99 checks passed
@github-project-automation github-project-automation Bot moved this from In Progress to Done in Unstructured Data Processing Jan 15, 2026
lowener pushed a commit to lowener/cuvs that referenced this pull request Jan 26, 2026
This PR introduces `create_numpy_file` in the file_io.hpp header. It creates NumPy files with pre-allocated space and writes the appropriate header based on the specified shape and data type. This removes duplicated code in CAGRA ACE in `cagra_build.cuh`. It addresses one follow-up item of rapidsai#1486.

CC @tfeher

Authors:
  - Julian Miller (https://github.com/julianmi)

Approvers:
  - Tamas Bela Feher (https://github.com/tfeher)

URL: rapidsai#1645
@julianmi julianmi deleted the ace-improve-numpy-headers branch April 27, 2026 08:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants