// RUN: npcomp-opt -refback-lower-to-llvm -split-input-file <%s | FileCheck %s --dump-input=fail // CHECK-LABEL: llvm.func @__refbackrt_wrapper_identity( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr>, // CHECK-SAME: %[[VAL_1:.*]]: !llvm.ptr>) { // CHECK: %[[VAL_2:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_3:.*]] = llvm.getelementptr %[[VAL_0]]{{\[}}%[[VAL_2]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_4:.*]] = llvm.bitcast %[[VAL_3]] : !llvm.ptr to !llvm.ptr> // CHECK: %[[VAL_5:.*]] = llvm.load %[[VAL_4]] : !llvm.ptr> // CHECK: %[[VAL_6:.*]] = llvm.call @identity(%[[VAL_5]]) : (!llvm.ptr) -> !llvm.ptr // CHECK: %[[VAL_7:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_8:.*]] = llvm.getelementptr %[[VAL_1]]{{\[}}%[[VAL_7]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_9:.*]] = llvm.bitcast %[[VAL_8]] : !llvm.ptr to !llvm.ptr> // CHECK: llvm.store %[[VAL_6]], %[[VAL_9]] : !llvm.ptr> // CHECK: llvm.return // CHECK: } // CHECK: llvm.func @__npcomp_compiler_rt_abort_if(!llvm.i1, !llvm.ptr) // CHECK: llvm.func @__npcomp_compiler_rt_to_memref(!llvm.ptr) -> !llvm.struct<(i64, ptr)> // CHECK: llvm.func @__npcomp_compiler_rt_from_memref(!llvm.i64, !llvm.ptr) -> !llvm.ptr // CHECK: llvm.func @__npcomp_compiler_rt_get_global(!llvm.ptr, ptr)>>) -> !llvm.struct<(i64, ptr)> // CHECK: llvm.mlir.global internal constant @__npcomp_internal_constant_identity("identity") // CHECK-LABEL: llvm.mlir.global internal constant @__npcomp_func_descriptors() : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> { // CHECK: %[[VAL_0:.*]] = llvm.mlir.undef : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_1:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_2:.*]] = llvm.mlir.constant(8 : i32) : !llvm.i32 // CHECK: %[[VAL_3:.*]] = llvm.insertvalue %[[VAL_2]], %[[VAL_0]][0 : i32, 0 : i32] : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_4:.*]] = llvm.mlir.addressof @__npcomp_internal_constant_identity : !llvm.ptr> // CHECK: %[[VAL_5:.*]] = llvm.getelementptr %[[VAL_4]]{{\[}}%[[VAL_1]], %[[VAL_1]]] : (!llvm.ptr>, !llvm.i32, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_6:.*]] = llvm.insertvalue %[[VAL_5]], %[[VAL_3]][0 : i32, 1 : i32] : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_7:.*]] = llvm.mlir.addressof @__refbackrt_wrapper_identity : !llvm.ptr>, ptr>)>> // CHECK: %[[VAL_8:.*]] = llvm.bitcast %[[VAL_7]] : !llvm.ptr>, ptr>)>> to !llvm.ptr // CHECK: %[[VAL_9:.*]] = llvm.insertvalue %[[VAL_8]], %[[VAL_6]][0 : i32, 2 : i32] : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_10:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_11:.*]] = llvm.insertvalue %[[VAL_10]], %[[VAL_9]][0 : i32, 3 : i32] : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_12:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_13:.*]] = llvm.insertvalue %[[VAL_12]], %[[VAL_11]][0 : i32, 4 : i32] : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: llvm.return %[[VAL_13]] : !llvm.array<1 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: } // CHECK-LABEL: llvm.mlir.global external constant @_mlir___npcomp_module_descriptor() : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> { // CHECK: %[[VAL_0:.*]] = llvm.mlir.undef : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: %[[VAL_1:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_2:.*]] = llvm.insertvalue %[[VAL_1]], %[[VAL_0]][0 : i32] : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: %[[VAL_3:.*]] = llvm.mlir.addressof @__npcomp_func_descriptors : !llvm.ptr, ptr, i32, i32)>>> // CHECK: %[[VAL_4:.*]] = llvm.bitcast %[[VAL_3]] : !llvm.ptr, ptr, i32, i32)>>> to !llvm.ptr, ptr, i32, i32)>> // CHECK: %[[VAL_5:.*]] = llvm.insertvalue %[[VAL_4]], %[[VAL_2]][1 : i32] : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: llvm.return %[[VAL_5]] : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: } refbackrt.module_metadata { refbackrt.func_metadata {funcName = @identity, numInputs = 1 : i32, numOutputs = 1 : i32} } // CHECK-LABEL: llvm.func @identity( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr) -> !llvm.ptr { // CHECK: llvm.return %[[VAL_0]] : !llvm.ptr // CHECK: } func @identity(%arg0: !refbackrt.tensor) -> !refbackrt.tensor { return %arg0 : !refbackrt.tensor } // ----- // Test input/output arg marshaling. // CHECK-LABEL: llvm.func @__refbackrt_wrapper_inputs1results2( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr>, // CHECK-SAME: %[[VAL_1:.*]]: !llvm.ptr>) { // CHECK: %[[VAL_2:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_3:.*]] = llvm.getelementptr %[[VAL_0]]{{\[}}%[[VAL_2]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_4:.*]] = llvm.bitcast %[[VAL_3]] : !llvm.ptr to !llvm.ptr> // CHECK: %[[VAL_5:.*]] = llvm.load %[[VAL_4]] : !llvm.ptr> // CHECK: %[[VAL_6:.*]] = llvm.call @inputs1results2(%[[VAL_5]]) : (!llvm.ptr) -> !llvm.struct<(ptr, ptr)> // CHECK: %[[VAL_7:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_8:.*]] = llvm.getelementptr %[[VAL_1]]{{\[}}%[[VAL_7]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_9:.*]] = llvm.bitcast %[[VAL_8]] : !llvm.ptr to !llvm.ptr> // CHECK: %[[VAL_10:.*]] = llvm.extractvalue %[[VAL_6]][0 : i32] : !llvm.struct<(ptr, ptr)> // CHECK: llvm.store %[[VAL_10]], %[[VAL_9]] : !llvm.ptr> // CHECK: %[[VAL_11:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_12:.*]] = llvm.getelementptr %[[VAL_1]]{{\[}}%[[VAL_11]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_13:.*]] = llvm.bitcast %[[VAL_12]] : !llvm.ptr to !llvm.ptr> // CHECK: %[[VAL_14:.*]] = llvm.extractvalue %[[VAL_6]][1 : i32] : !llvm.struct<(ptr, ptr)> // CHECK: llvm.store %[[VAL_14]], %[[VAL_13]] : !llvm.ptr> // CHECK: llvm.return // CHECK: } // CHECK-LABEL: llvm.func @__refbackrt_wrapper_inputs1results1( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr>, // CHECK-SAME: %[[VAL_1:.*]]: !llvm.ptr>) { // CHECK: %[[VAL_2:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_3:.*]] = llvm.getelementptr %[[VAL_0]]{{\[}}%[[VAL_2]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_4:.*]] = llvm.bitcast %[[VAL_3]] : !llvm.ptr to !llvm.ptr> // CHECK: %[[VAL_5:.*]] = llvm.load %[[VAL_4]] : !llvm.ptr> // CHECK: %[[VAL_6:.*]] = llvm.call @inputs1results1(%[[VAL_5]]) : (!llvm.ptr) -> !llvm.ptr // CHECK: %[[VAL_7:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_8:.*]] = llvm.getelementptr %[[VAL_1]]{{\[}}%[[VAL_7]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_9:.*]] = llvm.bitcast %[[VAL_8]] : !llvm.ptr to !llvm.ptr> // CHECK: llvm.store %[[VAL_6]], %[[VAL_9]] : !llvm.ptr> // CHECK: llvm.return // CHECK: } // CHECK-LABEL: llvm.func @__refbackrt_wrapper_inputs1results0( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr>, // CHECK-SAME: %[[VAL_1:.*]]: !llvm.ptr>) { // CHECK: %[[VAL_2:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_3:.*]] = llvm.getelementptr %[[VAL_0]]{{\[}}%[[VAL_2]]] : (!llvm.ptr>, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_4:.*]] = llvm.bitcast %[[VAL_3]] : !llvm.ptr to !llvm.ptr> // CHECK: %[[VAL_5:.*]] = llvm.load %[[VAL_4]] : !llvm.ptr> // CHECK: llvm.call @inputs1results0(%[[VAL_5]]) : (!llvm.ptr) -> () // CHECK: llvm.return // CHECK: } // CHECK: llvm.func @__npcomp_compiler_rt_abort_if(!llvm.i1, !llvm.ptr) // CHECK: llvm.func @__npcomp_compiler_rt_to_memref(!llvm.ptr) -> !llvm.struct<(i64, ptr)> // CHECK: llvm.func @__npcomp_compiler_rt_from_memref(!llvm.i64, !llvm.ptr) -> !llvm.ptr // CHECK: llvm.func @__npcomp_compiler_rt_get_global(!llvm.ptr, ptr)>>) -> !llvm.struct<(i64, ptr)> // CHECK: llvm.mlir.global internal constant @__npcomp_internal_constant_inputs1results0("inputs1results0") // CHECK: llvm.mlir.global internal constant @__npcomp_internal_constant_inputs1results1("inputs1results1") // CHECK: llvm.mlir.global internal constant @__npcomp_internal_constant_inputs1results2("inputs1results2") // CHECK-LABEL: llvm.mlir.global internal constant @__npcomp_func_descriptors() : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> { // CHECK: %[[VAL_0:.*]] = llvm.mlir.undef : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_1:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_2:.*]] = llvm.mlir.constant(15 : i32) : !llvm.i32 // CHECK: %[[VAL_3:.*]] = llvm.insertvalue %[[VAL_2]], %[[VAL_0]][0 : i32, 0 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_4:.*]] = llvm.mlir.addressof @__npcomp_internal_constant_inputs1results0 : !llvm.ptr> // CHECK: %[[VAL_5:.*]] = llvm.getelementptr %[[VAL_4]]{{\[}}%[[VAL_1]], %[[VAL_1]]] : (!llvm.ptr>, !llvm.i32, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_6:.*]] = llvm.insertvalue %[[VAL_5]], %[[VAL_3]][0 : i32, 1 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_7:.*]] = llvm.mlir.addressof @__refbackrt_wrapper_inputs1results0 : !llvm.ptr>, ptr>)>> // CHECK: %[[VAL_8:.*]] = llvm.bitcast %[[VAL_7]] : !llvm.ptr>, ptr>)>> to !llvm.ptr // CHECK: %[[VAL_9:.*]] = llvm.insertvalue %[[VAL_8]], %[[VAL_6]][0 : i32, 2 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_10:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_11:.*]] = llvm.insertvalue %[[VAL_10]], %[[VAL_9]][0 : i32, 3 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_12:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_13:.*]] = llvm.insertvalue %[[VAL_12]], %[[VAL_11]][0 : i32, 4 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_14:.*]] = llvm.mlir.constant(15 : i32) : !llvm.i32 // CHECK: %[[VAL_15:.*]] = llvm.insertvalue %[[VAL_14]], %[[VAL_13]][1 : i32, 0 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_16:.*]] = llvm.mlir.addressof @__npcomp_internal_constant_inputs1results1 : !llvm.ptr> // CHECK: %[[VAL_17:.*]] = llvm.getelementptr %[[VAL_16]]{{\[}}%[[VAL_1]], %[[VAL_1]]] : (!llvm.ptr>, !llvm.i32, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_18:.*]] = llvm.insertvalue %[[VAL_17]], %[[VAL_15]][1 : i32, 1 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_19:.*]] = llvm.mlir.addressof @__refbackrt_wrapper_inputs1results1 : !llvm.ptr>, ptr>)>> // CHECK: %[[VAL_20:.*]] = llvm.bitcast %[[VAL_19]] : !llvm.ptr>, ptr>)>> to !llvm.ptr // CHECK: %[[VAL_21:.*]] = llvm.insertvalue %[[VAL_20]], %[[VAL_18]][1 : i32, 2 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_22:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_23:.*]] = llvm.insertvalue %[[VAL_22]], %[[VAL_21]][1 : i32, 3 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_24:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_25:.*]] = llvm.insertvalue %[[VAL_24]], %[[VAL_23]][1 : i32, 4 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_26:.*]] = llvm.mlir.constant(15 : i32) : !llvm.i32 // CHECK: %[[VAL_27:.*]] = llvm.insertvalue %[[VAL_26]], %[[VAL_25]][2 : i32, 0 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_28:.*]] = llvm.mlir.addressof @__npcomp_internal_constant_inputs1results2 : !llvm.ptr> // CHECK: %[[VAL_29:.*]] = llvm.getelementptr %[[VAL_28]]{{\[}}%[[VAL_1]], %[[VAL_1]]] : (!llvm.ptr>, !llvm.i32, !llvm.i32) -> !llvm.ptr // CHECK: %[[VAL_30:.*]] = llvm.insertvalue %[[VAL_29]], %[[VAL_27]][2 : i32, 1 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_31:.*]] = llvm.mlir.addressof @__refbackrt_wrapper_inputs1results2 : !llvm.ptr>, ptr>)>> // CHECK: %[[VAL_32:.*]] = llvm.bitcast %[[VAL_31]] : !llvm.ptr>, ptr>)>> to !llvm.ptr // CHECK: %[[VAL_33:.*]] = llvm.insertvalue %[[VAL_32]], %[[VAL_30]][2 : i32, 2 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_34:.*]] = llvm.mlir.constant(1 : i32) : !llvm.i32 // CHECK: %[[VAL_35:.*]] = llvm.insertvalue %[[VAL_34]], %[[VAL_33]][2 : i32, 3 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: %[[VAL_36:.*]] = llvm.mlir.constant(2 : i32) : !llvm.i32 // CHECK: %[[VAL_37:.*]] = llvm.insertvalue %[[VAL_36]], %[[VAL_35]][2 : i32, 4 : i32] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: llvm.return %[[VAL_37]] : !llvm.array<3 x struct<(i32, ptr, ptr, i32, i32)>> // CHECK: } // CHECK-LABEL: llvm.mlir.global external constant @_mlir___npcomp_module_descriptor() : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> { // CHECK: %[[VAL_0:.*]] = llvm.mlir.undef : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: %[[VAL_1:.*]] = llvm.mlir.constant(3 : i32) : !llvm.i32 // CHECK: %[[VAL_2:.*]] = llvm.insertvalue %[[VAL_1]], %[[VAL_0]][0 : i32] : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: %[[VAL_3:.*]] = llvm.mlir.addressof @__npcomp_func_descriptors : !llvm.ptr, ptr, i32, i32)>>> // CHECK: %[[VAL_4:.*]] = llvm.bitcast %[[VAL_3]] : !llvm.ptr, ptr, i32, i32)>>> to !llvm.ptr, ptr, i32, i32)>> // CHECK: %[[VAL_5:.*]] = llvm.insertvalue %[[VAL_4]], %[[VAL_2]][1 : i32] : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: llvm.return %[[VAL_5]] : !llvm.struct<(i32, ptr, ptr, i32, i32)>>)> // CHECK: } refbackrt.module_metadata { refbackrt.func_metadata {funcName = @inputs1results0, numInputs = 1 : i32, numOutputs = 0 : i32} refbackrt.func_metadata {funcName = @inputs1results1, numInputs = 1 : i32, numOutputs = 1 : i32} refbackrt.func_metadata {funcName = @inputs1results2, numInputs = 1 : i32, numOutputs = 2 : i32} } // CHECK-LABEL: llvm.func @inputs1results0( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr) { // CHECK: llvm.return // CHECK: } func @inputs1results0(%arg0: !refbackrt.tensor) { return } // CHECK-LABEL: llvm.func @inputs1results1( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr) -> !llvm.ptr { // CHECK: llvm.return %[[VAL_0]] : !llvm.ptr // CHECK: } func @inputs1results1(%arg0: !refbackrt.tensor) -> !refbackrt.tensor { return %arg0 : !refbackrt.tensor } // CHECK-LABEL: llvm.func @inputs1results2( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr) -> !llvm.struct<(ptr, ptr)> { // CHECK: %[[VAL_1:.*]] = llvm.mlir.undef : !llvm.struct<(ptr, ptr)> // CHECK: %[[VAL_2:.*]] = llvm.insertvalue %[[VAL_0]], %[[VAL_1]][0] : !llvm.struct<(ptr, ptr)> // CHECK: %[[VAL_3:.*]] = llvm.insertvalue %[[VAL_0]], %[[VAL_2]][1] : !llvm.struct<(ptr, ptr)> // CHECK: llvm.return %[[VAL_3]] : !llvm.struct<(ptr, ptr)> // CHECK: } func @inputs1results2(%arg0: !refbackrt.tensor) -> (!refbackrt.tensor, !refbackrt.tensor) { return %arg0, %arg0 : !refbackrt.tensor, !refbackrt.tensor } // ----- // Test emission of compiler runtime functions. // CHECK: llvm.mlir.global internal constant @[[STRSYM:.*]]("msg\00") // CHECK: llvm.func @__npcomp_compiler_rt_abort_if(!llvm.i1, !llvm.ptr) // CHECK: llvm.func @__npcomp_compiler_rt_to_memref(!llvm.ptr) -> !llvm.struct<(i64, ptr)> // CHECK: llvm.func @__npcomp_compiler_rt_from_memref(!llvm.i64, !llvm.ptr) -> !llvm.ptr // CHECK: llvm.func @__npcomp_compiler_rt_get_global(!llvm.ptr, ptr)>>) -> !llvm.struct<(i64, ptr)> // CHECK-LABEL: llvm.func @calls_abort_if( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.i1) { // CHECK: %[[VAL_0:.*]] = llvm.mlir.addressof @[[STRSYM]] : !llvm.ptr> // CHECK: %[[VAL_1:.*]] = llvm.mlir.constant(0 : i32) : !llvm.i32 // CHECK: %[[VAL_2:.*]] = llvm.getelementptr %[[VAL_0]]{{\[}}%[[VAL_1]], %[[VAL_1]]] : (!llvm.ptr>, !llvm.i32, !llvm.i32) -> !llvm.ptr // CHECK: llvm.call @__npcomp_compiler_rt_abort_if(%[[VAL_3:.*]], %[[VAL_2]]) : (!llvm.i1, !llvm.ptr) -> () // CHECK: llvm.return func @calls_abort_if(%arg0: i1) { refbackrt.abort_if %arg0, "msg" return } // CHECK-LABEL: llvm.func @calls_to_memref( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.ptr) { // CHECK: %[[VAL_1:.*]] = llvm.call @__npcomp_compiler_rt_to_memref(%[[VAL_0]]) : (!llvm.ptr) -> !llvm.struct<(i64, ptr)> // CHECK: llvm.return // CHECK: } func @calls_to_memref(%arg0: !refbackrt.tensor) { %0 = refbackrt.to_memref %arg0 : memref<*xf32> return } // CHECK-LABEL: llvm.func @calls_from_memref( // CHECK-SAME: %[[VAL_0:.*]]: !llvm.i64, // CHECK-SAME: %[[VAL_1:.*]]: !llvm.ptr) -> !llvm.ptr { // CHECK: %[[VAL_2:.*]] = llvm.mlir.undef : !llvm.struct<(i64, ptr)> // CHECK: %[[VAL_3:.*]] = llvm.insertvalue %[[VAL_0]], %[[VAL_2]][0] : !llvm.struct<(i64, ptr)> // CHECK: %[[VAL_4:.*]] = llvm.insertvalue %[[VAL_1]], %[[VAL_3]][1] : !llvm.struct<(i64, ptr)> // CHECK: %[[VAL_5:.*]] = llvm.extractvalue %[[VAL_4]][0 : i32] : !llvm.struct<(i64, ptr)> // CHECK: %[[VAL_6:.*]] = llvm.extractvalue %[[VAL_4]][1 : i32] : !llvm.struct<(i64, ptr)> // CHECK: %[[VAL_7:.*]] = llvm.call @__npcomp_compiler_rt_from_memref(%[[VAL_5]], %[[VAL_6]]) : (!llvm.i64, !llvm.ptr) -> !llvm.ptr // CHECK: llvm.return %[[VAL_7]] : !llvm.ptr // CHECK: } func @calls_from_memref(%arg0: memref<*xf32>) -> !refbackrt.tensor { %0 = refbackrt.from_memref %arg0 : memref<*xf32> return %0 : !refbackrt.tensor }