@felixtensor/tree-sitter-mlir 0.1.2 → 0.1.3

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/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@felixtensor/tree-sitter-mlir",
3
- "version": "0.1.2",
3
+ "version": "0.1.3",
4
4
  "description": "MLIR grammar for tree-sitter",
5
5
  "type": "module",
6
6
  "repository": {
package/src/grammar.json CHANGED
@@ -3,20 +3,11 @@
3
3
  "name": "mlir",
4
4
  "rules": {
5
5
  "toplevel": {
6
- "type": "SEQ",
7
- "members": [
8
- {
9
- "type": "SYMBOL",
10
- "name": "_toplevel"
11
- },
12
- {
13
- "type": "REPEAT",
14
- "content": {
15
- "type": "SYMBOL",
16
- "name": "_toplevel"
17
- }
18
- }
19
- ]
6
+ "type": "REPEAT",
7
+ "content": {
8
+ "type": "SYMBOL",
9
+ "name": "_toplevel"
10
+ }
20
11
  },
21
12
  "_toplevel": {
22
13
  "type": "CHOICE",
@@ -378,20 +369,17 @@
378
369
  "type": "SEQ",
379
370
  "members": [
380
371
  {
381
- "type": "TOKEN",
382
- "content": {
383
- "type": "CHOICE",
384
- "members": [
385
- {
386
- "type": "STRING",
387
- "value": "dense"
388
- },
389
- {
390
- "type": "STRING",
391
- "value": "sparse"
392
- }
393
- ]
394
- }
372
+ "type": "CHOICE",
373
+ "members": [
374
+ {
375
+ "type": "SYMBOL",
376
+ "name": "_dense_keyword"
377
+ },
378
+ {
379
+ "type": "SYMBOL",
380
+ "name": "_sparse_keyword"
381
+ }
382
+ ]
395
383
  },
396
384
  {
397
385
  "type": "STRING",
@@ -1563,6 +1551,10 @@
1563
1551
  "name": "_affine_for_operation"
1564
1552
  }
1565
1553
  },
1554
+ {
1555
+ "type": "SYMBOL",
1556
+ "name": "_generic_custom_operation_with_location_attr_dict"
1557
+ },
1566
1558
  {
1567
1559
  "type": "SYMBOL",
1568
1560
  "name": "_generic_custom_operation"
@@ -1918,6 +1910,43 @@
1918
1910
  }
1919
1911
  }
1920
1912
  },
1913
+ "_generic_custom_operation_with_location_attr_dict": {
1914
+ "type": "PREC_DYNAMIC",
1915
+ "value": -1,
1916
+ "content": {
1917
+ "type": "PREC_RIGHT",
1918
+ "value": 0,
1919
+ "content": {
1920
+ "type": "SEQ",
1921
+ "members": [
1922
+ {
1923
+ "type": "FIELD",
1924
+ "name": "name",
1925
+ "content": {
1926
+ "type": "ALIAS",
1927
+ "content": {
1928
+ "type": "SYMBOL",
1929
+ "name": "_dotted_op_name"
1930
+ },
1931
+ "named": true,
1932
+ "value": "custom_op_name"
1933
+ }
1934
+ },
1935
+ {
1936
+ "type": "SYMBOL",
1937
+ "name": "_custom_body_location_attr_dict"
1938
+ },
1939
+ {
1940
+ "type": "REPEAT",
1941
+ "content": {
1942
+ "type": "SYMBOL",
1943
+ "name": "_custom_body_element"
1944
+ }
1945
+ }
1946
+ ]
1947
+ }
1948
+ }
1949
+ },
1921
1950
  "custom_op_name": {
1922
1951
  "type": "CHOICE",
1923
1952
  "members": [
@@ -2014,6 +2043,19 @@
2014
2043
  }
2015
2044
  },
2016
2045
  "_custom_body_element": {
2046
+ "type": "CHOICE",
2047
+ "members": [
2048
+ {
2049
+ "type": "SYMBOL",
2050
+ "name": "_custom_body_element_base"
2051
+ },
2052
+ {
2053
+ "type": "SYMBOL",
2054
+ "name": "_custom_body_successor_marker"
2055
+ }
2056
+ ]
2057
+ },
2058
+ "_custom_body_element_base": {
2017
2059
  "type": "CHOICE",
2018
2060
  "members": [
2019
2061
  {
@@ -2028,6 +2070,10 @@
2028
2070
  "type": "SYMBOL",
2029
2071
  "name": "successor"
2030
2072
  },
2073
+ {
2074
+ "type": "SYMBOL",
2075
+ "name": "_custom_body_complex_label"
2076
+ },
2031
2077
  {
2032
2078
  "type": "PREC",
2033
2079
  "value": 2,
@@ -2040,6 +2086,10 @@
2040
2086
  "type": "SYMBOL",
2041
2087
  "name": "attribute"
2042
2088
  },
2089
+ {
2090
+ "type": "SYMBOL",
2091
+ "name": "_custom_body_tuple_group"
2092
+ },
2043
2093
  {
2044
2094
  "type": "SYMBOL",
2045
2095
  "name": "region"
@@ -2048,6 +2098,14 @@
2048
2098
  "type": "SYMBOL",
2049
2099
  "name": "_custom_body_value_group"
2050
2100
  },
2101
+ {
2102
+ "type": "SYMBOL",
2103
+ "name": "_custom_body_ssa_dict"
2104
+ },
2105
+ {
2106
+ "type": "SYMBOL",
2107
+ "name": "_custom_body_sparse_operand"
2108
+ },
2051
2109
  {
2052
2110
  "type": "SYMBOL",
2053
2111
  "name": "_custom_body_paren"
@@ -2072,6 +2130,10 @@
2072
2130
  "type": "STRING",
2073
2131
  "value": "vector"
2074
2132
  },
2133
+ {
2134
+ "type": "STRING",
2135
+ "value": "tensor"
2136
+ },
2075
2137
  {
2076
2138
  "type": "STRING",
2077
2139
  "value": "ceildiv"
@@ -2088,6 +2150,10 @@
2088
2150
  "type": "SYMBOL",
2089
2151
  "name": "bare_id"
2090
2152
  },
2153
+ {
2154
+ "type": "SYMBOL",
2155
+ "name": "_custom_body_arrow"
2156
+ },
2091
2157
  {
2092
2158
  "type": "STRING",
2093
2159
  "value": ","
@@ -2231,6 +2297,271 @@
2231
2297
  }
2232
2298
  ]
2233
2299
  },
2300
+ "_custom_body_tuple_group": {
2301
+ "type": "SEQ",
2302
+ "members": [
2303
+ {
2304
+ "type": "STRING",
2305
+ "value": "{"
2306
+ },
2307
+ {
2308
+ "type": "SYMBOL",
2309
+ "name": "_custom_body_tuple"
2310
+ },
2311
+ {
2312
+ "type": "REPEAT",
2313
+ "content": {
2314
+ "type": "SEQ",
2315
+ "members": [
2316
+ {
2317
+ "type": "STRING",
2318
+ "value": ","
2319
+ },
2320
+ {
2321
+ "type": "SYMBOL",
2322
+ "name": "_custom_body_tuple"
2323
+ }
2324
+ ]
2325
+ }
2326
+ },
2327
+ {
2328
+ "type": "STRING",
2329
+ "value": "}"
2330
+ }
2331
+ ]
2332
+ },
2333
+ "_custom_body_tuple": {
2334
+ "type": "SEQ",
2335
+ "members": [
2336
+ {
2337
+ "type": "STRING",
2338
+ "value": "("
2339
+ },
2340
+ {
2341
+ "type": "SYMBOL",
2342
+ "name": "_value_use_list"
2343
+ },
2344
+ {
2345
+ "type": "STRING",
2346
+ "value": ")"
2347
+ }
2348
+ ]
2349
+ },
2350
+ "_custom_body_location_attr_dict": {
2351
+ "type": "SEQ",
2352
+ "members": [
2353
+ {
2354
+ "type": "SYMBOL",
2355
+ "name": "trailing_location"
2356
+ },
2357
+ {
2358
+ "type": "SYMBOL",
2359
+ "name": "dictionary_attribute"
2360
+ }
2361
+ ]
2362
+ },
2363
+ "_custom_body_complex_label": {
2364
+ "type": "PREC",
2365
+ "value": 1,
2366
+ "content": {
2367
+ "type": "SEQ",
2368
+ "members": [
2369
+ {
2370
+ "type": "SYMBOL",
2371
+ "name": "_complex_label_start"
2372
+ },
2373
+ {
2374
+ "type": "SYMBOL",
2375
+ "name": "value_use"
2376
+ }
2377
+ ]
2378
+ }
2379
+ },
2380
+ "_custom_body_ssa_dict": {
2381
+ "type": "SEQ",
2382
+ "members": [
2383
+ {
2384
+ "type": "STRING",
2385
+ "value": "{"
2386
+ },
2387
+ {
2388
+ "type": "REPEAT",
2389
+ "content": {
2390
+ "type": "SEQ",
2391
+ "members": [
2392
+ {
2393
+ "type": "SYMBOL",
2394
+ "name": "_custom_body_attr_dict_entry"
2395
+ },
2396
+ {
2397
+ "type": "STRING",
2398
+ "value": ","
2399
+ }
2400
+ ]
2401
+ }
2402
+ },
2403
+ {
2404
+ "type": "SYMBOL",
2405
+ "name": "_custom_body_ssa_dict_entry"
2406
+ },
2407
+ {
2408
+ "type": "REPEAT",
2409
+ "content": {
2410
+ "type": "SEQ",
2411
+ "members": [
2412
+ {
2413
+ "type": "STRING",
2414
+ "value": ","
2415
+ },
2416
+ {
2417
+ "type": "SYMBOL",
2418
+ "name": "_custom_body_mixed_dict_entry"
2419
+ }
2420
+ ]
2421
+ }
2422
+ },
2423
+ {
2424
+ "type": "STRING",
2425
+ "value": "}"
2426
+ }
2427
+ ]
2428
+ },
2429
+ "_custom_body_dict_key": {
2430
+ "type": "SYMBOL",
2431
+ "name": "string_literal"
2432
+ },
2433
+ "_custom_body_attr_dict_entry": {
2434
+ "type": "SEQ",
2435
+ "members": [
2436
+ {
2437
+ "type": "SYMBOL",
2438
+ "name": "_custom_body_dict_key"
2439
+ },
2440
+ {
2441
+ "type": "STRING",
2442
+ "value": "="
2443
+ },
2444
+ {
2445
+ "type": "SYMBOL",
2446
+ "name": "attribute_value"
2447
+ }
2448
+ ]
2449
+ },
2450
+ "_custom_body_ssa_dict_entry": {
2451
+ "type": "SEQ",
2452
+ "members": [
2453
+ {
2454
+ "type": "SYMBOL",
2455
+ "name": "_custom_body_dict_key"
2456
+ },
2457
+ {
2458
+ "type": "STRING",
2459
+ "value": "="
2460
+ },
2461
+ {
2462
+ "type": "CHOICE",
2463
+ "members": [
2464
+ {
2465
+ "type": "SYMBOL",
2466
+ "name": "value_use"
2467
+ },
2468
+ {
2469
+ "type": "SYMBOL",
2470
+ "name": "_custom_body_ssa_value_array"
2471
+ }
2472
+ ]
2473
+ }
2474
+ ]
2475
+ },
2476
+ "_custom_body_mixed_dict_entry": {
2477
+ "type": "SEQ",
2478
+ "members": [
2479
+ {
2480
+ "type": "SYMBOL",
2481
+ "name": "_custom_body_dict_key"
2482
+ },
2483
+ {
2484
+ "type": "STRING",
2485
+ "value": "="
2486
+ },
2487
+ {
2488
+ "type": "CHOICE",
2489
+ "members": [
2490
+ {
2491
+ "type": "SYMBOL",
2492
+ "name": "attribute_value"
2493
+ },
2494
+ {
2495
+ "type": "SYMBOL",
2496
+ "name": "value_use"
2497
+ },
2498
+ {
2499
+ "type": "SYMBOL",
2500
+ "name": "_custom_body_ssa_value_array"
2501
+ }
2502
+ ]
2503
+ }
2504
+ ]
2505
+ },
2506
+ "_custom_body_ssa_value_array": {
2507
+ "type": "SEQ",
2508
+ "members": [
2509
+ {
2510
+ "type": "STRING",
2511
+ "value": "["
2512
+ },
2513
+ {
2514
+ "type": "SYMBOL",
2515
+ "name": "value_use"
2516
+ },
2517
+ {
2518
+ "type": "REPEAT",
2519
+ "content": {
2520
+ "type": "SEQ",
2521
+ "members": [
2522
+ {
2523
+ "type": "STRING",
2524
+ "value": ","
2525
+ },
2526
+ {
2527
+ "type": "SYMBOL",
2528
+ "name": "value_use"
2529
+ }
2530
+ ]
2531
+ }
2532
+ },
2533
+ {
2534
+ "type": "STRING",
2535
+ "value": "]"
2536
+ }
2537
+ ]
2538
+ },
2539
+ "_custom_body_successor_marker": {
2540
+ "type": "SEQ",
2541
+ "members": [
2542
+ {
2543
+ "type": "STRING",
2544
+ "value": ">"
2545
+ },
2546
+ {
2547
+ "type": "SYMBOL",
2548
+ "name": "successor"
2549
+ }
2550
+ ]
2551
+ },
2552
+ "_custom_body_sparse_operand": {
2553
+ "type": "SEQ",
2554
+ "members": [
2555
+ {
2556
+ "type": "SYMBOL",
2557
+ "name": "_sparse_keyword"
2558
+ },
2559
+ {
2560
+ "type": "SYMBOL",
2561
+ "name": "_custom_body_paren"
2562
+ }
2563
+ ]
2564
+ },
2234
2565
  "_custom_body_angle_group": {
2235
2566
  "type": "SEQ",
2236
2567
  "members": [
@@ -2251,12 +2582,25 @@
2251
2582
  }
2252
2583
  ]
2253
2584
  },
