[all-commits] [llvm/llvm-project] d4159e: [MLIR][NVVM] Add support for griddepcontrol Ops (#...

[Srinivasa Ravi via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d4159e2a1d1d640077b2e5cde66b0a284049955f
      https://github.com/llvm/llvm-project/commit/d4159e2a1d1d640077b2e5cde66b0a284049955f
  Author: Srinivasa Ravi <srinivasar at nvidia.com>
  Date:   2025-01-29 (Wed, 29 Jan 2025)

  Changed paths:
    M mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
    M mlir/test/Dialect/LLVMIR/nvvm.mlir
    M mlir/test/Target/LLVMIR/nvvmir.mlir

  Log Message:
  -----------
  [MLIR][NVVM] Add support for griddepcontrol Ops (#124603)

Adds `griddepcontrol.wait` and `griddepcontrol.launch.dependents`
MLIR Ops to generate griddepcontrol instructions.

`griddepcontrol` - Allows dependent and prerequisite grids as defined by
the runtime to control execution in the following ways:

- `griddepcontrol.wait` - causes the executing thread to wait until all
  prerequisite grids in flight have completed and all the memory
  operations from the prerequisite grids are performed and made visible
  to the current grid.

- `griddepcontrol.launch.dependents` - signals that specific dependents
  the runtime system designated to react to this instruction can be
  scheduled as soon as all other CTAs in the grid issue the same
  instruction or have completed.

PTX Spec Reference:

https://docs.nvidia.com/cuda/parallel-thread-execution/#parallel-synchronization-and-communication-instructions-griddepcontrol



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications