MCPcopy Create free account
hub / github.com/zai-org/CodeGeeX / backward

Method backward

codegeex/megatron/mpu/mappings.py:94–95  ·  view source on GitHub ↗
(ctx, grad_output)

Source from the content-addressed store, hash-verified

92
93 @staticmethod
94 def backward(ctx, grad_output):
95 return _reduce(grad_output)
96
97
98class _ReduceFromModelParallelRegion(torch.autograd.Function):

Callers

nothing calls this directly

Calls 1

_reduceFunction · 0.85

Tested by

no test coverage detected