fhelium.backend.memory.operations

Registered Tensor execution for value placement transfers.

TorchMemoryTransferImplementation

class View source

TorchMemoryTransferImplementation(name: str = 'torch-memory-transfer', operation_types: tuple[type[memory.TransferOp], ...] = (memory.TransferOp,), supports_in_place: bool = False)

Move one Tensor payload to a caller-bound device and memory space.

Attributes

Name	Type	Default/value
name	str	'torch-memory-transfer'
operation_types	tuple[type[memory.TransferOp], ...]	(memory.TransferOp,)
supports_in_place	bool	False

resource_requirements

method

def resource_requirements(invocation: OperationInvocation) -> tuple[ResourceRequirement, ...]: ...

execute

method

def execute(invocation: OperationInvocation, inputs: tuple[torch.Tensor, ...], resources: tuple[BoundResource, ...], /, *, in_place: bool) -> tuple[torch.Tensor, ...]: ...