e4c74fd9dd
A splat attribute have a single element during printing so we should treat it as such when we decide if we elide it or not based on the flag intended to elide large attributes. Reviewed By: rriddle, mehdi_amini Differential Revision: https://reviews.llvm.org/D92165
22 lines
1.1 KiB
MLIR
22 lines
1.1 KiB
MLIR
// RUN: mlir-opt %s -mlir-elide-elementsattrs-if-larger=2 | FileCheck %s
|
|
// Ensure that the elided version is still parseable, although depending on
|
|
// what has been elided, it may not be semantically meaningful.
|
|
// In the typical case where what is being elided is a very large constant
|
|
// tensor which passes don't look at directly, this isn't an issue.
|
|
// RUN: mlir-opt %s -mlir-elide-elementsattrs-if-larger=2 | mlir-opt
|
|
|
|
// CHECK: opaque<"", "0xDEADBEEF"> : tensor<3xi32>
|
|
"test.dense_attr"() {foo.dense_attr = dense<[1, 2, 3]> : tensor<3xi32>} : () -> ()
|
|
|
|
// CHECK: dense<[1, 2]> : tensor<2xi32>
|
|
"test.non_elided_dense_attr"() {foo.dense_attr = dense<[1, 2]> : tensor<2xi32>} : () -> ()
|
|
|
|
// CHECK: opaque<"", "0xDEADBEEF"> : vector<1x1x1xf16>
|
|
"test.sparse_attr"() {foo.sparse_attr = sparse<[[1, 2, 3]], -2.0> : vector<1x1x1xf16>} : () -> ()
|
|
|
|
// CHECK: opaque<"", "0xDEADBEEF"> : tensor<100xf32>
|
|
"test.opaque_attr"() {foo.opaque_attr = opaque<"", "0xEBFE"> : tensor<100xf32> } : () -> ()
|
|
|
|
// CHECK: dense<1> : tensor<3xi32>
|
|
"test.dense_splat"() {foo.dense_attr = dense<1> : tensor<3xi32>} : () -> ()
|