Regenerate MLIR Bindings (#1456)

Co-authored-by: enzyme-ci-bot[bot] <78882869+enzyme-ci-bot[bot]@users.noreply.github.com>

Author: github-actions[bot]

src/mlir/Dialects/Nvvm.jl

@@ -1593,19 +1593,22 @@ end
 `elect_sync`
 
 The `elect.sync` instruction elects one predicated active leader
-thread from among a set of threads specified in membermask.
-The membermask is set to `0xFFFFFFFF` for the current version
-of this Op. The predicate result is set to `True` for the
-leader thread, and `False` for all other threads.
+thread from among a set of threads specified in the `membermask`.
+When the `membermask` is not provided explicitly, a default value
+of `0xFFFFFFFF` is used. The predicate result is set to `True` for
+the leader thread, and `False` for all other threads.
 
 [For more information, see PTX ISA](https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#parallel-synchronization-and-communication-instructions-elect-sync)
 """
-function elect_sync(; pred::IR.Type, location=Location())
+function elect_sync(
+    membermask=nothing::Union{Nothing,Value}; pred::IR.Type, location=Location()
+)
     op_ty_results = IR.Type[pred,]
     operands = Value[]
     owned_regions = Region[]
     successors = Block[]
     attributes = NamedAttribute[]
+    !isnothing(membermask) && push!(operands, membermask)
 
     return create_operation(
         "nvvm.elect.sync",

src/mlir/Dialects/TPU.jl

@@ -411,19 +411,18 @@ function enqueue_indirect_dma(
     source::Value,
     target::Value,
     offsets::Value,
-    semaphore::Value;
+    semaphore::Value,
+    offset_filter=nothing::Union{Nothing,Value};
     add=nothing,
-    offset_filter=nothing,
     location=Location(),
 )
     op_ty_results = IR.Type[]
     operands = Value[source, target, offsets, semaphore]
     owned_regions = Region[]
     successors = Block[]
     attributes = NamedAttribute[]
+    !isnothing(offset_filter) && push!(operands, offset_filter)
     !isnothing(add) && push!(attributes, namedattribute("add", add))
-    !isnothing(offset_filter) &&
-        push!(attributes, namedattribute("offset_filter", offset_filter))
 
     return create_operation(
         "tpu.enqueue_indirect_dma",
@@ -1579,6 +1578,25 @@ function wait_dma(semaphore::Value, ref::Value; location=Location())
     )
 end
 
+function wait_indirect_dma(semaphore::Value, src::Value, dst::Value; location=Location())
+    op_ty_results = IR.Type[]
+    operands = Value[semaphore, src, dst]
+    owned_regions = Region[]
+    successors = Block[]
+    attributes = NamedAttribute[]
+
+    return create_operation(
+        "tpu.wait_indirect_dma",
+        location;
+        operands,
+        owned_regions,
+        successors,
+        attributes,
+        results=op_ty_results,
+        result_inference=false,
+    )
+end
+
 function weird(input::Value; output::IR.Type, location=Location())
     op_ty_results = IR.Type[output,]
     operands = Value[input,]

src/mlir/libMLIR_h.jl

@@ -2061,6 +2061,44 @@ function mlirBlockPrint(block, callback, userData)
     )::Cvoid
 end
 
+"""
+    mlirBlockGetNumSuccessors(block)
+
+Returns the number of successor blocks of the block.
+"""
+function mlirBlockGetNumSuccessors(block)
+    @ccall mlir_c.mlirBlockGetNumSuccessors(block::MlirBlock)::intptr_t
+end
+
+"""
+    mlirBlockGetSuccessor(block, pos)
+
+Returns `pos`-th successor of the block.
+"""
+function mlirBlockGetSuccessor(block, pos)
+    @ccall mlir_c.mlirBlockGetSuccessor(block::MlirBlock, pos::intptr_t)::MlirBlock
+end
+
+"""
+    mlirBlockGetNumPredecessors(block)
+
+Returns the number of predecessor blocks of the block.
+"""
+function mlirBlockGetNumPredecessors(block)
+    @ccall mlir_c.mlirBlockGetNumPredecessors(block::MlirBlock)::intptr_t
+end
+
+"""
+    mlirBlockGetPredecessor(block, pos)
+
+Returns `pos`-th predecessor of the block.
+
+WARNING: This getter is more expensive than the others here because the impl actually iterates the use-def chain (of block operands) anew for each indexed access.
+"""
+function mlirBlockGetPredecessor(block, pos)
+    @ccall mlir_c.mlirBlockGetPredecessor(block::MlirBlock, pos::intptr_t)::MlirBlock
+end
+
 """
     mlirValueIsNull(value)