Add dotOperandDpasToLinearLayout (tensor B)

Signed-off-by: Whitney Tsang <[email protected]>
intel · Oct 3, 2024 · 803ef39 · 803ef39
1 parent c4ed65a
commit 803ef39
Show file tree

Hide file tree

Showing 4 changed files with 168 additions and 161 deletions.
diff --git a/lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp b/lib/Dialect/TritonGPU/IR/LinearLayoutConversions.cpp
@@ -1,5 +1,6 @@
 #include <vector>
 
+#include "intel/include/Dialect/TritonIntelGPU/IR/LinearLayoutConversions.h"
 #include "triton/Dialect/Triton/IR/Utility.h"
 #include "triton/Dialect/TritonGPU/IR/Attributes.h"
 #include "triton/Dialect/TritonGPU/IR/Dialect.h"
@@ -827,6 +828,9 @@ DotOperandEncodingAttr::toLinearLayout(ArrayRef<int64_t> shape) const {
  if (auto mfmaLayout = llvm::dyn_cast<AMDMfmaEncodingAttr>(getParent())) {
  return dotOperandMfmaToLinearLayout(*this, shape);
  }
+ if (auto dpasLayout = llvm::dyn_cast<intel::DpasEncodingAttr>(getParent())) {
+ return dotOperandDpasToLinearLayout(*this, shape);
+ }
 
  return std::nullopt;
 }