qimpy.mpi.BufferView
- class BufferView
Bases:
pybind11_objectExpose buffer protocol on torch.Tensor for mpi4py.
- __init__(self, tensor: torch.Tensor)
Construct the object inline, using BufferView(tensor) as an argument to MPI routines. Do not retain variables of type BufferView.
Methods
__init__(self, tensor: torch.Tensor) Construct the object inline, using BufferView(tensor) as an argument to MPI routines.