108 lines
5.4 KiB
JavaScript
108 lines
5.4 KiB
JavaScript
|
'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'))
|
||
|
|
}
|