MCPcopy Create free account
hub / github.com/Tiiny-AI/PowerInfer / main

Function main

smallthinker/tests/test-llama-grammar.cpp:12–408  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

10#include <stdexcept>
11
12int main()
13{
14 llama_grammar_parser parsed_grammar;
15
16 std::vector<std::pair<std::string, uint32_t>> expected = {
17 {"expr", 2},
18 {"expr_6", 6},
19 {"expr_7", 7},
20 {"ident", 8},
21 {"ident_10", 10},
22 {"num", 9},
23 {"num_11", 11},
24 {"root", 0},
25 {"root_1", 1},
26 {"root_5", 5},
27 {"term", 4},
28 {"ws", 3},
29 {"ws_12", 12},
30 };
31
32 std::vector<std::vector<llama_grammar_element>> expected_rules = {
33 {{LLAMA_GRETYPE_RULE_REF, 5}, {LLAMA_GRETYPE_END, 0}},
34 {
35 {LLAMA_GRETYPE_RULE_REF, 2},
36 {LLAMA_GRETYPE_CHAR, 61},
37 {LLAMA_GRETYPE_RULE_REF, 3},
38 {LLAMA_GRETYPE_RULE_REF, 4},
39 {LLAMA_GRETYPE_CHAR, 10},
40 {LLAMA_GRETYPE_END, 0},
41 },
42 {{LLAMA_GRETYPE_RULE_REF, 4}, {LLAMA_GRETYPE_RULE_REF, 7}, {LLAMA_GRETYPE_END, 0}},
43 {{LLAMA_GRETYPE_RULE_REF, 12}, {LLAMA_GRETYPE_END, 0}},
44 {
45 {LLAMA_GRETYPE_RULE_REF, 8},
46 {LLAMA_GRETYPE_ALT, 0},
47 {LLAMA_GRETYPE_RULE_REF, 9},
48 {LLAMA_GRETYPE_ALT, 0},
49 {LLAMA_GRETYPE_CHAR, 40},
50 {LLAMA_GRETYPE_RULE_REF, 3},
51 {LLAMA_GRETYPE_RULE_REF, 2},
52 {LLAMA_GRETYPE_CHAR, 41},
53 {LLAMA_GRETYPE_RULE_REF, 3},
54 {LLAMA_GRETYPE_END, 0},
55 },
56 {{LLAMA_GRETYPE_RULE_REF, 1}, {LLAMA_GRETYPE_RULE_REF, 5}, {LLAMA_GRETYPE_ALT, 0}, {LLAMA_GRETYPE_RULE_REF, 1}, {LLAMA_GRETYPE_END, 0}},
57 {
58 {LLAMA_GRETYPE_CHAR, 45},
59 {LLAMA_GRETYPE_CHAR_ALT, 43},
60 {LLAMA_GRETYPE_CHAR_ALT, 42},
61 {LLAMA_GRETYPE_CHAR_ALT, 47},
62 {LLAMA_GRETYPE_RULE_REF, 4},
63 {LLAMA_GRETYPE_END, 0},
64 },
65 {{LLAMA_GRETYPE_RULE_REF, 6}, {LLAMA_GRETYPE_RULE_REF, 7}, {LLAMA_GRETYPE_ALT, 0}, {LLAMA_GRETYPE_END, 0}},
66 {
67 {LLAMA_GRETYPE_CHAR, 97},
68 {LLAMA_GRETYPE_CHAR_RNG_UPPER, 122},
69 {LLAMA_GRETYPE_RULE_REF, 10},

Callers

nothing calls this directly

Calls 9

llama_grammar_init_implFunction · 0.85
fprintfFunction · 0.85
llama_grammar_free_implFunction · 0.85
push_backMethod · 0.45
c_rulesMethod · 0.45
dataMethod · 0.45
sizeMethod · 0.45
resizeMethod · 0.45

Tested by

no test coverage detected