0ebd265e01054e2d4d2b53eda8e4f6282a054f5b,chainermn/communicators/mpi_communicator_base.py,MpiCommunicatorBase,_multi_node_mean,#MpiCommunicatorBase#Any#Any#,680
Before Change
xp = chainer.backend.get_array_module(recvbuf)
// chainerx is planning to support copyto
// https://github.com/chainer/chainer/pull/7521
xp.copyto(recvbuf, array_b32.astype(numpy.float16), casting="no")
recvbuf *= 1.0 / self.mpi_comm.size
if chainer.is_debug():
After Change
self.mpi_comm.Allreduce(buffer_a, buffer_b)
if is_float16:
recvbuf[...] = array_b32.astype(numpy.float16)
recvbuf *= 1.0 / self.mpi_comm.size
if chainer.is_debug():
In pattern: SUPERPATTERN
Frequency: 3
Non-data size: 3
Instances
Project Name: chainer/chainer
Commit Name: 0ebd265e01054e2d4d2b53eda8e4f6282a054f5b
Time: 2019-10-22
Author: tianqi@preferred.jp
File Name: chainermn/communicators/mpi_communicator_base.py
Class Name: MpiCommunicatorBase
Method Name: _multi_node_mean
Project Name: SheffieldML/GPy
Commit Name: 4eac8a59655738b05cbc716a0834d3d94dfddd8c
Time: 2014-05-09
Author: ibinbei@gmail.com
File Name: GPy/core/parameterization/param.py
Class Name: ParamConcatenation
Method Name: __setitem__
Project Name: chainer/chainer
Commit Name: b85e17830cdbefd3bf70794b27274e9598c2ab74
Time: 2019-09-10
Author: nc.jky0@gmail.com
File Name: chainer/initializers/orthogonal.py
Class Name: Orthogonal
Method Name: __call__