'use strict';
module.exports = {
affine_dialect : $ => prec.right(choice(
seq('affine.apply',
field('operand',
seq($.attribute, $._dim_and_symbol_use_list)),
field('attributes', optional($.attribute))),
// operation ::= `affine.delinearize_index` $linear_index
// `into` ` `
// `(` $basis `)` attr-dict `:`
// type($multi_index)
seq('affine.delinearlize_index',
field('operand', $.value_use), 'into',
field('basis', $._value_use_list_parens),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `affine.dma_start` ssa-use `[`
// multi-dim-affine-map-of-ssa-ids `]`,
// `[` multi-dim-affine-map-of-ssa-ids `]`,
// `[` multi-dim-affine-map-of-ssa-ids `]`,
// ssa-use `:` memref-type
seq(choice('affine.dma_start', 'affine.dma_wait'),
field('operands',
seq($.value_use, $._multi_dim_affine_expr_sq,
repeat(seq(',', $.value_use,
$._multi_dim_affine_expr_sq)))),
',', field('numElements', $._value_use_list),
field('return', $._type_annotation)),
// operation ::= `affine.for` ssa-id `=` lower-bound `to`
// upper-bound
// (`step` integer-literal)? `{` op* `}`
seq('affine.for', field('iv', $.value_use), '=',
field('lowerBound',
seq(optional(token('max')), $._bound)),
token('to'),
field('upperBound',
seq(optional(token('min')), $._bound)),
field('step',
optional(seq(token('step'), $.integer_literal))),
field('iter_args',
optional(seq(token('iter_args'),
$._value_assignment_list))),
field('return', optional($._function_return)),
field('body', $.region),
field('attributes', optional($.attribute))),
// operation ::= `affine.if` if-op-cond `{` op* `}`
// (`else` `{` op* `}`)? if-op-cond ::= integer-set-attr
// dim-and-symbol-use-list
seq('affine.if',
field('condition',
seq($.attribute, $._dim_and_symbol_use_list)),
field('return', optional($._function_return)),
field('trueblk', $.region),
field('falseblk',
optional(seq(token('else'), $.region))),
field('attributes', optional($.attribute))),
// operation ::= `affine.load` ssa-use `[`
// multi-dim-affine-map-of-ssa-ids `]`
// `:` memref-type
seq(choice('affine.load', 'affine.vector_load'),
field('operand', $.value_use),
field('multiDimAffineMap',
$._multi_dim_affine_expr_sq),
field('return', $._type_annotation)),
// operation ::= `affine.min` affine-map-attribute
// dim-and-symbol-use-list
seq(choice('affine.min', 'affine.max'),
field('operand',
seq($.attribute, $._dim_and_symbol_use_list))),
seq('affine.parallel',
field('iv', $._value_use_list_parens), '=',
field('lowerBound', $._multi_dim_affine_expr_parens),
token('to'),
field('upperBound', $._multi_dim_affine_expr_parens),
field('step',
optional(seq(token('step'),
$._multi_dim_affine_expr_parens))),
field('reduce',
optional(seq(token('reduce'), '(',
$.string_literal,
repeat(seq(',', $.string_literal)),
')'))),
field('return', optional($._function_return)),
field('body', $.region)),
seq('affine.prefetch', field('source', $.value_use),
field('indices',
optional($._multi_dim_affine_expr_sq)),
',', field('isWrite', $.isWrite_attr), ',',
field('localityHint', $.localityHint_attr), ',',
field('isDataCache', $.isDataCache_attr),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `affine.store` ssa-use, ssa-use `[`
// multi-dim-affine-map-of-ssa-ids `]`
// `:` memref-type
seq(choice('affine.store', 'affine.vector_store'),
field('source', $.value_use), ',',
field('destination', $.value_use),
field('multiDimAffineMap',
$._multi_dim_affine_expr_sq),
field('return', $._type_annotation)),
// operation ::= `affine.yield` attr-dict ($operands^ `:`
// type($operands))?
seq('affine.yield',
field('attributes', optional($.attribute)),
field('results', optional($._value_use_type_list)))))
}
Codebase Browser
llvm