diff options
| author | Andy Davis <andydavis@google.com> | 2019-12-03 11:55:09 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-03 11:55:54 -0800 |
| commit | 2c13fd9f17207d9254c8b3ff1a7c5d92ace35441 (patch) | |
| tree | 5ccb8dcf9858011a6531d974b798e5bdcd76d192 /mlir/test/Dialect/VectorOps | |
| parent | 67515e8d7a3bf5bcaa8475890eea3c63d7ea732c (diff) | |
| download | bcm5719-llvm-2c13fd9f17207d9254c8b3ff1a7c5d92ace35441.tar.gz bcm5719-llvm-2c13fd9f17207d9254c8b3ff1a7c5d92ace35441.zip | |
Add CreateMaskOp to the VectorOps dialect.
PiperOrigin-RevId: 283591888
Diffstat (limited to 'mlir/test/Dialect/VectorOps')
| -rw-r--r-- | mlir/test/Dialect/VectorOps/invalid.mlir | 10 | ||||
| -rw-r--r-- | mlir/test/Dialect/VectorOps/ops.mlir | 12 |
2 files changed, 22 insertions, 0 deletions
diff --git a/mlir/test/Dialect/VectorOps/invalid.mlir b/mlir/test/Dialect/VectorOps/invalid.mlir index 0fbcb56f388..0f19033fb42 100644 --- a/mlir/test/Dialect/VectorOps/invalid.mlir +++ b/mlir/test/Dialect/VectorOps/invalid.mlir @@ -606,3 +606,13 @@ func @contraction(%arg0: vector<7x8x16x15xf32>, %arg1: vector<8x16x7x5xf32>, : vector<7x8x16x15xf32>, vector<8x16x7x5xf32> into vector<8x15x5xf32> return } + +// ----- + +func @create_mask() { + %c2 = constant 2 : index + %c3 = constant 3 : index + // expected-error@+1 {{must specify an operand for each result vector dimension}} + %0 = vector.create_mask %c3, %c2 : vector<4x3x7xi1> + return +} diff --git a/mlir/test/Dialect/VectorOps/ops.mlir b/mlir/test/Dialect/VectorOps/ops.mlir index 3824dfe20e4..0a52a1ea45b 100644 --- a/mlir/test/Dialect/VectorOps/ops.mlir +++ b/mlir/test/Dialect/VectorOps/ops.mlir @@ -124,3 +124,15 @@ func @contraction(%arg0 : vector<7x8x16x15xf32>, %arg1 : vector<8x16x7x5xf32>, : vector<7x8x16x15xf32>, vector<8x16x7x5xf32> into vector<8x15x8x5xf32> return } + +// CHECK-LABEL: create_vector_mask +func @create_vector_mask() { + // CHECK: %[[C2:.*]] = constant 2 : index + %c2 = constant 2 : index + // CHECK-NEXT: %[[C3:.*]] = constant 3 : index + %c3 = constant 3 : index + // CHECK-NEXT: vector.create_mask %[[C3]], %[[C2]] : vector<4x3xi1> + %0 = vector.create_mask %c3, %c2 : vector<4x3xi1> + + return +} |
