llvm/mlir/utils/tree-sitter-mlir/dialect/scf.js

'use strict';

module.exports = {
  scf_dialect : $ => prec.right(choice(
                  // operation ::= `scf.condition` `(` $condition `)` attr-dict
                  // ($args^ `:` type($args))?
                  seq('scf.condition',
                      field('condition', $._value_use_list_parens),
                      field('attributes', optional($.attribute)),
                      field('arguments', optional($._value_use_type_list))),

                  seq('scf.execute_region',
                      field('return', optional($._function_return)),
                      field('body', $.region),
                      field('attributes', optional($.attribute))),

                  seq('scf.if', field('condition', $.value_use),
                      field('return', optional($._function_return)),
                      field('trueblk', $.region),
                      field('falseblk',
                            optional(seq(token('else'), $.region)))),

                  // operation ::= `scf.index_switch` $arg attr-dict (`->`
                  // type($results)^)?
                  //               custom<SwitchCases>($cases, $caseRegions)
                  //               `\n`
                  //               `` `default` $defaultRegion
                  seq('scf.index_switch', field('flag', $._value_use_and_type),
                      field('attributes', optional($.attribute)),
                      field('result', optional($._function_return)),
                      $.scf_case_label, $.region,
                      repeat(seq($.scf_case_label, $.region))),

                  // scf.for %iv = %lb to %ub step %step {
                  // ... // body
                  // }
                  seq('scf.for', field('iv', $.value_use), '=',
                      field('lb', $.value_use), token('to'),
                      field('ub', $.value_use),
                      field('step', seq(token('step'), $.value_use)),
                      field('iter_args',
                            optional(seq(token('iter_args'),
                                         $._value_assignment_list))),
                      field('return', optional($._function_return)),
                      field('body', $.region),
                      field('attributes', optional($.attribute))),

                  seq('scf.forall', field('iv', $._value_use_list_parens),
                      field('bounds',
                            seq(choice(seq('=', $._value_use_list_parens,
                                           token('to')),
                                       token('in')),
                                $._value_use_list_parens)),
                      field('step', optional(seq(token('step'),
                                                 $._value_use_list_parens))),
                      field('shared_outs',
                            optional(seq(token('shared_outs'),
                                         $._value_assignment_list))),
                      field('return', optional($._function_return)),
                      field('body', $.region),
                      field('attributes', optional($.attribute))),

                  seq('scf.forall.in_parallel', field('body', $.region),
                      field('attributes', optional($.attribute))),

                  seq('scf.parallel', field('iv', $._value_use_list_parens),
                      '=', field('lb', $._value_use_list_parens), token('to'),
                      field('ub', $._value_use_list_parens),
                      field('step',
                            seq(token('step'), $._value_use_list_parens)),
                      field('init', optional(seq(token('init'),
                                                 $._value_use_list_parens))),
                      field('return', optional($._function_return)),
                      field('body', $.region),
                      field('attributes', optional($.attribute))),

                  seq('scf.reduce', field('operand', $._value_use_list_parens),
                      field('return', $._type_annotation),
                      field('body', $.region)),

                  // operation ::= `scf.reduce.return` $result attr-dict `:`
                  // type($result)
                  seq('scf.reduce.return', field('result', $.value_use),
                      field('attributes', optional($.attribute)),
                      field('return', $._type_annotation)),

                  // op ::= `scf.while` assignments `:` function-type region
                  // `do` region
                  //        `attributes` attribute-dict
                  // initializer ::= /* empty */ | `(` assignment-list `)`
                  seq('scf.while',
                      field('assignments', optional($._value_assignment_list)),
                      field('return', $._function_type_annotation),
                      field('condblk', $.region), 'do',
                      field('doblk', $.region),
                      field('attributes',
                            optional(seq('attributes', $.attribute)))),

                  // operation ::= `scf.yield` attr-dict ($results^ `:`
                  // type($results))?
                  seq('scf.yield', field('attributes', optional($.attribute)),
                      field('results', optional($._value_use_type_list))),
                  )),

  scf_case_label : $ => choice(seq(token('case'), $.integer_literal),
                               token('default'))
}