Python chunked_bcast_obj示例

示例#1

文件： mpi_communicator_base.py 项目： etckanikama/Congnition-for-OpenCV

 def bcast_obj(self, obj, max_buf_len=256 * 1024 * 1024, root=0):
     if self._check_obj_type_for_chainerx(obj):
         raise ValueError('calling bcast_obj on chainerx \
             with cuda is not supported')
     return chunked_bcast_obj(obj,
                              self.mpi_comm,
                              max_buf_len=max_buf_len,
                              root=root)

示例#2

文件： test_communication_utility.py 项目： asi1024/chainer

    def check_chunked_bcast_obj(self, data_size, max_buf_len):
        root = 0
        obj = np.arange(data_size)
        src = None
        if self.communicator.rank == root:
            src = obj

        dst = chunked_bcast_obj(src, self.communicator.mpi_comm,
                                max_buf_len, root)
        assert len(dst) == len(obj)
        for i in range(len(obj)):
            assert dst[i] == obj[i]

示例#3

    def check_chunked_bcast_obj(self, data_size, max_buf_len):
        root = 0
        obj = np.arange(data_size)
        src = None
        if self.communicator.rank == root:
            src = obj

        dst = chunked_bcast_obj(src, self.communicator.mpi_comm, max_buf_len,
                                root)
        assert len(dst) == len(obj)
        for i in range(len(obj)):
            assert dst[i] == obj[i]

示例#4

文件： mpi_communicator_base.py 项目： vermashresth/chainer

 def bcast_obj(self, obj, max_buf_len=256 * 1024 * 1024, root=0):
     return chunked_bcast_obj(obj, self.mpi_comm,
                              max_buf_len=max_buf_len,
                              root=root)

示例#5

文件： mpi_communicator_base.py 项目： jnishi/chainer

 def bcast_obj(self, obj, max_buf_len=256 * 1024 * 1024, root=0):
     return chunked_bcast_obj(obj, self.mpi_comm,
                              max_buf_len=max_buf_len,
                              root=root)