@@ -65,8 +65,8 @@ func.func @reshape_ranked_memref_to_ranked(%input : memref<2x3xf32>,
6565func.func  @reshape_unranked_memref_to_ranked (%input  : memref <2 x3 xf32 >,
6666                                        %shape  : memref <2 xindex >) {
6767  %unranked_input  = memref.cast  %input  : memref <2 x3 xf32 > to  memref <*xf32 >
68-   %output  = memref.reshape  %input  (%shape )
69-                 : (memref <2 x 3 x f32 >, memref <2 xindex >) -> memref <?x?xf32 >
68+   %output  = memref.reshape  %unranked_input  (%shape )
69+                 : (memref <*x f32 >, memref <2 xindex >) -> memref <?x?xf32 >
7070
7171  %unranked_output  = memref.cast  %output  : memref <?x?xf32 > to  memref <*xf32 >
7272  call  @printMemrefF32 (%unranked_output ) : (memref <*xf32 >) -> ()
@@ -95,8 +95,8 @@ func.func @reshape_unranked_memref_to_unranked(%input : memref<2x3xf32>,
9595                                          %shape  : memref <2 xindex >) {
9696  %unranked_input  = memref.cast  %input  : memref <2 x3 xf32 > to  memref <*xf32 >
9797  %dyn_size_shape  = memref.cast  %shape  : memref <2 xindex > to  memref <?xindex >
98-   %output  = memref.reshape  %input  (%dyn_size_shape )
99-                 : (memref <2 x 3 x f32 >, memref <?xindex >) -> memref <*xf32 >
98+   %output  = memref.reshape  %unranked_input  (%dyn_size_shape )
99+                 : (memref <*x f32 >, memref <?xindex >) -> memref <*xf32 >
100100
101101  call  @printMemrefF32 (%output ) : (memref <*xf32 >) -> ()
102102  // CHECK: rank = 2 offset = 0 sizes = [3, 2] strides = [2, 1] data = 
0 commit comments