From 2d83f13e902b5bf9d5e98e9972ad370735b460f4 Mon Sep 17 00:00:00 2001 From: "enzyme-ci-bot[bot]" <78882869+enzyme-ci-bot[bot]@users.noreply.github.com> Date: Wed, 15 Jan 2025 00:13:59 +0000 Subject: [PATCH] Regenerate MLIR Bindings --- src/mlir/Dialects/Nvvm.jl | 21 ++++++++++++--------- src/mlir/Dialects/TPU.jl | 23 ++++++++++++----------- src/mlir/Dialects/Triton.jl | 34 +++++++++++++++++----------------- 3 files changed, 41 insertions(+), 37 deletions(-) diff --git a/src/mlir/Dialects/Nvvm.jl b/src/mlir/Dialects/Nvvm.jl index 8c0017b36..60f59be07 100755 --- 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 index e0a43e178..ee104ff15 100644 --- 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 index d0eb666c8..f02e239e6 100755 --- 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))