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
505505
506506 LogicalResult
507507 inferReduceOpEncoding (Attribute operandEncoding, unsigned axis,
508- Attribute &resultEncoding) const override {
508+ Attribute &resultEncoding,
509+ std::optional<Location> loc) const override {
509510 resultEncoding =
510511 SliceEncodingAttr::get (getDialect ()->getContext (), axis,
511512 cast<DistributedEncodingTrait>(operandEncoding));
512513 return success ();
513514 }
514515
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 {
519521 // Not support TransOp on DPAS layout.
520522 return failure ();
521523 }
You can’t perform that action at this time.
0 commit comments