ArrayResponse

class cubie.memory.array_requests.ArrayResponse(arr: dict[str, FakeCUDAArray] = NOTHING, chunks: int = 1, chunk_length: int = 1, chunked_shapes: dict[str, tuple[int, ...]] = NOTHING)[source]

Bases: object

Result of an array allocation containing buffers and chunking data.

Parameters:

  • arr (dict[str, numba.cuda.simulator.cudadrv.devicearray.FakeCUDAArray]) – Dictionary mapping array labels to allocated device arrays.

  • chunks (int) – Number of chunks the allocation was divided into.

  • chunk_length (int) – Length of each chunk along the run axis (except possibly last).

  • chunked_shapes (dict[str, tuple[int, ...]]) – Mapping from array labels to their per-chunk shapes. Empty dict when no chunking occurs.

arr

Dictionary mapping array labels to allocated device arrays.

Type:

dict[str, numba.cuda.simulator.cudadrv.devicearray.FakeCUDAArray]

chunks

Number of chunks the allocation was divided into.

Type:

int

chunk_length

Length of each chunk along the run axis.

Type:

int

chunked_shapes

Mapping from array labels to their per-chunk shapes.

Type:

dict[str, tuple[int, …]]

arr: dict[str, FakeCUDAArray]
chunk_length: int
chunked_shapes: dict[str, tuple[int, ...]]
chunks: int