Changeset View
Changeset View
Standalone View
Standalone View
mlir/test/Conversion/MemRefToLLVM/memref-to-llvm.mlir
| Show First 20 Lines • Show All 611 Lines • ▼ Show 20 Lines | |||||
| // CHECK: llvm.insertvalue {{.*}}[3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64>)> | // CHECK: llvm.insertvalue {{.*}}[3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<3 x i64>, array<3 x i64>)> | ||||
| func.func @transpose(%arg0: memref<?x?x?xf32, offset: ?, strides: [?, ?, 1]>) { | func.func @transpose(%arg0: memref<?x?x?xf32, offset: ?, strides: [?, ?, 1]>) { | ||||
| %0 = memref.transpose %arg0 (i, j, k) -> (k, i, j) : memref<?x?x?xf32, offset: ?, strides: [?, ?, 1]> to memref<?x?x?xf32, affine_map<(d0, d1, d2)[s0, s1, s2] -> (d2 * s1 + s0 + d0 * s2 + d1)>> | %0 = memref.transpose %arg0 (i, j, k) -> (k, i, j) : memref<?x?x?xf32, offset: ?, strides: [?, ?, 1]> to memref<?x?x?xf32, affine_map<(d0, d1, d2)[s0, s1, s2] -> (d2 * s1 + s0 + d0 * s2 + d1)>> | ||||
| return | return | ||||
| } | } | ||||
| // ----- | // ----- | ||||
| // CHECK: llvm.mlir.global external @gv0() : !llvm.array<2 x f32> { | // CHECK: llvm.mlir.global external @gv0() {addr_space = 0 : i32} : !llvm.array<2 x f32> { | ||||
| // CHECK-NEXT: %0 = llvm.mlir.undef : !llvm.array<2 x f32> | // CHECK-NEXT: %0 = llvm.mlir.undef : !llvm.array<2 x f32> | ||||
| // CHECK-NEXT: llvm.return %0 : !llvm.array<2 x f32> | // CHECK-NEXT: llvm.return %0 : !llvm.array<2 x f32> | ||||
| // CHECK-NEXT: } | // CHECK-NEXT: } | ||||
| memref.global @gv0 : memref<2xf32> = uninitialized | memref.global @gv0 : memref<2xf32> = uninitialized | ||||
| // CHECK: llvm.mlir.global private @gv1() : !llvm.array<2 x f32> | // CHECK: llvm.mlir.global private @gv1() {addr_space = 0 : i32} : !llvm.array<2 x f32> | ||||
| memref.global "private" @gv1 : memref<2xf32> | memref.global "private" @gv1 : memref<2xf32> | ||||
| // CHECK: llvm.mlir.global external @gv2(dense<{{\[\[}}0.000000e+00, 1.000000e+00, 2.000000e+00], [3.000000e+00, 4.000000e+00, 5.000000e+00]]> : tensor<2x3xf32>) : !llvm.array<2 x array<3 x f32>> | // CHECK: llvm.mlir.global external @gv2(dense<{{\[\[}}0.000000e+00, 1.000000e+00, 2.000000e+00], [3.000000e+00, 4.000000e+00, 5.000000e+00]]> : tensor<2x3xf32>) {addr_space = 0 : i32} : !llvm.array<2 x array<3 x f32>> | ||||
| memref.global @gv2 : memref<2x3xf32> = dense<[[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]]> | memref.global @gv2 : memref<2x3xf32> = dense<[[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]]> | ||||
| // Test 1D memref. | // Test 1D memref. | ||||
| // CHECK-LABEL: func @get_gv0_memref | // CHECK-LABEL: func @get_gv0_memref | ||||
| func.func @get_gv0_memref() { | func.func @get_gv0_memref() { | ||||
| %0 = memref.get_global @gv0 : memref<2xf32> | %0 = memref.get_global @gv0 : memref<2xf32> | ||||
| // CHECK: %[[DIM:.*]] = llvm.mlir.constant(2 : index) : i64 | // CHECK: %[[DIM:.*]] = llvm.mlir.constant(2 : index) : i64 | ||||
| // CHECK: %[[STRIDE:.*]] = llvm.mlir.constant(1 : index) : i64 | // CHECK: %[[STRIDE:.*]] = llvm.mlir.constant(1 : index) : i64 | ||||
| Show All 33 Lines | func.func @get_gv2_memref() { | ||||
| // CHECK: llvm.insertvalue %[[DIM1]], {{.*}}[4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> | // CHECK: llvm.insertvalue %[[DIM1]], {{.*}}[4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> | ||||
| // CHECK: llvm.insertvalue %[[STRIDE1]], {{.*}}[4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> | // CHECK: llvm.insertvalue %[[STRIDE1]], {{.*}}[4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> | ||||
| %0 = memref.get_global @gv2 : memref<2x3xf32> | %0 = memref.get_global @gv2 : memref<2x3xf32> | ||||
| return | return | ||||
| } | } | ||||
| // Test scalar memref. | // Test scalar memref. | ||||
| // CHECK: llvm.mlir.global external @gv3(1.000000e+00 : f32) : f32 | // CHECK: llvm.mlir.global external @gv3(1.000000e+00 : f32) {addr_space = 0 : i32} : f32 | ||||
| memref.global @gv3 : memref<f32> = dense<1.0> | memref.global @gv3 : memref<f32> = dense<1.0> | ||||
| // CHECK-LABEL: func @get_gv3_memref | // CHECK-LABEL: func @get_gv3_memref | ||||
| func.func @get_gv3_memref() { | func.func @get_gv3_memref() { | ||||
| // CHECK: %[[ADDR:.*]] = llvm.mlir.addressof @gv3 : !llvm.ptr<f32> | // CHECK: %[[ADDR:.*]] = llvm.mlir.addressof @gv3 : !llvm.ptr<f32> | ||||
| // CHECK: %[[ZERO:.*]] = llvm.mlir.constant(0 : index) : i64 | // CHECK: %[[ZERO:.*]] = llvm.mlir.constant(0 : index) : i64 | ||||
| // CHECK: %[[GEP:.*]] = llvm.getelementptr %[[ADDR]][%[[ZERO]]] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32> | // CHECK: %[[GEP:.*]] = llvm.getelementptr %[[ADDR]][%[[ZERO]]] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32> | ||||
| // CHECK: %[[DEADBEEF:.*]] = llvm.mlir.constant(3735928559 : index) : i64 | // CHECK: %[[DEADBEEF:.*]] = llvm.mlir.constant(3735928559 : index) : i64 | ||||
| // CHECK: %[[DEADBEEFPTR:.*]] = llvm.inttoptr %[[DEADBEEF]] : i64 to !llvm.ptr<f32> | // CHECK: %[[DEADBEEFPTR:.*]] = llvm.inttoptr %[[DEADBEEF]] : i64 to !llvm.ptr<f32> | ||||
| // CHECK: llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | // CHECK: llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | ||||
| // CHECK: llvm.insertvalue %[[DEADBEEFPTR]], {{.*}}[0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | // CHECK: llvm.insertvalue %[[DEADBEEFPTR]], {{.*}}[0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | ||||
| // CHECK: llvm.insertvalue %[[GEP]], {{.*}}[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | // CHECK: llvm.insertvalue %[[GEP]], {{.*}}[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | ||||
| // CHECK: %[[OFFSET:.*]] = llvm.mlir.constant(0 : index) : i64 | // CHECK: %[[OFFSET:.*]] = llvm.mlir.constant(0 : index) : i64 | ||||
| // CHECK: llvm.insertvalue %[[OFFSET]], {{.*}}[2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | // CHECK: llvm.insertvalue %[[OFFSET]], {{.*}}[2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64)> | ||||
| %0 = memref.get_global @gv3 : memref<f32> | %0 = memref.get_global @gv3 : memref<f32> | ||||
| return | return | ||||
| } | } | ||||
| // Test scalar memref with an alignment. | // Test scalar memref with an alignment. | ||||
| // CHECK: llvm.mlir.global private @gv4(1.000000e+00 : f32) {alignment = 64 : i64} : f32 | // CHECK: llvm.mlir.global private @gv4(1.000000e+00 : f32) {addr_space = 0 : i32, alignment = 64 : i64} : f32 | ||||
| memref.global "private" @gv4 : memref<f32> = dense<1.0> {alignment = 64} | memref.global "private" @gv4 : memref<f32> = dense<1.0> {alignment = 64} | ||||
| // ----- | // ----- | ||||
| func.func @collapse_shape_static(%arg0: memref<1x3x4x1x5xf32>) -> memref<3x4x5xf32> { | func.func @collapse_shape_static(%arg0: memref<1x3x4x1x5xf32>) -> memref<3x4x5xf32> { | ||||
| %0 = memref.collapse_shape %arg0 [[0, 1], [2], [3, 4]] : | %0 = memref.collapse_shape %arg0 [[0, 1], [2], [3, 4]] : | ||||
| memref<1x3x4x1x5xf32> into memref<3x4x5xf32> | memref<1x3x4x1x5xf32> into memref<3x4x5xf32> | ||||
| return %0 : memref<3x4x5xf32> | return %0 : memref<3x4x5xf32> | ||||
| ▲ Show 20 Lines • Show All 424 Lines • Show Last 20 Lines | |||||