Skip to content
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

Non-uniform structors of arrays in numba #158

Open
jpmorgan98 opened this issue Feb 13, 2024 · 2 comments
Open

Non-uniform structors of arrays in numba #158

jpmorgan98 opened this issue Feb 13, 2024 · 2 comments
Assignees
Labels
enhancement New feature or request

Comments

@jpmorgan98
Copy link
Collaborator

Optimizations for how cross section data, tally meshes, and cells are stored. Right now with the np.ndarry the size of allocated memory for each item is that of the largest array and the same for all others. numba.jitclass is a remedy for this but is not GPU operable. Some initial ideas are

  • numba.jitclass with a puller function for GPU runs to nd.arrays
  • An offset scheme where data is stored as a single dim array with offsets
  • Use of other c-type data structures like pytorch or cupy
@jpmorgan98 jpmorgan98 added the enhancement New feature or request label Feb 13, 2024
@jpmorgan98
Copy link
Collaborator Author

@ilhamv how urgent is you're need to move towards jitclass?

@ilhamv
Copy link
Member

ilhamv commented Feb 28, 2024

@jpmorgan98 Per our discussion with Braxton yesterday, it sounds like it is easier for the GPU-Numba work, if we stick with the numpy structured array than if we move to use Jitclass, at least for now, perhaps until the first working GPU merge. But what do you think, @braxtoncuneo ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants