From e51a20e166b6c4c7e610fa323d8dcf1213ce88a3 Mon Sep 17 00:00:00 2001
From: Mehdi Amini <joker.eph@gmail.com>
Date: Tue, 25 Jan 2022 05:15:08 +0000
Subject: [PATCH] Fix python test to register all passes before using
 "normalize-memrefs"

The pass moved from mlir.transforms to the Memref dialect.
---
 mlir/test/python/pass_manager.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mlir/test/python/pass_manager.py b/mlir/test/python/pass_manager.py
index 380dc7ca3fb2..e5a42573a0e2 100644
--- a/mlir/test/python/pass_manager.py
+++ b/mlir/test/python/pass_manager.py
@@ -71,6 +71,7 @@ run(testParseFail)
 def testInvalidNesting():
   with Context():
     try:
+      import mlir.all_passes_registration
       pm = PassManager.parse("builtin.func(normalize-memrefs)")
     except ValueError as e:
       # CHECK: Can't add pass 'NormalizeMemRefs' restricted to 'builtin.module' on a PassManager intended to run on 'builtin.func', did you intend to nest?