Bob.Build.SM86

1module Bob.Build.SM86
2
3import Bob.ArenaPlan
4import Accelerator.SM86.Capability
5import Bob.Learner
6import Bob.System
7import Bob.TrainingRun
8import Model.Config
9import Model.Parameter
10import Model.Word32
11import Model.Word64
12import Representation.BuildPlan
13import Std.Physical
14import Std.Word
15
16family BobSM86BuildBinding : Type 0
17constructor BobSM86BuildBindingValue
18field unrestricted bobBuildSystem : (family BobSystemConfiguration)
19field unrestricted bobBuildRequirements : (family SM86CapabilityRequirements)
20field unrestricted bobBuildCheckpoint : (family BobCheckpointTransportSchema)
21field unrestricted bobBuildForwardRealizationIdentity : Bytes
22field unrestricted bobBuildLossRealizationIdentity : Bytes
23field unrestricted bobBuildOptimizerRealizationIdentity : Bytes
24field unrestricted bobBuildPredictionRealizationIdentity : Bytes
25
26end-family
27
28def bobSM86ProfileIdentity : Bytes =
29  b"bob-sm86-native-v1"
30
31def bobSM86MinimumVideoMemory : ByteCount =
32  (stdByteCount (modelWord64FromNaturalTruncated bobMinimumVideoBytes))
33
34def bobSM86MappedAddressEnd : DeviceAddress =
35  (stdDeviceAddress (modelWord64FromNaturalTruncated bobMappedEnd))
36
37def bobSM86MaximumGridX : GridDimension =
38  (stdGridDimension 256)
39
40def bobSM86CapabilityRequirements : (family SM86CapabilityRequirements) =
41  (record
42    SM86CapabilityRequirements
43    (sm86RequirementComputeMajor = 8)
44    (sm86RequirementComputeMinor = 6)
45    (sm86RequirementAvailableVideoMemoryBytes = (stdByteCountValue bobSM86MinimumVideoMemory))
46    (sm86RequirementMappedAddressEnd = (stdDeviceAddressValue bobSM86MappedAddressEnd))
47    (sm86RequirementMaximumGridX = (stdU32ToNatural (stdGridDimensionValue bobSM86MaximumGridX)))
48    (sm86RequirementUVM = 1)
49    (sm86RequirementResourceManager = 1)
50    (sm86RequirementQMD = 1)
51    (sm86RequirementPushbuffer = 1)
52    (sm86RequirementGPFIFO = 1)
53    (sm86RequirementDeviceTimestamps = 1)
54    (sm86RequirementNoHostFallback = 1))
55
56def bobSM86Admit =
57  (lambda unrestricted observed : (family SM86DeviceCapabilities) .
58    (sm86CapabilityAdmit bobSM86CapabilityRequirements observed))
59
60def bobSM86BuildBinding : (family BobSM86BuildBinding) =
61  (record
62    BobSM86BuildBinding
63    (bobBuildSystem = bobSystem)
64    (bobBuildRequirements = bobSM86CapabilityRequirements)
65    (bobBuildCheckpoint = bobCheckpointSchema)
66    (bobBuildForwardRealizationIdentity = b"sm86-byte-row-gather-f32-v1")
67    (bobBuildLossRealizationIdentity = b"sm86-exact-full-256-cross-entropy-v1")
68    (bobBuildOptimizerRealizationIdentity = b"sm86-adamw-f32-bounded-v1")
69    (bobBuildPredictionRealizationIdentity = b"sm86-greedy-argmax-256-v1"))
70
71def bobSM86BuildPlanEntries =
72  (lambda unrestricted targetIdentity : Bytes .
73    (lambda unrestricted capabilityEvidence : Bytes .
74      (lambda unrestricted finalELF : Bytes .
75        (buildPlanCons
76          b"system"
77          b"bob"
78          (buildPlanCons
79            b"model"
80            b"raw-byte-bigram-256x256-f32"
81            (buildPlanCons
82              b"objective"
83              bobObjectiveIdentity
84              (buildPlanCons
85                b"learner"
86                bobLearnerIdentity
87                (buildPlanCons
88                  b"target"
89                  targetIdentity
90                  (buildPlanCons
91                    b"requirements"
92                    bobSM86ProfileIdentity
93                    (buildPlanCons
94                      b"capability-evidence"
95                      capabilityEvidence
96                      (buildPlanCons
97                        b"schedule"
98                        b"row-gather-ece256-adamw-greedy"
99                        (buildPlanCons
100                          b"memory"
101                          b"bob-sm86-static-arena-16-mib"
102                          (buildPlanCons
103                            b"host-emitter"
104                            b"alpha-x86_64-host-emitter"
105                            (buildPlanCons
106                              b"final-host-elf"
107                              finalELF
108                              (buildPlanCons
109                                b"device-image"
110                                b"embedded-sm86-bob-program-table"
111                                (buildPlanCons
112                                  b"qmd"
113                                  b"embedded-sm86-qmd-table"
114                                  (buildPlanCons
115                                    b"pushbuffer"
116                                    b"embedded-sm86-pushbuffer"
117                                    (buildPlanCons
118                                      b"gpfifo"
119                                      b"embedded-sm86-gpfifo"
120                                      (buildPlanCons
121                                        b"checkpoint"
122                                        b"bob-raw-pmv-f32-v1"
123                                        (buildPlanCons
124                                        b"request-protocol"
125                                        b"alpha-text-artifact-request-v1"
126                                        (buildPlanCons
127                                        b"evidence"
128                                        b"r6-qualification-pending"
129                                        (buildPlanCons
130                                        b"physical-lane"
131                                        b"r6-qualification-pending"
132                                        buildPlanEmpty))))))))))))))))))))))