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

Method backward

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

Source from the content-addressed store, hash-verified

109
110 @staticmethod
111 def backward(ctx, grad_output):
112 return _gather(grad_output)
113
114
115class _GatherFromModelParallelRegion(torch.autograd.Function):

Callers

nothing calls this directly

Calls 1

_gatherFunction · 0.85

Tested by

no test coverage detected