Source/Packages

Realization.Nvidia.SM86.EmbeddingGatherF32SM86

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

672 lines106 declarations27.6 KiBSHA-256 1a27ac9fb9de

def · lines 267–339

embeddingGatherF32AddressProgramFor

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