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

Method backward

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

Source from the content-addressed store, hash-verified

124
125 @staticmethod
126 def backward(ctx, grad_output):
127 return _gather(grad_output)
128
129
130class _GatherFromModelParallelRegion(torch.autograd.Function):

Callers

nothing calls this directly

Calls 1

_gatherFunction · 0.85

Tested by

no test coverage detected