mirror of
https://gitea.com/gitea/tea.git
synced 2024-12-01 14:56:54 +01:00
d6df0a53b5
Co-authored-by: Norwin Roosen <git@nroo.de> Co-authored-by: Norwin <git@nroo.de> Reviewed-on: https://gitea.com/gitea/tea/pulls/390 Reviewed-by: 6543 <6543@obermui.de> Reviewed-by: Andrew Thornton <art27@cantab.net> Co-authored-by: Norwin <noerw@noreply.gitea.io> Co-committed-by: Norwin <noerw@noreply.gitea.io>
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package m
|
|
|
|
import (
|
|
. "github.com/alecthomas/chroma" // nolint
|
|
"github.com/alecthomas/chroma/lexers/internal"
|
|
)
|
|
|
|
// MLIR lexer.
|
|
var Mlir = internal.Register(MustNewLazyLexer(
|
|
&Config{
|
|
Name: "MLIR",
|
|
Aliases: []string{"mlir"},
|
|
Filenames: []string{"*.mlir"},
|
|
MimeTypes: []string{"text/x-mlir"},
|
|
},
|
|
mlirRules,
|
|
))
|
|
|
|
func mlirRules() Rules {
|
|
return Rules{
|
|
"root": {
|
|
Include("whitespace"),
|
|
{`c?"[^"]*?"`, LiteralString, nil},
|
|
{`\^([-a-zA-Z$._][\w\-$.0-9]*)\s*`, NameLabel, nil},
|
|
{`([\w\d_$.]+)\s*=`, NameLabel, nil},
|
|
Include("keyword"),
|
|
{`->`, Punctuation, nil},
|
|
{`@([\w_][\w\d_$.]*)`, NameFunction, nil},
|
|
{`[%#][\w\d_$.]+`, NameVariable, nil},
|
|
{`([1-9?][\d?]*\s*x)+`, LiteralNumber, nil},
|
|
{`0[xX][a-fA-F0-9]+`, LiteralNumber, nil},
|
|
{`-?\d+(?:[.]\d+)?(?:[eE][-+]?\d+(?:[.]\d+)?)?`, LiteralNumber, nil},
|
|
{`[=<>{}\[\]()*.,!:]|x\b`, Punctuation, nil},
|
|
{`[\w\d]+`, Text, nil},
|
|
},
|
|
"whitespace": {
|
|
{`(\n|\s)+`, Text, nil},
|
|
{`//.*?\n`, Comment, nil},
|
|
},
|
|
"keyword": {
|
|
{Words(``, ``, `constant`, `return`), KeywordType, nil},
|
|
{Words(``, ``, `func`, `loc`, `memref`, `tensor`, `vector`), KeywordType, nil},
|
|
{`bf16|f16|f32|f64|index`, Keyword, nil},
|
|
{`i[1-9]\d*`, Keyword, nil},
|
|
},
|
|
}
|
|
}
|