@@ -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