================ @@ -2,7 +2,7 @@ # This is just a smoke test that the dialect is functional. from mlir.ir import * -from mlir.dialects import amdgpu, func +from mlir.dialects import amdgpu, func, memref ---------------- tgymnich wrote: fixed https://github.com/llvm/llvm-project/pull/172309