qimpy.mpi.BufferView
- class BufferView
Bases:
pybind11_object
Expose 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
Construct the object inline, using BufferView(tensor) as an argument to MPI routines.