53 lines · plain
1// RUN: mlir-opt %s -one-shot-bufferize -split-input-file | FileCheck %s2 3// CHECK-LABEL: memref.global "private" @global 4ml_program.global private mutable @global(dense<0> : tensor<i64>) : tensor<i64>5 6// CHECK-LABEL: func.func @global_load_store7func.func @global_load_store() -> i64 {8 // CHECK-DAG: %[[CST127:.*]] = arith.constant 1279 // CHECK-DAG: %[[GLOBAL_1:.*]] = memref.get_global @global10 // CHECK: %[[VALUE:.*]] = memref.load %[[GLOBAL_1]][]11 // CHECK: %[[NEW_VALUE:.*]] = arith.muli %[[VALUE]], %[[CST127]]12 // CHECK: %[[ALLOC:.*]] = memref.alloc()13 // CHECK: memref.copy %[[GLOBAL_1]], %[[ALLOC]]14 // CHECK: memref.store %[[NEW_VALUE]], %[[ALLOC]][]15 // CHECK: %[[GLOBAL_2:.*]] = memref.get_global @global16 // CHECK: memref.copy %[[ALLOC]], %[[GLOBAL_2]]17 // CHECK: return %[[NEW_VALUE]]18 %c127 = arith.constant 127 : i6419 %0 = ml_program.global_load @global : tensor<i64>20 %extracted = tensor.extract %0[] : tensor<i64>21 %1 = arith.muli %extracted, %c127 : i6422 %inserted = tensor.insert %1 into %0[] : tensor<i64>23 ml_program.global_store @global = %inserted : tensor<i64>24 return %1 : i6425}26 27// -----28 29// CHECK-LABEL: memref.global "private" @global 30ml_program.global private mutable @global(dense<0> : tensor<i64>) : tensor<i64>31 32// CHECK-LABEL: func.func @raw_hazard33func.func @raw_hazard() -> i64 {34 // CHECK-DAG: %[[CST127:.*]] = arith.constant 12735 // CHECK-DAG: %[[GLOBAL_1:.*]] = memref.get_global @global36 // CHECK-DAG: %[[GLOBAL_2:.*]] = memref.get_global @global37 // CHECK-DAG: %[[ALLOC:.*]] = memref.alloc()38 // CHECK: memref.copy %[[GLOBAL_1]], %[[ALLOC]]39 // CHECK: memref.store %[[CST127]], %[[ALLOC]][]40 // CHECK: %[[VAL:.*]] = memref.load %[[GLOBAL_2]][]41 // CHECK: %[[GLOBAL_3:.*]] = memref.get_global @global42 // CHECK: memref.copy %[[ALLOC]], %[[GLOBAL_3]]43 // CHECK: return %[[VAL]]44 %c127 = arith.constant 127 : i6445 %0 = ml_program.global_load @global : tensor<i64>46 %1 = ml_program.global_load @global : tensor<i64>47 %inserted = tensor.insert %c127 into %0[] : tensor<i64>48 %extracted = tensor.extract %1[] : tensor<i64>49 ml_program.global_store @global = %inserted : tensor<i64>50 return %extracted : i6451}52 53