2585
+ "_custom_body_arrow": {
2586
+ "type": "SEQ",
2587
+ "members": [
2588
+ {
2589
+ "type": "STRING",
2590
+ "value": "<"
2591
+ },
2592
+ {
2593
+ "type": "STRING",
2594
+ "value": "-"
2595
+ }
2596
+ ]
2597
+ },
2254
2598
  "_nested_custom_body_element": {
2255
2599
  "type": "CHOICE",
2256
2600
  "members": [
2257
2601
  {
2258
2602
  "type": "SYMBOL",
2259
- "name": "_custom_body_element"
2603
+ "name": "_custom_body_element_base"
2260
2604
  },
2261
2605
  {
2262
2606
  "type": "SYMBOL",
@@ -2896,12 +3240,12 @@
2896
3240
  "name": "_literal"
2897
3241
  },
2898
3242
  {
2899
- "type": "STRING",
2900
- "value": "dense"
3243
+ "type": "SYMBOL",
3244
+ "name": "_dense_keyword"
2901
3245
  },
2902
3246
  {
2903
- "type": "STRING",
2904
- "value": "sparse"
3247
+ "type": "SYMBOL",
3248
+ "name": "_sparse_keyword"
2905
3249
  },
2906
3250
  {
2907
3251
  "type": "STRING",
@@ -3400,15 +3744,8 @@
3400
3744
  "type": "SEQ",
3401
3745
  "members": [
3402
3746
  {
3403
- "type": "TOKEN",
3404
- "content": {
3405
- "type": "STRING",
3406
- "value": "complex"
3407
- }
3408
- },
3409
- {
3410
- "type": "STRING",
3411
- "value": "<"
3747
+ "type": "SYMBOL",
3748
+ "name": "_complex_type_start"
3412
3749
  },
3413
3750
  {
3414
3751
  "type": "SYMBOL",
@@ -4179,8 +4516,12 @@
4179
4516
  {
4180
4517
  "type": "TOKEN",
4181
4518
  "content": {
4182
- "type": "STRING",
4183
- "value": "dense_resource"
4519
+ "type": "PREC",
4520
+ "value": 2,
4521
+ "content": {
4522
+ "type": "STRING",
4523
+ "value": "dense_resource"
4524
+ }
4184
4525
  }
4185
4526
  },
4186
4527
  {
@@ -4734,12 +5075,12 @@
4734
5075
  "name": "bare_id"
4735
5076
  },
4736
5077
  {
4737
- "type": "STRING",
4738
- "value": "dense"
5078
+ "type": "SYMBOL",
5079
+ "name": "_dense_keyword"
4739
5080
  },
4740
5081
  {
4741
- "type": "STRING",
4742
- "value": "sparse"
5082
+ "type": "SYMBOL",
5083
+ "name": "_sparse_keyword"
4743
5084
  }
4744
5085
  ]
4745
5086
  },
@@ -4761,12 +5102,12 @@
4761
5102
  "name": "bare_id"
4762
5103
  },
4763
5104
  {
4764
- "type": "STRING",
4765
- "value": "dense"
5105
+ "type": "SYMBOL",
5106
+ "name": "_dense_keyword"
4766
5107
  },
4767
5108
  {
4768
- "type": "STRING",
4769
- "value": "sparse"
5109
+ "type": "SYMBOL",
5110
+ "name": "_sparse_keyword"
4770
5111
  },
4771
5112
  {
4772
5113
  "type": "STRING",
@@ -5151,12 +5492,56 @@
5151
5492
  }
5152
5493
  ]
5153
5494
  }
