From 3b021d7f2e6bfd42593af76c02d2aa9c26beaaf0 Mon Sep 17 00:00:00 2001
From: Uday Bondhugula <bondhugula@google.com>
Date: Thu, 27 Dec 2018 18:15:31 -0800
Subject: [PATCH] Update vim syntax file to highlight core ops

PiperOrigin-RevId: 227082502
---
 mlir/utils/vim/mlir.vim | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mlir/utils/vim/mlir.vim b/mlir/utils/vim/mlir.vim
index 8f08b0db6bbd..7bc7f1154748 100644
--- a/mlir/utils/vim/mlir.vim
+++ b/mlir/utils/vim/mlir.vim
@@ -12,6 +12,7 @@ syn keyword mlirType memref tensor vector
 
 syntax keyword mlirKeywords extfunc cfgfunc mlfunc for to step return
 syntax keyword mlirConditional if else
+syntax keyword mlirCoreOps dim addf addi subf subi mulf muli cmpi select constant affine_apply call call_indirect extract_element getTensor memref_cast tensor_cast load store alloc dealloc dma_start dma_wait
 
 syn match mlirInt "-\=\<\d\+\>"
 syn match mlirFloat "-\=\<\d\+\.\d\+\>"
@@ -23,6 +24,7 @@ syn region mlirString matchgroup=mlirString start=+"+ end=+"+
 
 hi def link mlirComment      Comment
 hi def link mlirKeywords     Statement
+hi def link mlirCoreOps      Statement
 hi def link mlirInt          Constant
 hi def link mlirType         Type
 hi def link mlirMapOutline   PreProc