Skip to content

Commit 9715099

Browse files
committedFeb 6, 2025
format 3
1 parent 4effaf9 commit 9715099

File tree

2 files changed

+28
-29
lines changed

2 files changed

+28
-29
lines changed
 

‎src/mlir/Dialects/TPU.jl

+11-12
Original file line numberDiff line numberDiff line change
@@ -1085,18 +1085,17 @@ function sem_signal(
10851085
attributes = NamedAttribute[]
10861086
!isnothing(device_id) && push!(operands, device_id)
10871087
!isnothing(core_id) && push!(operands, core_id)
1088-
push!(
1089-
attributes,
1090-
operandsegmentsizes([
1091-
1, 1, if (device_id == nothing)
1092-
0
1093-
elseif 1(core_id == nothing)
1094-
0
1095-
else
1096-
1
1097-
end
1098-
]),
1099-
)
1088+
push!(attributes, operandsegmentsizes([
1089+
1,
1090+
1,
1091+
if (device_id == nothing)
1092+
0
1093+
elseif 1(core_id == nothing)
1094+
0
1095+
else
1096+
1
1097+
end,
1098+
]))
11001099
!isnothing(core_type) && push!(attributes, namedattribute("core_type", core_type))
11011100

11021101
return create_operation(

‎src/mlir/Dialects/Triton.jl

+17-17
Original file line numberDiff line numberDiff line change
@@ -588,18 +588,18 @@ function dot_scaled(
588588
]
589589
!isnothing(lhs_scale) && push!(operands, lhs_scale)
590590
!isnothing(rhs_scale) && push!(operands, rhs_scale)
591-
push!(
592-
attributes,
593-
operandsegmentsizes([
594-
1, 1, 1, if (lhs_scale == nothing)
595-
0
596-
elseif 1(rhs_scale == nothing)
597-
0
598-
else
599-
1
600-
end
601-
]),
602-
)
591+
push!(attributes, operandsegmentsizes([
592+
1,
593+
1,
594+
1,
595+
if (lhs_scale == nothing)
596+
0
597+
elseif 1(rhs_scale == nothing)
598+
0
599+
else
600+
1
601+
end,
602+
]))
603603

604604
return create_operation(
605605
"tt.dot_scaled",
@@ -1081,16 +1081,16 @@ function load(
10811081
attributes = NamedAttribute[]
10821082
!isnothing(mask) && push!(operands, mask)
10831083
!isnothing(other) && push!(operands, other)
1084-
push!(
1085-
attributes,
1086-
operandsegmentsizes([1, if (mask == nothing)
1084+
push!(attributes, operandsegmentsizes([
1085+
1,
1086+
if (mask == nothing)
10871087
0
10881088
elseif 1(other == nothing)
10891089
0
10901090
else
10911091
1
1092-
end]),
1093-
)
1092+
end,
1093+
]))
10941094
!isnothing(result) && push!(op_ty_results, result)
10951095
!isnothing(boundaryCheck) &&
10961096
push!(attributes, namedattribute("boundaryCheck", boundaryCheck))

0 commit comments

Comments
 (0)