Skip to content

Commit ca98c17

Browse files
Regenerate MLIR Bindings (#513)
Co-authored-by: enzyme-ci-bot[bot] <78882869+enzyme-ci-bot[bot]@users.noreply.github.com>
1 parent 37942c8 commit ca98c17

File tree

3 files changed

+41
-37
lines changed

3 files changed

+41
-37
lines changed

src/mlir/Dialects/Nvvm.jl

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -78,15 +78,18 @@ function barrier(
7878
attributes = NamedAttribute[]
7979
!isnothing(barrierId) && push!(operands, barrierId)
8080
!isnothing(numberOfThreads) && push!(operands, numberOfThreads)
81-
push!(attributes, operandsegmentsizes([
82-
if (barrierId == nothing)
83-
0
84-
elseif 1(numberOfThreads == nothing)
85-
0
86-
else
87-
1
88-
end,
89-
]))
81+
push!(
82+
attributes,
83+
operandsegmentsizes([
84+
if (barrierId == nothing)
85+
0
86+
elseif 1(numberOfThreads == nothing)
87+
0
88+
else
89+
1
90+
end
91+
]),
92+
)
9093

9194
return create_operation(
9295
"nvvm.barrier",

src/mlir/Dialects/TPU.jl

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -902,17 +902,18 @@ function sem_signal(
902902
attributes = NamedAttribute[]
903903
!isnothing(device_id) && push!(operands, device_id)
904904
!isnothing(core_id) && push!(operands, core_id)
905-
push!(attributes, operandsegmentsizes([
906-
1,
907-
1,
908-
if (device_id == nothing)
909-
0
910-
elseif 1(core_id == nothing)
911-
0
912-
else
913-
1
914-
end,
915-
]))
905+
push!(
906+
attributes,
907+
operandsegmentsizes([
908+
1, 1, if (device_id == nothing)
909+
0
910+
elseif 1(core_id == nothing)
911+
0
912+
else
913+
1
914+
end
915+
]),
916+
)
916917
!isnothing(core_type) && push!(attributes, namedattribute("core_type", core_type))
917918

918919
return create_operation(

src/mlir/Dialects/Triton.jl

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -482,18 +482,18 @@ function dot_scaled(
482482
]
483483
!isnothing(lhs_scale) && push!(operands, lhs_scale)
484484
!isnothing(rhs_scale) && push!(operands, rhs_scale)
485-
push!(attributes, operandsegmentsizes([
486-
1,
487-
1,
488-
1,
489-
if (lhs_scale == nothing)
490-
0
491-
elseif 1(rhs_scale == nothing)
492-
0
493-
else
494-
1
495-
end,
496-
]))
485+
push!(
486+
attributes,
487+
operandsegmentsizes([
488+
1, 1, 1, if (lhs_scale == nothing)
489+
0
490+
elseif 1(rhs_scale == nothing)
491+
0
492+
else
493+
1
494+
end
495+
]),
496+
)
497497

498498
return create_operation(
499499
"tt.dot_scaled",
@@ -949,16 +949,16 @@ function load(
949949
attributes = NamedAttribute[]
950950
!isnothing(mask) && push!(operands, mask)
951951
!isnothing(other) && push!(operands, other)
952-
push!(attributes, operandsegmentsizes([
953-
1,
954-
if (mask == nothing)
952+
push!(
953+
attributes,
954+
operandsegmentsizes([1, if (mask == nothing)
955955
0
956956
elseif 1(other == nothing)
957957
0
958958
else
959959
1
960-
end,
961-
]))
960+
end]),
961+
)
962962
!isnothing(result) && push!(op_ty_results, result)
963963
!isnothing(boundaryCheck) &&
964964
push!(attributes, namedattribute("boundaryCheck", boundaryCheck))

0 commit comments

Comments
 (0)