megengine.distributed.Client.set_is_grad

Client.set_is_grad(key, is_grad)

用 key 来标记 send/recv 是需要梯度的。

参数

• key – key to match send/recv op.

• is_grad – whether this op need grad.

上一页

megengine.distributed.Client.group_barrier

下一页

megengine.distributed.Client.set_remote_tracer