From 1778d3b8245b9a7787bbd0b00f60f879ed4689c9 Mon Sep 17 00:00:00 2001
From: Nishant Patel <nishant.b.patel@intel.com>
Date: Wed, 14 May 2025 15:53:58 -0700
Subject: [mlir] [vector] Add linearization pattern for vector.create_mask
 (#138214)

This PR is a breakdown [3 / 4] of the PR #136193
The PR adds linearization patterns for vector.create_mask
---
 mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')

diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
index eda2594fbc7c..54defd949c26 100644
--- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
+++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
@@ -973,7 +973,7 @@ struct TestVectorLinearize final
     return "Linearizes ND vectors for N >= 2 into 1D vectors";
   }
   void getDependentDialects(DialectRegistry &registry) const override {
-    registry.insert<vector::VectorDialect>();
+    registry.insert<vector::VectorDialect, arith::ArithDialect>();
   }
 
   void runOnOperation() override {
-- 
cgit v1.2.3