c8ca486573
resource keys have the problem that you can’t parse them from mlir
assembly if they have special or non-printable characters, but nothing
prevents you from specifying such a key when you create e.g. a
DenseResourceElementsAttr, and it works fine in other ways, including
bytecode emission and parsing
this PR solves the parsing by quoting and escaping keys with special or
non-printable characters in mlir assembly, in the same way as symbols,
e.g.:
```
module attributes {
fst = dense_resource<resource_fst> : tensor<2xf16>,
snd = dense_resource<"resource\09snd"> : tensor<2xf16>
} {}
{-#
dialect_resources: {
builtin: {
resource_fst: "0x0200000001000200",
"resource\09snd": "0x0200000008000900"
}
}
#-}
```
by not quoting keys without special or non-printable characters, the
change is effectively backwards compatible
the change is tested by:
1. adding a test with a dense resource handle key with special
characters to `dense-resource-elements-attr.mlir`
2. adding special and unprintable characters to some resource keys in
the existing lit tests `pretty-resources-print.mlir` and
`mlir/test/Bytecode/resources.mlir`
25 lines
987 B
MLIR
25 lines
987 B
MLIR
// RUN: mlir-opt -emit-bytecode %s | mlir-opt | FileCheck %s
|
|
|
|
// CHECK-LABEL: @TestDialectResources
|
|
module @TestDialectResources attributes {
|
|
// CHECK: bytecode.test = dense_resource<decl_resource> : tensor<2xui32>
|
|
// CHECK: bytecode.test2 = dense_resource<resource> : tensor<4xf64>
|
|
// CHECK: bytecode.test3 = dense_resource<"resource\09two"> : tensor<4xf64>
|
|
bytecode.test = dense_resource<decl_resource> : tensor<2xui32>,
|
|
bytecode.test2 = dense_resource<resource> : tensor<4xf64>,
|
|
bytecode.test3 = dense_resource<"resource\09two"> : tensor<4xf64>
|
|
} {}
|
|
|
|
// CHECK: builtin: {
|
|
// CHECK-NEXT: resource: "0x08000000010000000000000002000000000000000300000000000000"
|
|
// CHECK-NEXT: "resource\09two": "0x08000000010000000000000002000000000000000300000000000000"
|
|
|
|
{-#
|
|
dialect_resources: {
|
|
builtin: {
|
|
resource: "0x08000000010000000000000002000000000000000300000000000000",
|
|
"resource\09two": "0x08000000010000000000000002000000000000000300000000000000"
|
|
}
|
|
}
|
|
#-}
|