Skip to content

Commit a5fb8cb

Browse files
Format code (#520)
Co-authored-by: enzyme-ci-bot[bot] <78882869+enzyme-ci-bot[bot]@users.noreply.github.com>
1 parent ef88fdb commit a5fb8cb

File tree

3 files changed

+37
-41
lines changed

3 files changed

+37
-41
lines changed

src/mlir/Dialects/Nvvm.jl

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

9491
return create_operation(
9592
"nvvm.barrier",

src/mlir/Dialects/TPU.jl

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -902,18 +902,17 @@ function sem_signal(
902902
attributes = NamedAttribute[]
903903
!isnothing(device_id) && push!(operands, device_id)
904904
!isnothing(core_id) && push!(operands, core_id)
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-
)
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+
]))
917916
!isnothing(core_type) && push!(attributes, namedattribute("core_type", core_type))
918917

919918
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!(
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-
)
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+
]))
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!(
953-
attributes,
954-
operandsegmentsizes([1, if (mask == nothing)
952+
push!(attributes, operandsegmentsizes([
953+
1,
954+
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)