@felixtensor/tree-sitter-mlir 0.1.4 → 0.1.6

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/src/grammar.json CHANGED
@@ -1553,6 +1553,19 @@
1553
1553
  ]
1554
1554
  },
1555
1555
  "custom_operation": {
1556
+ "type": "CHOICE",
1557
+ "members": [
1558
+ {
1559
+ "type": "SYMBOL",
1560
+ "name": "_tier1_custom_operation"
1561
+ },
1562
+ {
1563
+ "type": "SYMBOL",
1564
+ "name": "_tier2_custom_operation"
1565
+ }
1566
+ ]
1567
+ },
1568
+ "_tier1_custom_operation": {
1556
1569
  "type": "CHOICE",
1557
1570
  "members": [
1558
1571
  {
@@ -1578,7 +1591,12 @@
1578
1591
  "type": "SYMBOL",
1579
1592
  "name": "_affine_for_operation"
1580
1593
  }
1581
- },
1594
+ }
1595
+ ]
1596
+ },
1597
+ "_tier2_custom_operation": {
1598
+ "type": "CHOICE",
1599
+ "members": [
1582
1600
  {
1583
1601
  "type": "SYMBOL",
1584
1602
  "name": "_pdl_interp_record_match_operation"
@@ -2143,32 +2161,75 @@
2143
2161
  "members": [
2144
2162
  {
2145
2163
  "type": "SYMBOL",
2146
- "name": "value_use"
2164
+ "name": "_custom_body_reference_element"
2147
2165
  },
2148
2166
  {
2149
2167
  "type": "SYMBOL",
2150
- "name": "symbol_ref_id"
2168
+ "name": "_custom_body_type_element"
2151
2169
  },
2152
2170
  {
2153
2171
  "type": "SYMBOL",
2154
- "name": "successor"
2172
+ "name": "_custom_body_attribute_or_braced_element"
2155
2173
  },
2156
2174
  {
2157
2175
  "type": "SYMBOL",
2158
- "name": "_custom_body_complex_label"
2176
+ "name": "_custom_body_dialect_marker"
2159
2177
  },
2160
2178
  {
2161
- "type": "PREC",
2162
- "value": 2,
2163
- "content": {
2164
- "type": "SYMBOL",
2165
- "name": "type"
2166
- }
2179
+ "type": "SYMBOL",
2180
+ "name": "_custom_body_group"
2181
+ },
2182
+ {
2183
+ "type": "SYMBOL",
2184
+ "name": "_custom_body_atom"
2167
2185
  },
2186
+ {
2187
+ "type": "SYMBOL",
2188
+ "name": "_custom_body_punctuation"
2189
+ }
2190
+ ]
2191
+ },
2192
+ "_custom_body_reference_element": {
2193
+ "type": "CHOICE",
2194
+ "members": [
2195
+ {
2196
+ "type": "SYMBOL",
2197
+ "name": "value_use"
2198
+ },
2199
+ {
2200
+ "type": "SYMBOL",
2201
+ "name": "symbol_ref_id"
2202
+ },
2203
+ {
2204
+ "type": "SYMBOL",
2205
+ "name": "successor"
2206
+ }
2207
+ ]
2208
+ },
2209
+ "_custom_body_type_element": {
2210
+ "type": "PREC",
2211
+ "value": 2,
2212
+ "content": {
2213
+ "type": "SYMBOL",
2214
+ "name": "type"
2215
+ }
2216
+ },
2217
+ "_custom_body_attribute_or_braced_element": {
2218
+ "type": "CHOICE",
2219
+ "members": [
2168
2220
  {
2169
2221
  "type": "SYMBOL",
2170
2222
  "name": "attribute"
2171
2223
  },
2224
+ {
2225
+ "type": "SYMBOL",
2226
+ "name": "_custom_body_brace_payload"
2227
+ }
2228
+ ]
2229
+ },
2230
+ "_custom_body_brace_payload": {
2231
+ "type": "CHOICE",
2232
+ "members": [
2172
2233
  {
2173
2234
  "type": "SYMBOL",
2174
2235
  "name": "_custom_body_tuple_group"
@@ -2184,6 +2245,19 @@
2184
2245
  {
2185
2246
  "type": "SYMBOL",
2186
2247
  "name": "_custom_body_ssa_dict"
2248
+ }
2249
+ ]
2250
+ },
2251
+ "_custom_body_dialect_marker": {
2252
+ "type": "CHOICE",
2253
+ "members": [
2254
+ {
2255
+ "type": "SYMBOL",
2256
+ "name": "_custom_body_arrow"
2257
+ },
2258
+ {
2259
+ "type": "SYMBOL",
2260
+ "name": "_custom_body_complex_label"
2187
2261
  },
2188
2262
  {
2189
2263
  "type": "SYMBOL",
@@ -2192,7 +2266,12 @@
2192
2266
  {
2193
2267
  "type": "SYMBOL",
2194
2268
  "name": "_custom_body_sparse_operand"
2195
- },
2269
+ }
2270
+ ]
2271
+ },
2272
+ "_custom_body_group": {
2273
+ "type": "CHOICE",
2274
+ "members": [
2196
2275
  {
2197
2276
  "type": "SYMBOL",
2198
2277
  "name": "_custom_body_paren"
@@ -2204,7 +2283,29 @@
2204
2283
  {
2205
2284
  "type": "SYMBOL",
2206
2285
  "name": "_custom_body_angle_group"
2286
+ }
2287
+ ]
2288
+ },
2289
+ "_custom_body_atom": {
2290
+ "type": "CHOICE",
2291
+ "members": [
2292
+ {
2293
+ "type": "SYMBOL",
2294
+ "name": "_custom_body_literal_element"
2295
+ },
2296
+ {
2297
+ "type": "SYMBOL",
2298
+ "name": "_custom_body_reserved_keyword"
2207
2299
  },
2300
+ {
2301
+ "type": "SYMBOL",
2302
+ "name": "bare_id"
2303
+ }
2304
+ ]
2305
+ },
2306
+ "_custom_body_literal_element": {
2307
+ "type": "CHOICE",
2308
+ "members": [
2208
2309
  {
2209
2310
  "type": "SYMBOL",
2210
2311
  "name": "variadic"
@@ -2212,19 +2313,41 @@
2212
2313
  {
2213
2314
  "type": "SYMBOL",
2214
2315
  "name": "_literal"
2215
- },
2316
+ }
2317
+ ]
2318
+ },
2319
+ "_custom_body_reserved_keyword": {
2320
+ "type": "CHOICE",
2321
+ "members": [
2216
2322
  {
2217
- "type": "STRING",
2218
- "value": "array"
2323
+ "type": "SYMBOL",
2324
+ "name": "_custom_body_array_keyword"
2219
2325
  },
2220
2326
  {
2221
2327
  "type": "STRING",
2222
2328
  "value": "vector"
2223
2329
  },
2224
2330
  {
2225
- "type": "STRING",
2226
- "value": "tensor"
2331
+ "type": "SYMBOL",
2332
+ "name": "_custom_body_tensor_keyword"
2227
2333
  },
2334
+ {
2335
+ "type": "SYMBOL",
2336
+ "name": "_custom_body_affine_keyword"
2337
+ }
2338
+ ]
2339
+ },
2340
+ "_custom_body_array_keyword": {
2341
+ "type": "STRING",
2342
+ "value": "array"
2343
+ },
2344
+ "_custom_body_tensor_keyword": {
2345
+ "type": "STRING",
2346
+ "value": "tensor"
2347
+ },
2348
+ "_custom_body_affine_keyword": {
2349
+ "type": "CHOICE",
2350
+ "members": [
2228
2351
  {
2229
2352
  "type": "STRING",
2230
2353
  "value": "ceildiv"
@@ -2236,15 +2359,25 @@
2236
2359
  {
2237
2360
  "type": "STRING",
2238
2361
  "value": "mod"
2239
- },
2362
+ }
2363
+ ]
2364
+ },
2365
+ "_custom_body_punctuation": {
2366
+ "type": "CHOICE",
2367
+ "members": [
2240
2368
  {
2241
2369
  "type": "SYMBOL",
2242
- "name": "bare_id"
2370
+ "name": "_custom_body_separator_punctuation"
2243
2371
  },
2244
2372
  {
2245
2373
  "type": "SYMBOL",
2246
- "name": "_custom_body_arrow"
2247
- },
2374
+ "name": "_custom_body_operator_punctuation"
2375
+ }
2376
+ ]
2377
+ },
2378
+ "_custom_body_separator_punctuation": {
2379
+ "type": "CHOICE",
2380
+ "members": [
2248
2381
  {
2249
2382
  "type": "STRING",
2250
2383
  "value": ","
@@ -2260,7 +2393,12 @@
2260
2393
  {
2261
2394
  "type": "STRING",
2262
2395
  "value": "->"
2263
- },
2396
+ }
2397
+ ]
2398
+ },
2399
+ "_custom_body_operator_punctuation": {
2400
+ "type": "CHOICE",
2401
+ "members": [
2264
2402
  {
2265
2403
  "type": "STRING",
2266
2404
  "value": "*"
@@ -2278,8 +2416,8 @@
2278
2416
  "value": "+"
2279
2417
  },
2280
2418
  {
2281
- "type": "STRING",
2282
- "value": "-"
2419
+ "type": "SYMBOL",
2420
+ "name": "_custom_body_minus_punctuation"
2283
2421
  },
2284
2422
  {
2285
2423
  "type": "STRING",
@@ -2299,6 +2437,10 @@
2299
2437
  }
2300
2438
  ]
2301
2439
  },
2440
+ "_custom_body_minus_punctuation": {
2441
+ "type": "STRING",
2442
+ "value": "-"
2443
+ },
2302
2444
  "_custom_body_paren": {
2303
2445
  "type": "SEQ",
2304
2446
  "members": [
@@ -2738,17 +2880,21 @@
2738
2880
  ]
2739
2881
  },
2740
2882
  "_custom_body_arrow": {
2741
- "type": "SEQ",
2742
- "members": [
2743
- {
2744
- "type": "STRING",
2745
- "value": "<"
2746
- },
2747
- {
2748
- "type": "STRING",
2749
- "value": "-"
2750
- }
2751
- ]
2883
+ "type": "PREC",
2884
+ "value": 1,
2885
+ "content": {
2886
+ "type": "SEQ",
2887
+ "members": [
2888
+ {
2889
+ "type": "STRING",
2890
+ "value": "<"
2891
+ },
2892
+ {
2893
+ "type": "STRING",
2894
+ "value": "-"
2895
+ }
2896
+ ]
2897
+ }
2752
2898
  },
2753
2899
  "_nested_custom_body_element": {
2754
2900
  "type": "CHOICE",
@@ -2965,36 +3111,40 @@
2965
3111
  ]
2966
3112
  },
2967
3113
  "region": {
2968
- "type": "SEQ",
2969
- "members": [
2970
- {
2971
- "type": "STRING",
2972
- "value": "{"
2973
- },
2974
- {
2975
- "type": "CHOICE",
2976
- "members": [
2977
- {
3114
+ "type": "PREC",
3115
+ "value": 1,
3116
+ "content": {
3117
+ "type": "SEQ",
3118
+ "members": [
3119
+ {
3120
+ "type": "STRING",
3121
+ "value": "{"
3122
+ },
3123
+ {
3124
+ "type": "CHOICE",
3125
+ "members": [
3126
+ {
3127
+ "type": "SYMBOL",
3128
+ "name": "entry_block"
3129
+ },
3130
+ {
3131
+ "type": "BLANK"
3132
+ }
3133
+ ]
3134
+ },
3135
+ {
3136
+ "type": "REPEAT",
3137
+ "content": {
2978
3138
  "type": "SYMBOL",
2979
- "name": "entry_block"
2980
- },
2981
- {
2982
- "type": "BLANK"
3139
+ "name": "block"
2983
3140
  }
2984
- ]
2985
- },
2986
- {
2987
- "type": "REPEAT",
2988
- "content": {
2989
- "type": "SYMBOL",
2990
- "name": "block"
3141
+ },
3142
+ {
3143
+ "type": "STRING",
3144
+ "value": "}"
2991
3145
  }
2992
- },
2993
- {
2994
- "type": "STRING",
2995
- "value": "}"
2996
- }
2997
- ]
3146
+ ]
3147
+ }
2998
3148
  },
2999
3149
  "entry_block": {
3000
3150
  "type": "REPEAT1",
@@ -3364,119 +3514,19 @@
3364
3514
  "members": [
3365
3515
  {
3366
3516
  "type": "SYMBOL",
3367
- "name": "pretty_dialect_item_body"
3368
- },
3369
- {
3370
- "type": "SYMBOL",
3371
- "name": "_pretty_dialect_bang_body_token"
3372
- },
3373
- {
3374
- "type": "SYMBOL",
3375
- "name": "_pretty_dialect_body_attribute"
3376
- },
3377
- {
3378
- "type": "SYMBOL",
3379
- "name": "dialect_dim_list"
3380
- },
3381
- {
3382
- "type": "SYMBOL",
3383
- "name": "type"
3384
- },
3385
- {
3386
- "type": "PREC",
3387
- "value": 2,
3388
- "content": {
3389
- "type": "SYMBOL",
3390
- "name": "attribute"
3391
- }
3517
+ "name": "_pretty_dialect_structural_content"
3392
3518
  },
3393
3519
  {
3394
3520
  "type": "SYMBOL",
3395
- "name": "_literal"
3521
+ "name": "_pretty_dialect_value_content"
3396
3522
  },
3397
3523
  {
3398
3524
  "type": "SYMBOL",
3399
- "name": "_dense_keyword"
3400
- },
3401
- {
3402
- "type": "SYMBOL",
3403
- "name": "_sparse_keyword"
3404
- },
3405
- {
3406
- "type": "STRING",
3407
- "value": "array"
3408
- },
3409
- {
3410
- "type": "STRING",
3411
- "value": "vector"
3412
- },
3413
- {
3414
- "type": "STRING",
3415
- "value": "tensor"
3416
- },
3417
- {
3418
- "type": "STRING",
3419
- "value": "opaque"
3525
+ "name": "_pretty_dialect_keyword_content"
3420
3526
  },
3421
3527
  {
3422
3528
  "type": "SYMBOL",
3423
- "name": "bare_id"
3424
- },
3425
- {
3426
- "type": "STRING",
3427
- "value": ","
3428
- },
3429
- {
3430
- "type": "STRING",
3431
- "value": ":"
3432
- },
3433
- {
3434
- "type": "STRING",
3435
- "value": "="
3436
- },
3437
- {
3438
- "type": "STRING",
3439
- "value": "->"
3440
- },
3441
- {
3442
- "type": "STRING",
3443
- "value": "("
3444
- },
3445
- {
3446
- "type": "STRING",
3447
- "value": ")"
3448
- },
3449
- {
3450
- "type": "STRING",
3451
- "value": "["
3452
- },
3453
- {
3454
- "type": "STRING",
3455
- "value": "]"
3456
- },
3457
- {
3458
- "type": "STRING",
3459
- "value": "{"
3460
- },
3461
- {
3462
- "type": "STRING",
3463
- "value": "}"
3464
- },
3465
- {
3466
- "type": "STRING",
3467
- "value": "*"
3468
- },
3469
- {
3470
- "type": "STRING",
3471
- "value": "?"
3472
- },
3473
- {
3474
- "type": "STRING",
3475
- "value": "@"
3476
- },
3477
- {
3478
- "type": "STRING",
3479
- "value": "#"
3529
+ "name": "_pretty_dialect_punctuation_content"
3480
3530
  },
3481
3531
  {
3482
3532
  "type": "TOKEN",
@@ -3492,6 +3542,142 @@
3492
3542
  ]
3493
3543
  }
3494
3544
  },
3545
+ "_pretty_dialect_structural_content": {
3546
+ "type": "CHOICE",
3547
+ "members": [
3548
+ {
3549
+ "type": "SYMBOL",
3550
+ "name": "pretty_dialect_item_body"
3551
+ },
3552
+ {
3553
+ "type": "SYMBOL",
3554
+ "name": "_pretty_dialect_bang_body_token"
3555
+ },
3556
+ {
3557
+ "type": "SYMBOL",
3558
+ "name": "_pretty_dialect_body_attribute"
3559
+ }
3560
+ ]
3561
+ },
3562
+ "_pretty_dialect_value_content": {
3563
+ "type": "CHOICE",
3564
+ "members": [
3565
+ {
3566
+ "type": "SYMBOL",
3567
+ "name": "dialect_dim_list"
3568
+ },
3569
+ {
3570
+ "type": "SYMBOL",
3571
+ "name": "type"
3572
+ },
3573
+ {
3574
+ "type": "PREC",
3575
+ "value": 2,
3576
+ "content": {
3577
+ "type": "SYMBOL",
3578
+ "name": "attribute"
3579
+ }
3580
+ },
3581
+ {
3582
+ "type": "SYMBOL",
3583
+ "name": "_literal"
3584
+ }
3585
+ ]
3586
+ },
3587
+ "_pretty_dialect_keyword_content": {
3588
+ "type": "CHOICE",
3589
+ "members": [
3590
+ {
3591
+ "type": "SYMBOL",
3592
+ "name": "_dense_keyword"
3593
+ },
3594
+ {
3595
+ "type": "SYMBOL",
3596
+ "name": "_sparse_keyword"
3597
+ },
3598
+ {
3599
+ "type": "STRING",
3600
+ "value": "array"
3601
+ },
3602
+ {
3603
+ "type": "STRING",
3604
+ "value": "vector"
3605
+ },
3606
+ {
3607
+ "type": "STRING",
3608
+ "value": "tensor"
3609
+ },
3610
+ {
3611
+ "type": "STRING",
3612
+ "value": "opaque"
3613
+ },
3614
+ {
3615
+ "type": "SYMBOL",
3616
+ "name": "bare_id"
3617
+ }
3618
+ ]
3619
+ },
3620
+ "_pretty_dialect_punctuation_content": {
3621
+ "type": "CHOICE",
3622
+ "members": [
3623
+ {
3624
+ "type": "STRING",
3625
+ "value": ","
3626
+ },
3627
+ {
3628
+ "type": "STRING",
3629
+ "value": ":"
3630
+ },
3631
+ {
3632
+ "type": "STRING",
3633
+ "value": "="
3634
+ },
3635
+ {
3636
+ "type": "STRING",
3637
+ "value": "->"
3638
+ },
3639
+ {
3640
+ "type": "STRING",
3641
+ "value": "("
3642
+ },
3643
+ {
3644
+ "type": "STRING",
3645
+ "value": ")"
3646
+ },
3647
+ {
3648
+ "type": "STRING",
3649
+ "value": "["
3650
+ },
3651
+ {
3652
+ "type": "STRING",
3653
+ "value": "]"
3654
+ },
3655
+ {
3656
+ "type": "STRING",
3657
+ "value": "{"
3658
+ },
3659
+ {
3660
+ "type": "STRING",
3661
+ "value": "}"
3662
+ },
3663
+ {
3664
+ "type": "STRING",
3665
+ "value": "*"
3666
+ },
3667
+ {
3668
+ "type": "STRING",
3669
+ "value": "?"
3670
+ },
3671
+ {
3672
+ "type": "STRING",
3673
+ "value": "@"
3674
+ },
3675
+ {
3676
+ "type": "STRING",
3677
+ "value": "#"
3678
+ }
3679
+ ]
3680
+ },
3495
3681
  "_pretty_dialect_bang_body_token": {
3496
3682
  "type": "TOKEN",
3497
3683
  "content": {
@@ -5708,14 +5894,6 @@
5708
5894
  "_static_dim_list",
5709
5895
  "_static_dim_list"
5710
5896
  ],
5711
- [
5712
- "dictionary_attribute",
5713
- "region"
5714
- ],
5715
- [
5716
- "custom_op_name",
5717
- "attribute_entry"
5718
- ],
5719
5897
  [
5720
5898
  "type_alias",
5721
5899
  "dialect_namespace"
@@ -5728,41 +5906,61 @@
5728
5906
  "pretty_dialect_item"
5729
5907
  ],
5730
5908
  [
5731
- "array_literal",
5732
- "_custom_body_element_base"
5909
+ "_value_use_list",
5910
+ "_value_use_and_type"
5733
5911
  ],
5734
5912
  [
5735
- "_custom_body_element_base",
5736
- "tensor_type"
5913
+ "_type_list_no_parens",
5914
+ "_type_or_func_type"
5737
5915
  ],
5738
5916
  [
5739
- "_custom_body_element_base",
5740
- "_custom_body_arrow"
5917
+ "_type_list_parens",
5918
+ "_multi_dim_affine_expr_parens"
5741
5919
  ],
5742
5920
  [
5743
- "_generic_custom_operation_with_location_attr_dict",
5744
- "custom_op_name"
5921
+ "custom_op_name",
5922
+ "attribute_entry"
5745
5923
  ],
5746
5924
  [
5747
- "_custom_body_dict_key",
5748
- "attribute_entry"
5925
+ "array_literal",
5926
+ "_custom_body_array_keyword"
5749
5927
  ],
5750
5928
  [
5751
- "_value_use_list",
5752
- "_value_use_and_type"
5929
+ "_custom_body_tensor_keyword",
5930
+ "tensor_type"
5753
5931
  ],
5754
5932
  [
5755
- "_type_list_no_parens",
5756
- "_type_or_func_type"
5933
+ "_generic_custom_operation_with_location_attr_dict",
5934
+ "custom_op_name"
5757
5935
  ],
5758
5936
  [
5759
- "_type_list_parens",
5760
- "_multi_dim_affine_expr_parens"
5937
+ "_custom_body_dict_key",
5938
+ "attribute_entry"
5761
5939
  ]
5762
5940
  ],
5763
5941
  "precedences": [],
5764
5942
  "externals": [],
5765
- "inline": [],
5943
+ "inline": [
5944
+ "_tier1_custom_operation",
5945
+ "_tier2_custom_operation",
5946
+ "_custom_body_reference_element",
5947
+ "_custom_body_type_element",
5948
+ "_custom_body_attribute_or_braced_element",
5949
+ "_custom_body_dialect_marker",
5950
+ "_custom_body_group",
5951
+ "_custom_body_atom",
5952
+ "_custom_body_literal_element",
5953
+ "_custom_body_reserved_keyword",
5954
+ "_custom_body_affine_keyword",
5955
+ "_custom_body_brace_payload",
5956
+ "_custom_body_punctuation",
5957
+ "_custom_body_separator_punctuation",
5958
+ "_custom_body_operator_punctuation",
5959
+ "_pretty_dialect_structural_content",
5960
+ "_pretty_dialect_value_content",
5961
+ "_pretty_dialect_keyword_content",
5962
+ "_pretty_dialect_punctuation_content"
5963
+ ],
5766
5964
  "supertypes": [],
5767
5965
  "reserved": {}
5768
5966
  }