Skip to content

cuda out of memory error for large lookup tables #10

@giancarlo-pereira

Description

@giancarlo-pereira

The codebase currently handles use_gpu = True by sending the whole lookup table (LUT) to the GPU when first loaded. This will not work for large LUTs.

Additionally, operations like LUT = lut[mask] need memory allocation as it generates a copy instead of a view.

Possible options:

  • substitute masking for slicing or other ways to create view of array
  • keep everything in CPU until blockLookUp is called -- the function would then need a use_gpu boolean argument
  • dask for distributed arrays?

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions