Skip to content

Remove _helperlib dep., implement attempt_nocopy_reshape in python for CUDA-specific changes #383

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

VijayKandiah
Copy link
Contributor

We are only using the method attempt_nocopy_reshape from numba's _helperlib C extension. This PR introduces a python implementation of attempt_nocopy_reshape for CUDA-specific customizations and to remove dependency on _helperlib.

@VijayKandiah VijayKandiah self-assigned this Aug 6, 2025
@VijayKandiah VijayKandiah added the 3 - Ready for Review Ready for review by team label Aug 6, 2025
Copy link

copy-pr-bot bot commented Aug 6, 2025

Auto-sync is disabled for ready for review pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@VijayKandiah VijayKandiah requested a review from gmarkall August 6, 2025 22:40
Copy link
Contributor

@gmarkall gmarkall left a comment

Choose a reason for hiding this comment

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

A quick note (not got really round to reviewing this yet) - numba_attempt_nocopy_reshape is already implemented in reshape_funcs.cu. There may be some merit to having the implementation in Python, but I think there is also the risk of it being less efficient in terms of register usage as I would guess it would use 64-bit ints in places where the .cu version uses int types.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
3 - Ready for Review Ready for review by team
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants