@@ -13,12 +13,17 @@ endif
13
13
syn case match
14
14
15
15
" Types.
16
+ "
16
17
syn keyword mlirType index f16 f32 f64
17
- " Integer type .
18
+ " Signless integer types .
18
19
syn match mlirType / \< i\d\+\> /
20
+ " Unsigned integer types.
21
+ syn match mlirType / \< ui\d\+\> /
22
+ " Signed integer types.
23
+ syn match mlirType / \< si\d\+\> /
19
24
20
25
" Elemental types inside memref, tensor, or vector types.
21
- syn match mlirType / x\s *\z s\( f16\| f32\| f64\| i\d\+\) /
26
+ syn match mlirType / x\s *\z s\( f16\| f32\| f64\| i\d\+\| ui \d\+\| si \d\+\ )/
22
27
23
28
" Shaped types.
24
29
syn match mlirType / \< memref\z e\s *<.*>/
@@ -29,13 +34,14 @@ syn match mlirType /\<vector\ze\s*<.*>/
29
34
syn match mlirType / x\s *\z svector/
30
35
31
36
" Operations.
32
- " Core ops (not exhaustive yet).
33
- " TODO: the list is not exhaustive.
34
- syn keyword mlirOps alloc alloca addf addi call call_indirect cmpf cmpi constant
35
- syn keyword mlirOps dealloc divf dma_start dma_wait dim extract_element
36
- syn keyword mlirOps getTensor index_cast load memref_cast memref_shape_cast
37
- syn keyword mlirOps mulf muli negf prefetch sitofp splat store select subf subi
38
- syn keyword mlirOps subview tensor_cast view
37
+ " Standard dialect ops.
38
+ " TODO: this list is not exhaustive.
39
+ syn keyword mlirOps alloc alloca addf addi and call call_indirect cmpf cmpi
40
+ syn keyword mlirOps constant dealloc divf dma_start dma_wait dim exp
41
+ syn keyword mlirOps extract_element getTensor index_cast load log memref_cast
42
+ syn keyword mlirOps memref_shape_cast mulf muli negf powf prefetch rsqrt sitofp
43
+ syn keyword mlirOps splat store select sqrt subf subi subview tanh tensor_cast
44
+ syn keyword mlirOps view
39
45
40
46
" Affine ops.
41
47
syn match mlirOps / \< affine\. apply\> /
@@ -44,6 +50,7 @@ syn match mlirOps /\<affine\.dma_wait\>/
44
50
syn match mlirOps / \< affine\. for\> /
45
51
syn match mlirOps / \< affine\. if\> /
46
52
syn match mlirOps / \< affine\. load\> /
53
+ syn match mlirOps / \< affine\. parallel\> /
47
54
syn match mlirOps / \< affine\. prefetch\> /
48
55
syn match mlirOps / \< affine\. store\> /
49
56
syn match mlirOps / \< loop\. for\> /
0 commit comments