Source/Packages

Realization.Nvidia.SM86.EmbeddingGatherF32SM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/EmbeddingGatherF32SM86.alpha

681 lines108 declarations28.0 KiBSHA-256 d1c5e905f898

def · lines 276–348

embeddingGatherF32AddressProgramFor

Full file
276def embeddingGatherF32AddressProgramFor =
277  (lambda unrestricted width : (family EmbeddingGatherF32Width) .
278    (embeddingGatherF32Append
279      (embeddingGatherF32Instruction
280        (constructor
281          SM86InstructionBody
282          SM86IntegerMultiplyAddImmediate
283          (embeddingGatherF32Register (byte 6))
284          (embeddingGatherF32Register (byte 0))
285          embeddingGatherF32FourImmediate
286          sm86ZeroRegister
287          sm86SafeControl))
288      (embeddingGatherF32Append
289        (embeddingGatherF32Instruction
290          (constructor
291            SM86InstructionBody
292            SM86IntegerMultiplyAddWideConstant
293            (embeddingGatherF32Register (byte 8))
294            (embeddingGatherF32Register (byte 4))
295            (embeddingGatherF32Register (byte 3))
296            (byte 0)
297            embeddingGatherF32TokenIdsPointer
298            sm86SafeControl))
299        (embeddingGatherF32Append
300          (embeddingGatherF32Instruction
301            (constructor
302              SM86InstructionBody
303              SM86LoadGlobal
304              (embeddingGatherF32Register (byte 10))
305              (embeddingGatherF32Register (byte 8))
306              embeddingGatherF32ZeroImmediate
307              embeddingGatherF32SetBarrier0))
308          (embeddingGatherF32Append
309            (embeddingGatherF32Instruction
310              (constructor
311                SM86InstructionBody
312                SM86IntegerMultiplyAddImmediate
313                (embeddingGatherF32Register (byte 18))
314                (embeddingGatherF32Register (byte 4))
315                (embeddingGatherF32WidthImmediateFor width)
316                (embeddingGatherF32Register (byte 6))
317                sm86SafeControl))
318            (embeddingGatherF32Append
319              (embeddingGatherF32Instruction
320                (constructor
321                  SM86InstructionBody
322                  SM86IntegerMultiplyAddWideConstant
323                  (embeddingGatherF32Register (byte 20))
324                  (embeddingGatherF32Register (byte 18))
325                  (embeddingGatherF32Register (byte 3))
326                  (byte 0)
327                  embeddingGatherF32OutputPointer
328                  sm86SafeControl))
329              (embeddingGatherF32Append
330                (embeddingGatherF32Instruction
331                  (constructor
332                    SM86InstructionBody
333                    SM86IntegerMultiplyAddImmediate
334                    (embeddingGatherF32Register (byte 12))
335                    (embeddingGatherF32Register (byte 10))
336                    (embeddingGatherF32WidthImmediateFor width)
337                    (embeddingGatherF32Register (byte 6))
338                    embeddingGatherF32WaitToken))
339                (embeddingGatherF32Instruction
340                  (constructor
341                    SM86InstructionBody
342                    SM86IntegerMultiplyAddWideConstant
343                    (embeddingGatherF32Register (byte 14))
344                    (embeddingGatherF32Register (byte 12))
345                    (embeddingGatherF32Register (byte 3))
346                    (byte 0)
347                    embeddingGatherF32EmbeddingPointer
348                    sm86SafeControl)))))))))

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.