@@ -3917,7 +3917,8 @@ Speculation::Speculatability MatmulOp::getSpeculatability() {
3917
3917
return getGenericSpeculatabilityImpl (cast<LinalgOp>(getOperation ()));
3918
3918
}
3919
3919
3920
- SmallVector<AffineMap> MatmulTransposeAOp::getDefaultIndexingMaps (OpBuilder &builder) {
3920
+ SmallVector<AffineMap>
3921
+ MatmulTransposeAOp::getDefaultIndexingMaps (OpBuilder &builder) {
3921
3922
AffineExpr d0, d1, d2;
3922
3923
MLIRContext *context = builder.getContext ();
3923
3924
bindDims (context, d0, d1, d2);
@@ -4006,10 +4007,12 @@ MatmulTransposeAOp::create(OpBuilder &builder, Location location,
4006
4007
4007
4008
bool MatmulTransposeAOp::classof (Operation *op) {
4008
4009
return dyn_cast_or_null<linalg::MatmulOp>(op) &&
4009
- MatmulTransposeAOp::isDefaultIndexingMaps (op->getAttr (" indexing_maps" ));
4010
+ MatmulTransposeAOp::isDefaultIndexingMaps (
4011
+ op->getAttr (" indexing_maps" ));
4010
4012
}
4011
4013
4012
- SmallVector<AffineMap> MatmulTransposeBOp::getDefaultIndexingMaps (OpBuilder &builder) {
4014
+ SmallVector<AffineMap>
4015
+ MatmulTransposeBOp::getDefaultIndexingMaps (OpBuilder &builder) {
4013
4016
AffineExpr d0, d1, d2;
4014
4017
MLIRContext *context = builder.getContext ();
4015
4018
bindDims (context, d0, d1, d2);
@@ -4098,7 +4101,8 @@ MatmulTransposeBOp::create(OpBuilder &builder, Location location,
4098
4101
4099
4102
bool MatmulTransposeBOp::classof (Operation *op) {
4100
4103
return dyn_cast_or_null<linalg::MatmulOp>(op) &&
4101
- MatmulTransposeBOp::isDefaultIndexingMaps (op->getAttr (" indexing_maps" ));
4104
+ MatmulTransposeBOp::isDefaultIndexingMaps (
4105
+ op->getAttr (" indexing_maps" ));
4102
4106
}
4103
4107
4104
4108
SmallVector<AffineMap>
@@ -4130,7 +4134,8 @@ void linalg::BatchMatmulTransposeAOp::build(
4130
4134
OpBuilder &builder, OperationState &result, ValueRange inputs,
4131
4135
ValueRange outputs, ArrayRef<NamedAttribute> attributes) {
4132
4136
buildMatmulOp (builder, result, std::nullopt, inputs, outputs, attributes,
4133
- BatchMatmulOp::getRegionBuilder (), getDefaultIndexingMaps (builder));
4137
+ BatchMatmulOp::getRegionBuilder (),
4138
+ getDefaultIndexingMaps (builder));
4134
4139
}
4135
4140
4136
4141
BatchMatmulTransposeAOp
@@ -4149,7 +4154,8 @@ void linalg::BatchMatmulTransposeAOp::build(
4149
4154
ValueRange inputs, ValueRange outputs,
4150
4155
ArrayRef<NamedAttribute> attributes) {
4151
4156
buildMatmulOp (builder, result, resultTensorTypes, inputs, outputs, attributes,
4152
- BatchMatmulOp::getRegionBuilder (), getDefaultIndexingMaps (builder));
4157
+ BatchMatmulOp::getRegionBuilder (),
4158
+ getDefaultIndexingMaps (builder));
4153
4159
}
4154
4160
4155
4161
BatchMatmulTransposeAOp
@@ -4170,7 +4176,8 @@ void linalg::BatchMatmulTransposeAOp::build(
4170
4176
ArrayRef<NamedAttribute> attributes) {
4171
4177
result.addAttribute (" cast" , cast);
4172
4178
buildMatmulOp (builder, result, resultTensorTypes, inputs, outputs, attributes,
4173
- BatchMatmulOp::getRegionBuilder (), getDefaultIndexingMaps (builder));
4179
+ BatchMatmulOp::getRegionBuilder (),
4180
+ getDefaultIndexingMaps (builder));
4174
4181
}
4175
4182
4176
4183
BatchMatmulTransposeAOp
@@ -4220,7 +4227,8 @@ void linalg::BatchMatmulTransposeBOp::build(
4220
4227
OpBuilder &builder, OperationState &result, ValueRange inputs,
4221
4228
ValueRange outputs, ArrayRef<NamedAttribute> attributes) {
4222
4229
buildMatmulOp (builder, result, std::nullopt, inputs, outputs, attributes,
4223
- BatchMatmulOp::getRegionBuilder (), getDefaultIndexingMaps (builder));
4230
+ BatchMatmulOp::getRegionBuilder (),
4231
+ getDefaultIndexingMaps (builder));
4224
4232
}
4225
4233
4226
4234
BatchMatmulTransposeBOp
@@ -4239,7 +4247,8 @@ void linalg::BatchMatmulTransposeBOp::build(
4239
4247
ValueRange inputs, ValueRange outputs,
4240
4248
ArrayRef<NamedAttribute> attributes) {
4241
4249
buildMatmulOp (builder, result, resultTensorTypes, inputs, outputs, attributes,
4242
- BatchMatmulOp::getRegionBuilder (), getDefaultIndexingMaps (builder));
4250
+ BatchMatmulOp::getRegionBuilder (),
4251
+ getDefaultIndexingMaps (builder));
4243
4252
}
4244
4253
4245
4254
BatchMatmulTransposeBOp
@@ -4260,7 +4269,8 @@ void linalg::BatchMatmulTransposeBOp::build(
4260
4269
ArrayRef<NamedAttribute> attributes) {
4261
4270
result.addAttribute (" cast" , cast);
4262
4271
buildMatmulOp (builder, result, resultTensorTypes, inputs, outputs, attributes,
4263
- BatchMatmulOp::getRegionBuilder (), getDefaultIndexingMaps (builder));
4272
+ BatchMatmulOp::getRegionBuilder (),
4273
+ getDefaultIndexingMaps (builder));
4264
4274
}
4265
4275
4266
4276
BatchMatmulTransposeBOp
0 commit comments