MCPcopy Create free account
hub / github.com/THUDM/GLM / backward

Method backward

mpu/mappings.py:87–88  ·  view source on GitHub ↗
(ctx, grad_output)

Source from the content-addressed store, hash-verified

85
86 @staticmethod
87 def backward(ctx, grad_output):
88 return _reduce(grad_output)
89
90
91class _ReduceFromModelParallelRegion(torch.autograd.Function):

Callers

nothing calls this directly

Calls 1

_reduceFunction · 0.85

Tested by

no test coverage detected