14 lines · plain
1// RUN: mlir-opt %s -pass-pipeline='builtin.module(func.func(test-affine-reify-value-bounds))' -verify-diagnostics \2// RUN: -split-input-file | FileCheck %s3 4// CHECK-LABEL: func @vector_transfer_write(5// CHECK-SAME: %[[t:.*]]: tensor<?xf32>6// CHECK: %[[c0:.*]] = arith.constant 0 : index7// CHECK: %[[dim:.*]] = tensor.dim %[[t]], %[[c0]]8// CHECK: return %[[dim]]9func.func @vector_transfer_write(%t: tensor<?xf32>, %v: vector<5xf32>, %pos: index) -> index {10 %0 = vector.transfer_write %v, %t[%pos] : vector<5xf32>, tensor<?xf32>11 %1 = "test.reify_bound"(%0) {dim = 0} : (tensor<?xf32>) -> (index)12 return %1 : index13}14