5495
+ },
5496
+ "_dense_keyword": {
5497
+ "type": "TOKEN",
5498
+ "content": {
5499
+ "type": "PREC",
5500
+ "value": 1,
5501
+ "content": {
5502
+ "type": "STRING",
5503
+ "value": "dense"
5504
+ }
5505
+ }
5506
+ },
5507
+ "_sparse_keyword": {
5508
+ "type": "TOKEN",
5509
+ "content": {
5510
+ "type": "PREC",
5511
+ "value": 1,
5512
+ "content": {
5513
+ "type": "STRING",
5514
+ "value": "sparse"
5515
+ }
5516
+ }
5517
+ },
5518
+ "_complex_label_start": {
5519
+ "type": "TOKEN",
5520
+ "content": {
5521
+ "type": "PREC",
5522
+ "value": 2,
5523
+ "content": {
5524
+ "type": "PATTERN",
5525
+ "value": "complex:"
5526
+ }
5527
+ }
5528
+ },
5529
+ "_complex_type_start": {
5530
+ "type": "TOKEN",
5531
+ "content": {
5532
+ "type": "PREC",
5533
+ "value": 1,
5534
+ "content": {
5535
+ "type": "PATTERN",
5536
+ "value": "complex<"
5537
+ }
5538
+ }
5154
5539
  }
5155
5540
  },
5156
5541
  "extras": [
5157
5542
  {
5158
5543
  "type": "PATTERN",
5159
- "value": "\\s"
5544
+ "value": "[\\s\\x00]"
5160
5545
  },
5161
5546
  {
5162
5547
  "type": "SYMBOL",
@@ -5189,7 +5574,23 @@
5189
5574
  ],
5190
5575
  [
5191
5576
  "array_literal",
5192
- "_custom_body_element"
5577
+ "_custom_body_element_base"
5578
+ ],
5579
+ [
5580
+ "_custom_body_element_base",
5581
+ "tensor_type"
5582
+ ],
5583
+ [
5584
+ "_custom_body_element_base",
5585
+ "_custom_body_arrow"
5586
+ ],
5587
+ [
5588
+ "_generic_custom_operation_with_location_attr_dict",
5589
+ "custom_op_name"
5590
+ ],
5591
+ [
5592
+ "_custom_body_dict_key",
5593
+ "attribute_entry"
5193
5594
  ],
5194
5595
  [
5195
5596
  "_value_use_list",