Regenerate MLIR Bindings

EnzymeAD · Jan 15, 2025 · 2d83f13 · 2d83f13
1 parent 0e8a17a
commit 2d83f13
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 37 deletions.
diff --git a/src/mlir/Dialects/Nvvm.jl b/src/mlir/Dialects/Nvvm.jl
@@ -78,15 +78,18 @@ function barrier(
     attributes = NamedAttribute[]
     !isnothing(barrierId) && push!(operands, barrierId)
     !isnothing(numberOfThreads) && push!(operands, numberOfThreads)
-    push!(attributes, operandsegmentsizes([
-        if (barrierId == nothing)
-            0
-        elseif 1(numberOfThreads == nothing)
-            0
-        else
-            1
-        end,
-    ]))
+    push!(
+        attributes,
+        operandsegmentsizes([
+            if (barrierId == nothing)
+                0
+            elseif 1(numberOfThreads == nothing)
+                0
+            else
+                1
+            end
+        ]),
+    )
 
     return create_operation(
         "nvvm.barrier",

diff --git a/src/mlir/Dialects/TPU.jl b/src/mlir/Dialects/TPU.jl
@@ -902,17 +902,18 @@ function sem_signal(
     attributes = NamedAttribute[]
     !isnothing(device_id) && push!(operands, device_id)
     !isnothing(core_id) && push!(operands, core_id)
-    push!(attributes, operandsegmentsizes([
-        1,
-        1,
-        if (device_id == nothing)
-            0
-        elseif 1(core_id == nothing)
-            0
-        else
-            1
-        end,
-    ]))
+    push!(
+        attributes,
+        operandsegmentsizes([
+            1, 1, if (device_id == nothing)
+                0
+            elseif 1(core_id == nothing)
+                0
+            else
+                1
+            end
+        ]),
+    )
     !isnothing(core_type) && push!(attributes, namedattribute("core_type", core_type))
 
     return create_operation(

diff --git a/src/mlir/Dialects/Triton.jl b/src/mlir/Dialects/Triton.jl
@@ -482,18 +482,18 @@ function dot_scaled(
     ]
     !isnothing(lhs_scale) && push!(operands, lhs_scale)
     !isnothing(rhs_scale) && push!(operands, rhs_scale)
-    push!(attributes, operandsegmentsizes([
-        1,
-        1,
-        1,
-        if (lhs_scale == nothing)
-            0
-        elseif 1(rhs_scale == nothing)
-            0
-        else
-            1
-        end,
-    ]))
+    push!(
+        attributes,
+        operandsegmentsizes([
+            1, 1, 1, if (lhs_scale == nothing)
+                0
+            elseif 1(rhs_scale == nothing)
+                0
+            else
+                1
+            end
+        ]),
+    )
 
     return create_operation(
         "tt.dot_scaled",
@@ -949,16 +949,16 @@ function load(
     attributes = NamedAttribute[]
     !isnothing(mask) && push!(operands, mask)
     !isnothing(other) && push!(operands, other)
-    push!(attributes, operandsegmentsizes([
-        1,
-        if (mask == nothing)
+    push!(
+        attributes,
+        operandsegmentsizes([1, if (mask == nothing)
             0
         elseif 1(other == nothing)
             0
         else
             1
-        end,
-    ]))
+        end]),
+    )
     !isnothing(result) && push!(op_ty_results, result)
     !isnothing(boundaryCheck) &&
         push!(attributes, namedattribute("boundaryCheck", boundaryCheck))