'use strict';
module.exports = {
memref_dialect : $ => choice(
// operation ::= `memref.assume_alignment` $memref `,`
// $alignment attr-dict `:` type($memref)
seq('memref.assume_alignment',
field('memref', $.value_use), ',',
field('alignment', $.integer_literal),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.alloc` `(`$dynamicSizes`)` (`` `[`
// $symbolOperands^ `]`)? attr-dict
// `:` type($memref)
// operation ::= `memref.alloca` `(`$dynamicSizes`)` (``
// `[` $symbolOperands^ `]`)? attr-dict
// `:` type($memref)
seq(choice('memref.alloc', 'memref.alloca'),
field('dynamicSizes', $._value_use_list_parens),
field('symbolOperands', optional($._dense_idx_list)),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.cast` $source attr-dict `:`
// type($source) `to` type($dest)
seq('memref.cast', field('in', $.value_use),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.copy` $source `,` $target
// attr-dict
// `:` type($source) `to` type($target)
seq('memref.copy', field('source', $.value_use), ',',
field('target', $.value_use),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.collapse_shape` $src
// $reassociation attr-dict
// `:` type($src) `into` type($result)
// operation ::= `memref.expand_shape` $src $reassociation
// attr-dict
// `:` type($src) `into` type($result)
seq(choice('memref.collapse_shape', 'memref.expand_shape'),
field('source', $.value_use),
field('reassociation', $.nested_idx_list),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.dealloc` $memref attr-dict `:`
// type($memref)
seq('memref.dealloc', field('memref', $.value_use),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.dim` attr-dict $source `,` $index
// `:` type($source)
seq('memref.dim',
field('attributes', optional($.attribute)),
field('source', $.value_use), ',',
field('index', $.value_use),
field('return', $._type_annotation)),
// operation ::= `memref.load` $memref `[` $indices `]`
// attr-dict `:` type($memref)
seq('memref.load',
field('memref', seq($.value_use, $._dense_idx_list)),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
seq('memref.prefetch', field('source', $.value_use),
field('indices', optional($._dense_idx_list)), ',',
field('isWrite', $.isWrite_attr), ',',
field('localityHint', $.localityHint_attr), ',',
field('isDataCache', $.isDataCache_attr),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.rank` $memref attr-dict `:`
// type($memref)
seq('memref.rank', field('memref', $.value_use),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.realloc` $source (`(`
// $dynamicResultSize^ `)`)? attr-dict
// `:` type($source) `to` type(results)
seq('memref.realloc', field('source', $.value_use),
field('dynamicResultSize',
optional($._value_use_list_parens)),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.reinterpret_cast` $source `to`
// `offset` `` `:`
// custom<DynamicIndexList>($offsets,
// $static_offsets)
// `` `,` `sizes` `` `:`
// custom<DynamicIndexList>($sizes,
// $static_sizes)
// `` `,` `strides` `` `:`
// custom<DynamicIndexList>($strides,
// $static_strides) attr-dict `:`
// type($source) `to` type($result)
seq('memref.reinterpret_cast',
field('source', $.value_use), token('to'),
field('offset', seq(token('offset'), ':',
$._dense_idx_list, ',')),
field('sizes', seq(token('sizes'), ':',
$._dense_idx_list, ',')),
field('strides',
seq(token('strides'), ':', $._dense_idx_list)),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.reshape` $source `(` $shape `)`
// attr-dict
// `:` functional-type(operands, results)
seq('memref.reshape', field('source', $.value_use),
field('shape', seq('(', $.value_use, ')')),
field('attributes', optional($.attribute)),
field('return', $._function_type_annotation)),
// operation ::= `memref.store` $value `,` $memref `[`
// $indices `]` attr-dict
// `:` type($memref)
seq('memref.store', field('source', $.value_use), ',',
field('destination', $.value_use),
field('indices', $._dense_idx_list),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.subview` $source ``
// custom<DynamicIndexList>($offsets,
// $static_offsets)
// custom<DynamicIndexList>($sizes,
// $static_sizes)
// custom<DynamicIndexList>($strides,
// $static_strides) attr-dict `:`
// type($source) `to` type($result)
seq('memref.subview', field('source', $.value_use),
field('offsets', $._dense_idx_list),
field('sizes', $._dense_idx_list),
field('strides', $._dense_idx_list),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)),
// operation ::= `memref.view` $source `[` $byte_shift `]`
// `` `[` $sizes `]` attr-dict
// `:` type($source) `to` type(results)
seq('memref.view', field('source', $.value_use),
field('byte_shift', $._dense_idx_list),
field('sizes', $._dense_idx_list),
field('attributes', optional($.attribute)),
field('return', $._type_annotation)))
}
Codebase Browser
llvm