legate.core.task.InputStore#
- class legate.core.task.InputStore#
Convenience class for specifying input stores for Legate task variants.
This class can be used as a type annotation in order to mark parameters as inputs that should be taken from
TaskContext.inputswhen the task code is invoked:def task_function(in: InputStore, out: OutputStore) -> None
- __init__(*args, **kwargs)#
Methods
__dlpack__(self, *, stream, ...[, copy])Exports the store for consumption by from_dlpack() as a DLPack capsule.
__dlpack_device__(self)Returns device type and device ID in DLPack format.
__init__(*args, **kwargs)bind_data(self, TaskLocalBuffer buffer[, extent])Binds a buffer to the store.
create_output_buffer(self, shape, bool bind=True)Create a buffer for the store to potentially bind to later on.
get_inline_allocation(self)Get the InlineAllocation for this store.
Attributes
__array_interface__dict[str, Any]
__cuda_array_interface__dict[str, Any]
__doc__domainDomain_t
ndimint32_t
targetStoreTarget
typeType