Skip to content

Commit c2570b7

Browse files
Add dotOperandDpasToLinearLayout (Tensor B) (#2422)
`dotOperandMfmaToLinearLayout` for tensor B is added in upstream commit triton-lang/triton@ff02a46 recently. CI: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/11170777152 Signed-off-by: Whitney Tsang <[email protected]>
1 parent 549c9e1 commit c2570b7

File tree

4 files changed

+168
-161
lines changed

4 files changed

+168
-161
lines changed

lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
#include <vector>
22

3+
#include "intel/include/Dialect/TritonIntelGPU/IR/LinearLayoutConversions.h"
34
#include "triton/Dialect/Triton/IR/Utility.h"
45
#include "triton/Dialect/TritonGPU/IR/Attributes.h"
56
#include "triton/Dialect/TritonGPU/IR/Dialect.h"
@@ -827,6 +828,9 @@ DotOperandEncodingAttr::toLinearLayout(ArrayRef<int64_t> shape) const {
827828
if (auto mfmaLayout = llvm::dyn_cast<AMDMfmaEncodingAttr>(getParent())) {
828829
return dotOperandMfmaToLinearLayout(*this, shape);
829830
}
831+
if (auto dpasLayout = llvm::dyn_cast<intel::DpasEncodingAttr>(getParent())) {
832+
return dotOperandDpasToLinearLayout(*this, shape);
833+
}
830834

831835
return std::nullopt;
832836
}

0 commit comments

Comments
 (0)