File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed
third_party/intel/lib/Dialect/TritonIntelGPU/IR Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -505,17 +505,19 @@ struct TritonIntelGPUInferLayoutInterface
505
505
506
506
LogicalResult
507
507
inferReduceOpEncoding (Attribute operandEncoding, unsigned axis,
508
- Attribute &resultEncoding) const override {
508
+ Attribute &resultEncoding,
509
+ std::optional<Location> loc) const override {
509
510
resultEncoding =
510
511
SliceEncodingAttr::get (getDialect ()->getContext (), axis,
511
512
cast<DistributedEncodingTrait>(operandEncoding));
512
513
return success ();
513
514
}
514
515
515
- LogicalResult inferTransOpEncoding (Attribute operandEncoding,
516
- ArrayRef<int64_t > shape,
517
- ArrayRef<int32_t > order, // trans order
518
- Attribute &resultEncoding) const override {
516
+ LogicalResult
517
+ inferTransOpEncoding (Attribute operandEncoding, ArrayRef<int64_t > shape,
518
+ ArrayRef<int32_t > order, // trans order
519
+ Attribute &resultEncoding,
520
+ std::optional<Location> loc) const override {
519
521
// Not support TransOp on DPAS layout.
520
522
return failure ();
521
523
}
You can’t perform that action at this time.
0 commit comments