Method load_vm_module

python/tvm/runtime/disco/session.py:302–323

Load a VM module from a file. Parameters ---------- path : str The path to the VM module file. device : Optional[Device] = None The device to load the VM module to. Default to the default device of each worker. Returns ------

(
        self,
        path: str,
        device: Device | None = None,
    )

        return remote_array

    def load_vm_module(
        self,
        path: str,
        device: Device | None = None,
    ) -> DModule:
        """Load a VM module from a file.

        Parameters
        ----------
        path : str
            The path to the VM module file.

        device : Optional[Device] = None
            The device to load the VM module to. Default to the default device of each worker.

        Returns
        -------
        module : DModule
            The loaded VM module.
        """
        func = self._get_cached_method("runtime.disco.load_vm_module")
        return DModule(func(path, device), self)

    def init_ccl(self, ccl: str, *device_ids):
        """Initialize the underlying communication collective library.