This reverts commit 95db7b4a. This breaks vectorize_2d.mlir and vectorize_3d.mlir test under ASAN (use after free).