Skip to content

[mlir][x86vector] Lower vector.contract to sequence of FMAs #163382

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

[mlir][x86vector] Lower vector.contract to sequence of FMAs #163382

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
ce42a09
lower vector.contract to sequence of FMAs
arun-thmn Oct 14, 2025
176cb06
initial code to wrap as a Transform dialect
arun-thmn Oct 22, 2025
15f4c5d
fix build errors in transform code
arun-thmn Oct 22, 2025
ca4e252
Round1: env set up + clang format fix
arun-thmn Oct 24, 2025
d9334cf
Merge remote-tracking branch 'origin' into nano-kernels-fp32
arun-thmn Oct 24, 2025
ca52bdc
rebase from the main branch
arun-thmn Oct 24, 2025
f34f50d
added more commnets + more test-cases
arun-thmn Oct 27, 2025
a3385e2
renaming test-cases
arun-thmn Oct 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions mlir/include/mlir/Dialect/X86Vector/Transforms.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@

#include "mlir/IR/Value.h"

#include "mlir/Dialect/Vector/Transforms/VectorTransforms.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/PatternMatch.h"

namespace mlir {

class ImplicitLocOpBuilder;
Expand Down Expand Up @@ -79,6 +83,11 @@ struct MaskHelper {
}
};

//===----------------------------------------------------------------------===//
// Nano-kernels
LogicalResult nanoKernels(RewriterBase &rewriter,
vector::ContractionOp contractOp, int64_t vectorSize);

//===----------------------------------------------------------------------===//
/// Helpers extracted from:
/// - clang/lib/Headers/avxintrin.h
Expand Down
1 change: 1 addition & 0 deletions mlir/lib/Dialect/X86Vector/Transforms/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
add_mlir_dialect_library(MLIRX86VectorTransforms
AVXTranspose.cpp
LegalizeForLLVMExport.cpp
NanoKernels.cpp

LINK_LIBS PUBLIC
MLIRArithDialect
Expand Down
Loading
Loading