AlphaGitHub
Source/Packages

Runtime.NativeDataParallelOwned

packages/execution/src/Runtime/NativeDataParallelOwned.alpha

2,874 lines126 declarations142.2 KiBSHA-256 de18258423c4
Download exact fileView at revisionbcc11a4e7647d31628

def · lines 1420–1437

dpoRound

Full file
1420def dpoRound =
1421  (lambda unrestricted owners : Nat .
1422    (lambda unrestricted p : Nat .
1423      (lambda unrestricted sum : (family GlobalProtocol (family OrderedCombineOperation (family RegionTransferIdentity)) (family RegionTransferIdentity) (family ProtocolNoDecision)) .
1424        (lambda unrestricted assemble : (family GlobalProtocol (family OrderedCombineOperation (family RegionTransferIdentity)) (family RegionTransferIdentity) (family ProtocolNoDecision)) .
1425          (lambda unrestricted gradients : (family StdList (family GlobalProtocol (family OrderedCombineOperation (family RegionTransferIdentity)) (family RegionTransferIdentity) (family ProtocolNoDecision))) .
1426            (nat-eliminate
1427              (lambda unrestricted current : Nat .
1428                (family
1429                  GlobalProtocol
1430                  (family DataParallelOwnedOperation)
1431                  (family RegionTransferIdentity)
1432                  (family ProtocolNoDecision)))
1433              (dpoReplicatedRound owners sum assemble gradients)
1434              (lambda unrestricted previous : Nat .
1435                (lambda unrestricted ignored : (family GlobalProtocol (family DataParallelOwnedOperation) (family RegionTransferIdentity) (family ProtocolNoDecision)) .
1436                  (dpoPartitionedRound owners sum assemble gradients)))
1437              p))))))

The compiler supplied declaration spans and resolved links from this source snapshot. This page does not assert that this file belongs to a checked closure.