Method debug_copy_from

python/tvm/runtime/disco/session.py:58–75

Copy an Tensor value to remote for debugging purposes. Parameters ---------- worker_id : int The id of the worker to be copied to. value : Union[numpy.ndarray, Tensor] The value to be copied.

(
        self,
        worker_id: int,
        value: np.ndarray | Tensor,
    )

        return _ffi_api.DRefDebugGetFromRemote(self, worker_id)  # type: ignore # pylint: disable=no-member

    def debug_copy_from(
        self,
        worker_id: int,
        value: np.ndarray | Tensor,
    ) -> None:
        """Copy an Tensor value to remote for debugging purposes.

        Parameters
        ----------
        worker_id : int
            The id of the worker to be copied to.

        value : Union[numpy.ndarray, Tensor]
            The value to be copied.
        """
        if not isinstance(value, Tensor):
            value = _as_Tensor(value)
        return _ffi_api.DRefDebugCopyFrom(self, worker_id, value)  # type: ignore # pylint: disable=no-member


class DPackedFunc(DRef):