File tree Expand file tree Collapse file tree 3 files changed +41
-37
lines changed Expand file tree Collapse file tree 3 files changed +41
-37
lines changed Original file line number Diff line number Diff line change @@ -78,15 +78,18 @@ function barrier(
78
78
attributes = NamedAttribute[]
79
79
! isnothing (barrierId) && push! (operands, barrierId)
80
80
! 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
+ )
90
93
91
94
return create_operation (
92
95
" nvvm.barrier" ,
Original file line number Diff line number Diff line change @@ -902,17 +902,18 @@ function sem_signal(
902
902
attributes = NamedAttribute[]
903
903
! isnothing (device_id) && push! (operands, device_id)
904
904
! 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
+ )
916
917
! isnothing (core_type) && push! (attributes, namedattribute (" core_type" , core_type))
917
918
918
919
return create_operation (
Original file line number Diff line number Diff line change @@ -482,18 +482,18 @@ function dot_scaled(
482
482
]
483
483
! isnothing (lhs_scale) && push! (operands, lhs_scale)
484
484
! 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
+ )
497
497
498
498
return create_operation (
499
499
" tt.dot_scaled" ,
@@ -949,16 +949,16 @@ function load(
949
949
attributes = NamedAttribute[]
950
950
! isnothing (mask) && push! (operands, mask)
951
951
! 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 )
955
955
0
956
956
elseif 1 (other == nothing )
957
957
0
958
958
else
959
959
1
960
- end ,
961
- ]) )
960
+ end ]) ,
961
+ )
962
962
! isnothing (result) && push! (op_ty_results, result)
963
963
! isnothing (boundaryCheck) &&
964
964
push! (attributes, namedattribute (" boundaryCheck" , boundaryCheck))
You can’t perform that action at this time.
0 commit comments