From: 雾雨魔理沙 Date: Fri, 28 Jun 2019 00:35:03 +0000 (-0700) Subject: [Relay][Parser] simplify build script, remove python 2 support (#3419) X-Git-Tag: upstream/0.7.0~2247 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b05dc12ff4656d17c55f08fb14144435f47727f6;p=platform%2Fupstream%2Ftvm.git [Relay][Parser] simplify build script, remove python 2 support (#3419) * simplify build script, remove python 2 support * remove py2 file * update py3 --- diff --git a/cmake/modules/ANTLR.cmake b/cmake/modules/ANTLR.cmake index d1e8e4e..0ba8f8d 100644 --- a/cmake/modules/ANTLR.cmake +++ b/cmake/modules/ANTLR.cmake @@ -16,44 +16,19 @@ # under the License. if(USE_ANTLR) - file(GLOB_RECURSE ANTLR4 - /usr/local/lib/antlr-*-complete.jar - /usr/local/Cellar/*antlr-*-complete.jar) + set(RELAY_PARSER_DIR + ${CMAKE_CURRENT_SOURCE_DIR}/python/tvm/relay/grammar) - if(DEFINED ANTLR4) - # Get the first element of the list of antlr jars. - # Sort and reverse the list so the item selected is the highest - # version in lib or else in Cellar if no lib installation exists. - list(SORT ANTLR4) - list(REVERSE ANTLR4) - list(GET ANTLR4 0 ANTLR4) - set(RELAY_PARSER_DIR - ${CMAKE_CURRENT_SOURCE_DIR}/python/tvm/relay/grammar) + set(RELAY_PARSER + ${RELAY_PARSER_DIR}/py3/RelayVisitor.py + ${RELAY_PARSER_DIR}/py3/RelayParser.py + ${RELAY_PARSER_DIR}/py3/RelayLexer.py) - set(RELAY_PARSER - ${RELAY_PARSER_DIR}/py2/RelayVisitor.py - ${RELAY_PARSER_DIR}/py2/RelayParser.py - ${RELAY_PARSER_DIR}/py2/RelayLexer.py + # Generate ANTLR grammar for parsing. + add_custom_command(OUTPUT ${RELAY_PARSER} + COMMAND antlr4 -visitor -no-listener -Dlanguage=Python3 ${RELAY_PARSER_DIR}/Relay.g4 -o ${RELAY_PARSER_DIR}/py3 + DEPENDS ${RELAY_PARSER_DIR}/Relay.g4 + WORKING_DIRECTORY ${RELAY_PARSER_DIR}) - ${RELAY_PARSER_DIR}/py3/RelayVisitor.py - ${RELAY_PARSER_DIR}/py3/RelayParser.py - ${RELAY_PARSER_DIR}/py3/RelayLexer.py) - - set(JAVA_HOME $ENV{JAVA_HOME}) - if (NOT DEFINED JAVA_HOME) - # Hack to get system to search for Java itself. - set(JAVA_HOME "/usr") - endif() - - # Generate ANTLR grammar for parsing. - add_custom_command(OUTPUT ${RELAY_PARSER} - COMMAND ${JAVA_HOME}/bin/java -jar ${ANTLR4} -visitor -no-listener -Dlanguage=Python2 ${RELAY_PARSER_DIR}/Relay.g4 -o ${RELAY_PARSER_DIR}/py2 - COMMAND ${JAVA_HOME}/bin/java -jar ${ANTLR4} -visitor -no-listener -Dlanguage=Python3 ${RELAY_PARSER_DIR}/Relay.g4 -o ${RELAY_PARSER_DIR}/py3 - DEPENDS ${RELAY_PARSER_DIR}/Relay.g4 - WORKING_DIRECTORY ${RELAY_PARSER_DIR}) - - add_custom_target(relay_parser ALL DEPENDS ${RELAY_PARSER}) - else() - message(FATAL_ERROR "Can't find ANTLR4: ANTLR4=" ${ANTLR4}) - endif() + add_custom_target(relay_parser ALL DEPENDS ${RELAY_PARSER}) endif(USE_ANTLR) diff --git a/python/tvm/relay/_parser.py b/python/tvm/relay/_parser.py index 303f694..3bb05b0 100644 --- a/python/tvm/relay/_parser.py +++ b/python/tvm/relay/_parser.py @@ -43,14 +43,9 @@ class ParseError(Exception): PYTHON_VERSION = sys.version_info.major try: - if PYTHON_VERSION == 2: - from .grammar.py2.RelayVisitor import RelayVisitor - from .grammar.py2.RelayParser import RelayParser - from .grammar.py2.RelayLexer import RelayLexer - else: - from .grammar.py3.RelayVisitor import RelayVisitor - from .grammar.py3.RelayParser import RelayParser - from .grammar.py3.RelayLexer import RelayLexer + from .grammar.py3.RelayVisitor import RelayVisitor + from .grammar.py3.RelayParser import RelayParser + from .grammar.py3.RelayLexer import RelayLexer except ImportError: raise ParseError("Couldn't find ANTLR parser. Try building with USE_ANTLR=ON.") diff --git a/python/tvm/relay/grammar/py2/.gitattributes b/python/tvm/relay/grammar/py2/.gitattributes deleted file mode 100644 index 4adf65f..0000000 --- a/python/tvm/relay/grammar/py2/.gitattributes +++ /dev/null @@ -1,3 +0,0 @@ -Relay* binary -Relay* linguist-generated=true -Relay* linguist-detectable=false \ No newline at end of file diff --git a/python/tvm/relay/grammar/py2/Relay.interp b/python/tvm/relay/grammar/py2/Relay.interp deleted file mode 100644 index c6893d0..0000000 --- a/python/tvm/relay/grammar/py2/Relay.interp +++ /dev/null @@ -1,109 +0,0 @@ -token literal names: -null -'(' -')' -',' -'[' -']' -'if' -'else' -'let' -'=' -';' -'{' -'}' -'fn' -'->' -'def' -':' -'Tensor' -'_' -'v0.0.2' -null -null -null -'*' -'/' -'+' -'-' -'<' -'>' -'<=' -'>=' -'==' -'!=' -null -null -null -'mut' -null -null -null -null - -token symbolic names: -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -SEMVER -WS -LINE_COMMENT -COMMENT -MUL -DIV -ADD -SUB -LT -GT -LE -GE -EQ -NE -GLOBAL_VAR -LOCAL_VAR -GRAPH_VAR -MUT -BOOL_LIT -FLOAT -NAT -CNAME - -rule names: -opIdent -prog -expr -func -defn -argList -varList -var -attrList -attr -typeParamSeq -type_ -shapeSeq -shape -typeIdent -body -scalar -ident - - -atn: -[3, 24715, 42794, 33075, 47597, 16764, 15335, 30598, 22884, 3, 42, 332, 4, 2, 9, 2, 4, 3, 9, 3, 4, 4, 9, 4, 4, 5, 9, 5, 4, 6, 9, 6, 4, 7, 9, 7, 4, 8, 9, 8, 4, 9, 9, 9, 4, 10, 9, 10, 4, 11, 9, 11, 4, 12, 9, 12, 4, 13, 9, 13, 4, 14, 9, 14, 4, 15, 9, 15, 4, 16, 9, 16, 4, 17, 9, 17, 4, 18, 9, 18, 4, 19, 9, 19, 3, 2, 3, 2, 3, 3, 3, 3, 7, 3, 43, 10, 3, 12, 3, 14, 3, 46, 11, 3, 3, 3, 5, 3, 49, 10, 3, 3, 3, 3, 3, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 6, 4, 72, 10, 4, 13, 4, 14, 4, 73, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 7, 4, 82, 10, 4, 12, 4, 14, 4, 85, 11, 4, 5, 4, 87, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 5, 4, 100, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 5, 4, 110, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 5, 4, 128, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 7, 4, 150, 10, 4, 12, 4, 14, 4, 153, 11, 4, 5, 4, 155, 10, 4, 3, 4, 7, 4, 158, 10, 4, 12, 4, 14, 4, 161, 11, 4, 3, 5, 3, 5, 5, 5, 165, 10, 5, 3, 5, 3, 5, 3, 5, 3, 5, 3, 5, 5, 5, 172, 10, 5, 3, 5, 3, 5, 3, 6, 3, 6, 3, 6, 5, 6, 179, 10, 6, 3, 6, 3, 6, 3, 6, 3, 6, 3, 6, 5, 6, 186, 10, 6, 3, 6, 3, 6, 3, 7, 3, 7, 3, 7, 3, 7, 3, 7, 3, 7, 5, 7, 196, 10, 7, 3, 8, 3, 8, 3, 8, 7, 8, 201, 10, 8, 12, 8, 14, 8, 204, 11, 8, 5, 8, 206, 10, 8, 3, 9, 3, 9, 3, 9, 5, 9, 211, 10, 9, 3, 10, 3, 10, 3, 10, 7, 10, 216, 10, 10, 12, 10, 14, 10, 219, 11, 10, 5, 10, 221, 10, 10, 3, 11, 3, 11, 3, 11, 3, 11, 3, 12, 3, 12, 3, 12, 3, 12, 3, 12, 3, 12, 7, 12, 233, 10, 12, 12, 12, 14, 12, 236, 11, 12, 3, 12, 3, 12, 5, 12, 240, 10, 12, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 6, 13, 253, 10, 13, 13, 13, 14, 13, 254, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 5, 13, 269, 10, 13, 3, 13, 3, 13, 3, 13, 3, 13, 7, 13, 275, 10, 13, 12, 13, 14, 13, 278, 11, 13, 5, 13, 280, 10, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 5, 13, 287, 10, 13, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 6, 14, 300, 10, 14, 13, 14, 14, 14, 301, 3, 14, 3, 14, 5, 14, 306, 10, 14, 3, 15, 3, 15, 3, 15, 3, 15, 3, 15, 5, 15, 313, 10, 15, 3, 16, 3, 16, 3, 17, 3, 17, 3, 17, 3, 17, 3, 18, 3, 18, 3, 18, 5, 18, 324, 10, 18, 3, 19, 3, 19, 3, 19, 3, 19, 5, 19, 330, 10, 19, 3, 19, 2, 3, 6, 20, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 2, 6, 3, 2, 25, 26, 3, 2, 27, 28, 3, 2, 29, 32, 3, 2, 33, 34, 2, 373, 2, 38, 3, 2, 2, 2, 4, 40, 3, 2, 2, 2, 6, 127, 3, 2, 2, 2, 8, 162, 3, 2, 2, 2, 10, 175, 3, 2, 2, 2, 12, 195, 3, 2, 2, 2, 14, 205, 3, 2, 2, 2, 16, 207, 3, 2, 2, 2, 18, 220, 3, 2, 2, 2, 20, 222, 3, 2, 2, 2, 22, 239, 3, 2, 2, 2, 24, 286, 3, 2, 2, 2, 26, 305, 3, 2, 2, 2, 28, 312, 3, 2, 2, 2, 30, 314, 3, 2, 2, 2, 32, 316, 3, 2, 2, 2, 34, 323, 3, 2, 2, 2, 36, 329, 3, 2, 2, 2, 38, 39, 7, 42, 2, 2, 39, 3, 3, 2, 2, 2, 40, 48, 7, 21, 2, 2, 41, 43, 5, 10, 6, 2, 42, 41, 3, 2, 2, 2, 43, 46, 3, 2, 2, 2, 44, 42, 3, 2, 2, 2, 44, 45, 3, 2, 2, 2, 45, 49, 3, 2, 2, 2, 46, 44, 3, 2, 2, 2, 47, 49, 5, 6, 4, 2, 48, 44, 3, 2, 2, 2, 48, 47, 3, 2, 2, 2, 49, 50, 3, 2, 2, 2, 50, 51, 7, 2, 2, 3, 51, 5, 3, 2, 2, 2, 52, 53, 8, 4, 1, 2, 53, 54, 7, 3, 2, 2, 54, 55, 5, 6, 4, 2, 55, 56, 7, 4, 2, 2, 56, 128, 3, 2, 2, 2, 57, 58, 7, 28, 2, 2, 58, 128, 5, 6, 4, 19, 59, 128, 5, 8, 5, 2, 60, 61, 7, 3, 2, 2, 61, 128, 7, 4, 2, 2, 62, 63, 7, 3, 2, 2, 63, 64, 5, 6, 4, 2, 64, 65, 7, 5, 2, 2, 65, 66, 7, 4, 2, 2, 66, 128, 3, 2, 2, 2, 67, 68, 7, 3, 2, 2, 68, 71, 5, 6, 4, 2, 69, 70, 7, 5, 2, 2, 70, 72, 5, 6, 4, 2, 71, 69, 3, 2, 2, 2, 72, 73, 3, 2, 2, 2, 73, 71, 3, 2, 2, 2, 73, 74, 3, 2, 2, 2, 74, 75, 3, 2, 2, 2, 75, 76, 7, 4, 2, 2, 76, 128, 3, 2, 2, 2, 77, 86, 7, 6, 2, 2, 78, 83, 5, 6, 4, 2, 79, 80, 7, 5, 2, 2, 80, 82, 5, 6, 4, 2, 81, 79, 3, 2, 2, 2, 82, 85, 3, 2, 2, 2, 83, 81, 3, 2, 2, 2, 83, 84, 3, 2, 2, 2, 84, 87, 3, 2, 2, 2, 85, 83, 3, 2, 2, 2, 86, 78, 3, 2, 2, 2, 86, 87, 3, 2, 2, 2, 87, 88, 3, 2, 2, 2, 88, 128, 7, 7, 2, 2, 89, 90, 7, 8, 2, 2, 90, 91, 7, 3, 2, 2, 91, 92, 5, 6, 4, 2, 92, 93, 7, 4, 2, 2, 93, 94, 5, 32, 17, 2, 94, 95, 7, 9, 2, 2, 95, 96, 5, 32, 17, 2, 96, 128, 3, 2, 2, 2, 97, 99, 7, 10, 2, 2, 98, 100, 7, 38, 2, 2, 99, 98, 3, 2, 2, 2, 99, 100, 3, 2, 2, 2, 100, 101, 3, 2, 2, 2, 101, 102, 5, 16, 9, 2, 102, 103, 7, 11, 2, 2, 103, 104, 5, 6, 4, 2, 104, 105, 7, 12, 2, 2, 105, 106, 5, 6, 4, 8, 106, 128, 3, 2, 2, 2, 107, 109, 7, 10, 2, 2, 108, 110, 7, 38, 2, 2, 109, 108, 3, 2, 2, 2, 109, 110, 3, 2, 2, 2, 110, 111, 3, 2, 2, 2, 111, 112, 5, 16, 9, 2, 112, 113, 7, 11, 2, 2, 113, 114, 7, 13, 2, 2, 114, 115, 5, 6, 4, 2, 115, 116, 7, 14, 2, 2, 116, 117, 7, 12, 2, 2, 117, 118, 5, 6, 4, 7, 118, 128, 3, 2, 2, 2, 119, 120, 5, 36, 19, 2, 120, 121, 7, 11, 2, 2, 121, 122, 5, 6, 4, 2, 122, 123, 7, 12, 2, 2, 123, 124, 5, 6, 4, 5, 124, 128, 3, 2, 2, 2, 125, 128, 5, 36, 19, 2, 126, 128, 5, 34, 18, 2, 127, 52, 3, 2, 2, 2, 127, 57, 3, 2, 2, 2, 127, 59, 3, 2, 2, 2, 127, 60, 3, 2, 2, 2, 127, 62, 3, 2, 2, 2, 127, 67, 3, 2, 2, 2, 127, 77, 3, 2, 2, 2, 127, 89, 3, 2, 2, 2, 127, 97, 3, 2, 2, 2, 127, 107, 3, 2, 2, 2, 127, 119, 3, 2, 2, 2, 127, 125, 3, 2, 2, 2, 127, 126, 3, 2, 2, 2, 128, 159, 3, 2, 2, 2, 129, 130, 12, 18, 2, 2, 130, 131, 9, 2, 2, 2, 131, 158, 5, 6, 4, 19, 132, 133, 12, 17, 2, 2, 133, 134, 9, 3, 2, 2, 134, 158, 5, 6, 4, 18, 135, 136, 12, 16, 2, 2, 136, 137, 9, 4, 2, 2, 137, 158, 5, 6, 4, 17, 138, 139, 12, 15, 2, 2, 139, 140, 9, 5, 2, 2, 140, 158, 5, 6, 4, 16, 141, 142, 12, 6, 2, 2, 142, 143, 7, 12, 2, 2, 143, 158, 5, 6, 4, 7, 144, 145, 12, 20, 2, 2, 145, 154, 7, 3, 2, 2, 146, 151, 5, 6, 4, 2, 147, 148, 7, 5, 2, 2, 148, 150, 5, 6, 4, 2, 149, 147, 3, 2, 2, 2, 150, 153, 3, 2, 2, 2, 151, 149, 3, 2, 2, 2, 151, 152, 3, 2, 2, 2, 152, 155, 3, 2, 2, 2, 153, 151, 3, 2, 2, 2, 154, 146, 3, 2, 2, 2, 154, 155, 3, 2, 2, 2, 155, 156, 3, 2, 2, 2, 156, 158, 7, 4, 2, 2, 157, 129, 3, 2, 2, 2, 157, 132, 3, 2, 2, 2, 157, 135, 3, 2, 2, 2, 157, 138, 3, 2, 2, 2, 157, 141, 3, 2, 2, 2, 157, 144, 3, 2, 2, 2, 158, 161, 3, 2, 2, 2, 159, 157, 3, 2, 2, 2, 159, 160, 3, 2, 2, 2, 160, 7, 3, 2, 2, 2, 161, 159, 3, 2, 2, 2, 162, 164, 7, 15, 2, 2, 163, 165, 5, 22, 12, 2, 164, 163, 3, 2, 2, 2, 164, 165, 3, 2, 2, 2, 165, 166, 3, 2, 2, 2, 166, 167, 7, 3, 2, 2, 167, 168, 5, 12, 7, 2, 168, 171, 7, 4, 2, 2, 169, 170, 7, 16, 2, 2, 170, 172, 5, 24, 13, 2, 171, 169, 3, 2, 2, 2, 171, 172, 3, 2, 2, 2, 172, 173, 3, 2, 2, 2, 173, 174, 5, 32, 17, 2, 174, 9, 3, 2, 2, 2, 175, 176, 7, 17, 2, 2, 176, 178, 5, 36, 19, 2, 177, 179, 5, 22, 12, 2, 178, 177, 3, 2, 2, 2, 178, 179, 3, 2, 2, 2, 179, 180, 3, 2, 2, 2, 180, 181, 7, 3, 2, 2, 181, 182, 5, 12, 7, 2, 182, 185, 7, 4, 2, 2, 183, 184, 7, 16, 2, 2, 184, 186, 5, 24, 13, 2, 185, 183, 3, 2, 2, 2, 185, 186, 3, 2, 2, 2, 186, 187, 3, 2, 2, 2, 187, 188, 5, 32, 17, 2, 188, 11, 3, 2, 2, 2, 189, 196, 5, 14, 8, 2, 190, 196, 5, 18, 10, 2, 191, 192, 5, 14, 8, 2, 192, 193, 7, 5, 2, 2, 193, 194, 5, 18, 10, 2, 194, 196, 3, 2, 2, 2, 195, 189, 3, 2, 2, 2, 195, 190, 3, 2, 2, 2, 195, 191, 3, 2, 2, 2, 196, 13, 3, 2, 2, 2, 197, 202, 5, 16, 9, 2, 198, 199, 7, 5, 2, 2, 199, 201, 5, 16, 9, 2, 200, 198, 3, 2, 2, 2, 201, 204, 3, 2, 2, 2, 202, 200, 3, 2, 2, 2, 202, 203, 3, 2, 2, 2, 203, 206, 3, 2, 2, 2, 204, 202, 3, 2, 2, 2, 205, 197, 3, 2, 2, 2, 205, 206, 3, 2, 2, 2, 206, 15, 3, 2, 2, 2, 207, 210, 5, 36, 19, 2, 208, 209, 7, 18, 2, 2, 209, 211, 5, 24, 13, 2, 210, 208, 3, 2, 2, 2, 210, 211, 3, 2, 2, 2, 211, 17, 3, 2, 2, 2, 212, 217, 5, 20, 11, 2, 213, 214, 7, 5, 2, 2, 214, 216, 5, 20, 11, 2, 215, 213, 3, 2, 2, 2, 216, 219, 3, 2, 2, 2, 217, 215, 3, 2, 2, 2, 217, 218, 3, 2, 2, 2, 218, 221, 3, 2, 2, 2, 219, 217, 3, 2, 2, 2, 220, 212, 3, 2, 2, 2, 220, 221, 3, 2, 2, 2, 221, 19, 3, 2, 2, 2, 222, 223, 7, 42, 2, 2, 223, 224, 7, 11, 2, 2, 224, 225, 5, 6, 4, 2, 225, 21, 3, 2, 2, 2, 226, 227, 7, 6, 2, 2, 227, 240, 7, 7, 2, 2, 228, 229, 7, 6, 2, 2, 229, 234, 5, 36, 19, 2, 230, 231, 7, 5, 2, 2, 231, 233, 5, 36, 19, 2, 232, 230, 3, 2, 2, 2, 233, 236, 3, 2, 2, 2, 234, 232, 3, 2, 2, 2, 234, 235, 3, 2, 2, 2, 235, 237, 3, 2, 2, 2, 236, 234, 3, 2, 2, 2, 237, 238, 7, 7, 2, 2, 238, 240, 3, 2, 2, 2, 239, 226, 3, 2, 2, 2, 239, 228, 3, 2, 2, 2, 240, 23, 3, 2, 2, 2, 241, 242, 7, 3, 2, 2, 242, 287, 7, 4, 2, 2, 243, 244, 7, 3, 2, 2, 244, 245, 5, 24, 13, 2, 245, 246, 7, 5, 2, 2, 246, 247, 7, 4, 2, 2, 247, 287, 3, 2, 2, 2, 248, 249, 7, 3, 2, 2, 249, 252, 5, 24, 13, 2, 250, 251, 7, 5, 2, 2, 251, 253, 5, 24, 13, 2, 252, 250, 3, 2, 2, 2, 253, 254, 3, 2, 2, 2, 254, 252, 3, 2, 2, 2, 254, 255, 3, 2, 2, 2, 255, 256, 3, 2, 2, 2, 256, 257, 7, 4, 2, 2, 257, 287, 3, 2, 2, 2, 258, 287, 5, 30, 16, 2, 259, 260, 7, 19, 2, 2, 260, 261, 7, 6, 2, 2, 261, 262, 5, 26, 14, 2, 262, 263, 7, 5, 2, 2, 263, 264, 5, 24, 13, 2, 264, 265, 7, 7, 2, 2, 265, 287, 3, 2, 2, 2, 266, 268, 7, 15, 2, 2, 267, 269, 5, 22, 12, 2, 268, 267, 3, 2, 2, 2, 268, 269, 3, 2, 2, 2, 269, 270, 3, 2, 2, 2, 270, 279, 7, 3, 2, 2, 271, 276, 5, 24, 13, 2, 272, 273, 7, 5, 2, 2, 273, 275, 5, 24, 13, 2, 274, 272, 3, 2, 2, 2, 275, 278, 3, 2, 2, 2, 276, 274, 3, 2, 2, 2, 276, 277, 3, 2, 2, 2, 277, 280, 3, 2, 2, 2, 278, 276, 3, 2, 2, 2, 279, 271, 3, 2, 2, 2, 279, 280, 3, 2, 2, 2, 280, 281, 3, 2, 2, 2, 281, 282, 7, 4, 2, 2, 282, 283, 7, 16, 2, 2, 283, 287, 5, 24, 13, 2, 284, 287, 7, 20, 2, 2, 285, 287, 7, 41, 2, 2, 286, 241, 3, 2, 2, 2, 286, 243, 3, 2, 2, 2, 286, 248, 3, 2, 2, 2, 286, 258, 3, 2, 2, 2, 286, 259, 3, 2, 2, 2, 286, 266, 3, 2, 2, 2, 286, 284, 3, 2, 2, 2, 286, 285, 3, 2, 2, 2, 287, 25, 3, 2, 2, 2, 288, 289, 7, 3, 2, 2, 289, 306, 7, 4, 2, 2, 290, 291, 7, 3, 2, 2, 291, 292, 5, 28, 15, 2, 292, 293, 7, 5, 2, 2, 293, 294, 7, 4, 2, 2, 294, 306, 3, 2, 2, 2, 295, 296, 7, 3, 2, 2, 296, 299, 5, 28, 15, 2, 297, 298, 7, 5, 2, 2, 298, 300, 5, 28, 15, 2, 299, 297, 3, 2, 2, 2, 300, 301, 3, 2, 2, 2, 301, 299, 3, 2, 2, 2, 301, 302, 3, 2, 2, 2, 302, 303, 3, 2, 2, 2, 303, 304, 7, 4, 2, 2, 304, 306, 3, 2, 2, 2, 305, 288, 3, 2, 2, 2, 305, 290, 3, 2, 2, 2, 305, 295, 3, 2, 2, 2, 306, 27, 3, 2, 2, 2, 307, 308, 7, 3, 2, 2, 308, 309, 5, 28, 15, 2, 309, 310, 7, 4, 2, 2, 310, 313, 3, 2, 2, 2, 311, 313, 7, 41, 2, 2, 312, 307, 3, 2, 2, 2, 312, 311, 3, 2, 2, 2, 313, 29, 3, 2, 2, 2, 314, 315, 7, 42, 2, 2, 315, 31, 3, 2, 2, 2, 316, 317, 7, 13, 2, 2, 317, 318, 5, 6, 4, 2, 318, 319, 7, 14, 2, 2, 319, 33, 3, 2, 2, 2, 320, 324, 7, 40, 2, 2, 321, 324, 7, 41, 2, 2, 322, 324, 7, 39, 2, 2, 323, 320, 3, 2, 2, 2, 323, 321, 3, 2, 2, 2, 323, 322, 3, 2, 2, 2, 324, 35, 3, 2, 2, 2, 325, 330, 5, 2, 2, 2, 326, 330, 7, 35, 2, 2, 327, 330, 7, 36, 2, 2, 328, 330, 7, 37, 2, 2, 329, 325, 3, 2, 2, 2, 329, 326, 3, 2, 2, 2, 329, 327, 3, 2, 2, 2, 329, 328, 3, 2, 2, 2, 330, 37, 3, 2, 2, 2, 36, 44, 48, 73, 83, 86, 99, 109, 127, 151, 154, 157, 159, 164, 171, 178, 185, 195, 202, 205, 210, 217, 220, 234, 239, 254, 268, 276, 279, 286, 301, 305, 312, 323, 329] \ No newline at end of file diff --git a/python/tvm/relay/grammar/py2/Relay.tokens b/python/tvm/relay/grammar/py2/Relay.tokens deleted file mode 100644 index 41f3ee6..0000000 --- a/python/tvm/relay/grammar/py2/Relay.tokens +++ /dev/null @@ -1,70 +0,0 @@ -T__0=1 -T__1=2 -T__2=3 -T__3=4 -T__4=5 -T__5=6 -T__6=7 -T__7=8 -T__8=9 -T__9=10 -T__10=11 -T__11=12 -T__12=13 -T__13=14 -T__14=15 -T__15=16 -T__16=17 -T__17=18 -SEMVER=19 -WS=20 -LINE_COMMENT=21 -COMMENT=22 -MUL=23 -DIV=24 -ADD=25 -SUB=26 -LT=27 -GT=28 -LE=29 -GE=30 -EQ=31 -NE=32 -GLOBAL_VAR=33 -LOCAL_VAR=34 -GRAPH_VAR=35 -MUT=36 -BOOL_LIT=37 -FLOAT=38 -NAT=39 -CNAME=40 -'('=1 -')'=2 -','=3 -'['=4 -']'=5 -'if'=6 -'else'=7 -'let'=8 -'='=9 -';'=10 -'{'=11 -'}'=12 -'fn'=13 -'->'=14 -'def'=15 -':'=16 -'Tensor'=17 -'_'=18 -'v0.0.2'=19 -'*'=23 -'/'=24 -'+'=25 -'-'=26 -'<'=27 -'>'=28 -'<='=29 -'>='=30 -'=='=31 -'!='=32 -'mut'=36 diff --git a/python/tvm/relay/grammar/py2/RelayLexer.interp b/python/tvm/relay/grammar/py2/RelayLexer.interp deleted file mode 100644 index 092b358..0000000 --- a/python/tvm/relay/grammar/py2/RelayLexer.interp +++ /dev/null @@ -1,140 +0,0 @@ -token literal names: -null -'(' -')' -',' -'[' -']' -'if' -'else' -'let' -'=' -';' -'{' -'}' -'fn' -'->' -'def' -':' -'Tensor' -'_' -'v0.0.2' -null -null -null -'*' -'/' -'+' -'-' -'<' -'>' -'<=' -'>=' -'==' -'!=' -null -null -null -'mut' -null -null -null -null - -token symbolic names: -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -null -SEMVER -WS -LINE_COMMENT -COMMENT -MUL -DIV -ADD -SUB -LT -GT -LE -GE -EQ -NE -GLOBAL_VAR -LOCAL_VAR -GRAPH_VAR -MUT -BOOL_LIT -FLOAT -NAT -CNAME - -rule names: -T__0 -T__1 -T__2 -T__3 -T__4 -T__5 -T__6 -T__7 -T__8 -T__9 -T__10 -T__11 -T__12 -T__13 -T__14 -T__15 -T__16 -T__17 -SEMVER -WS -LINE_COMMENT -COMMENT -MUL -DIV -ADD -SUB -LT -GT -LE -GE -EQ -NE -GLOBAL_VAR -LOCAL_VAR -GRAPH_VAR -MUT -BOOL_LIT -FLOAT -NAT -EXP -CNAME -LETTER -DIGIT - -channel names: -DEFAULT_TOKEN_CHANNEL -HIDDEN - -mode names: -DEFAULT_MODE - -atn: -[3, 24715, 42794, 33075, 47597, 16764, 15335, 30598, 22884, 2, 42, 267, 8, 1, 4, 2, 9, 2, 4, 3, 9, 3, 4, 4, 9, 4, 4, 5, 9, 5, 4, 6, 9, 6, 4, 7, 9, 7, 4, 8, 9, 8, 4, 9, 9, 9, 4, 10, 9, 10, 4, 11, 9, 11, 4, 12, 9, 12, 4, 13, 9, 13, 4, 14, 9, 14, 4, 15, 9, 15, 4, 16, 9, 16, 4, 17, 9, 17, 4, 18, 9, 18, 4, 19, 9, 19, 4, 20, 9, 20, 4, 21, 9, 21, 4, 22, 9, 22, 4, 23, 9, 23, 4, 24, 9, 24, 4, 25, 9, 25, 4, 26, 9, 26, 4, 27, 9, 27, 4, 28, 9, 28, 4, 29, 9, 29, 4, 30, 9, 30, 4, 31, 9, 31, 4, 32, 9, 32, 4, 33, 9, 33, 4, 34, 9, 34, 4, 35, 9, 35, 4, 36, 9, 36, 4, 37, 9, 37, 4, 38, 9, 38, 4, 39, 9, 39, 4, 40, 9, 40, 4, 41, 9, 41, 4, 42, 9, 42, 4, 43, 9, 43, 4, 44, 9, 44, 3, 2, 3, 2, 3, 3, 3, 3, 3, 4, 3, 4, 3, 5, 3, 5, 3, 6, 3, 6, 3, 7, 3, 7, 3, 7, 3, 8, 3, 8, 3, 8, 3, 8, 3, 8, 3, 9, 3, 9, 3, 9, 3, 9, 3, 10, 3, 10, 3, 11, 3, 11, 3, 12, 3, 12, 3, 13, 3, 13, 3, 14, 3, 14, 3, 14, 3, 15, 3, 15, 3, 15, 3, 16, 3, 16, 3, 16, 3, 16, 3, 17, 3, 17, 3, 18, 3, 18, 3, 18, 3, 18, 3, 18, 3, 18, 3, 18, 3, 19, 3, 19, 3, 20, 3, 20, 3, 20, 3, 20, 3, 20, 3, 20, 3, 20, 3, 21, 6, 21, 149, 10, 21, 13, 21, 14, 21, 150, 3, 21, 3, 21, 3, 22, 3, 22, 3, 22, 3, 22, 7, 22, 159, 10, 22, 12, 22, 14, 22, 162, 11, 22, 3, 22, 3, 22, 3, 22, 3, 22, 3, 23, 3, 23, 3, 23, 3, 23, 7, 23, 172, 10, 23, 12, 23, 14, 23, 175, 11, 23, 3, 23, 3, 23, 3, 23, 3, 23, 3, 23, 3, 24, 3, 24, 3, 25, 3, 25, 3, 26, 3, 26, 3, 27, 3, 27, 3, 28, 3, 28, 3, 29, 3, 29, 3, 30, 3, 30, 3, 30, 3, 31, 3, 31, 3, 31, 3, 32, 3, 32, 3, 32, 3, 33, 3, 33, 3, 33, 3, 34, 3, 34, 3, 34, 3, 35, 3, 35, 3, 35, 3, 36, 3, 36, 3, 36, 3, 37, 3, 37, 3, 37, 3, 37, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 5, 38, 228, 10, 38, 3, 39, 3, 39, 3, 39, 3, 39, 5, 39, 234, 10, 39, 3, 39, 3, 39, 3, 39, 5, 39, 239, 10, 39, 3, 40, 6, 40, 242, 10, 40, 13, 40, 14, 40, 243, 3, 41, 3, 41, 5, 41, 248, 10, 41, 3, 41, 3, 41, 3, 42, 3, 42, 5, 42, 254, 10, 42, 3, 42, 3, 42, 3, 42, 7, 42, 259, 10, 42, 12, 42, 14, 42, 262, 11, 42, 3, 43, 3, 43, 3, 44, 3, 44, 4, 160, 173, 2, 45, 3, 3, 5, 4, 7, 5, 9, 6, 11, 7, 13, 8, 15, 9, 17, 10, 19, 11, 21, 12, 23, 13, 25, 14, 27, 15, 29, 16, 31, 17, 33, 18, 35, 19, 37, 20, 39, 21, 41, 22, 43, 23, 45, 24, 47, 25, 49, 26, 51, 27, 53, 28, 55, 29, 57, 30, 59, 31, 61, 32, 63, 33, 65, 34, 67, 35, 69, 36, 71, 37, 73, 38, 75, 39, 77, 40, 79, 41, 81, 2, 83, 42, 85, 2, 87, 2, 3, 2, 7, 5, 2, 11, 12, 15, 15, 34, 34, 4, 2, 71, 71, 103, 103, 4, 2, 45, 45, 47, 47, 4, 2, 67, 92, 99, 124, 3, 2, 50, 59, 2, 275, 2, 3, 3, 2, 2, 2, 2, 5, 3, 2, 2, 2, 2, 7, 3, 2, 2, 2, 2, 9, 3, 2, 2, 2, 2, 11, 3, 2, 2, 2, 2, 13, 3, 2, 2, 2, 2, 15, 3, 2, 2, 2, 2, 17, 3, 2, 2, 2, 2, 19, 3, 2, 2, 2, 2, 21, 3, 2, 2, 2, 2, 23, 3, 2, 2, 2, 2, 25, 3, 2, 2, 2, 2, 27, 3, 2, 2, 2, 2, 29, 3, 2, 2, 2, 2, 31, 3, 2, 2, 2, 2, 33, 3, 2, 2, 2, 2, 35, 3, 2, 2, 2, 2, 37, 3, 2, 2, 2, 2, 39, 3, 2, 2, 2, 2, 41, 3, 2, 2, 2, 2, 43, 3, 2, 2, 2, 2, 45, 3, 2, 2, 2, 2, 47, 3, 2, 2, 2, 2, 49, 3, 2, 2, 2, 2, 51, 3, 2, 2, 2, 2, 53, 3, 2, 2, 2, 2, 55, 3, 2, 2, 2, 2, 57, 3, 2, 2, 2, 2, 59, 3, 2, 2, 2, 2, 61, 3, 2, 2, 2, 2, 63, 3, 2, 2, 2, 2, 65, 3, 2, 2, 2, 2, 67, 3, 2, 2, 2, 2, 69, 3, 2, 2, 2, 2, 71, 3, 2, 2, 2, 2, 73, 3, 2, 2, 2, 2, 75, 3, 2, 2, 2, 2, 77, 3, 2, 2, 2, 2, 79, 3, 2, 2, 2, 2, 83, 3, 2, 2, 2, 3, 89, 3, 2, 2, 2, 5, 91, 3, 2, 2, 2, 7, 93, 3, 2, 2, 2, 9, 95, 3, 2, 2, 2, 11, 97, 3, 2, 2, 2, 13, 99, 3, 2, 2, 2, 15, 102, 3, 2, 2, 2, 17, 107, 3, 2, 2, 2, 19, 111, 3, 2, 2, 2, 21, 113, 3, 2, 2, 2, 23, 115, 3, 2, 2, 2, 25, 117, 3, 2, 2, 2, 27, 119, 3, 2, 2, 2, 29, 122, 3, 2, 2, 2, 31, 125, 3, 2, 2, 2, 33, 129, 3, 2, 2, 2, 35, 131, 3, 2, 2, 2, 37, 138, 3, 2, 2, 2, 39, 140, 3, 2, 2, 2, 41, 148, 3, 2, 2, 2, 43, 154, 3, 2, 2, 2, 45, 167, 3, 2, 2, 2, 47, 181, 3, 2, 2, 2, 49, 183, 3, 2, 2, 2, 51, 185, 3, 2, 2, 2, 53, 187, 3, 2, 2, 2, 55, 189, 3, 2, 2, 2, 57, 191, 3, 2, 2, 2, 59, 193, 3, 2, 2, 2, 61, 196, 3, 2, 2, 2, 63, 199, 3, 2, 2, 2, 65, 202, 3, 2, 2, 2, 67, 205, 3, 2, 2, 2, 69, 208, 3, 2, 2, 2, 71, 211, 3, 2, 2, 2, 73, 214, 3, 2, 2, 2, 75, 227, 3, 2, 2, 2, 77, 238, 3, 2, 2, 2, 79, 241, 3, 2, 2, 2, 81, 245, 3, 2, 2, 2, 83, 253, 3, 2, 2, 2, 85, 263, 3, 2, 2, 2, 87, 265, 3, 2, 2, 2, 89, 90, 7, 42, 2, 2, 90, 4, 3, 2, 2, 2, 91, 92, 7, 43, 2, 2, 92, 6, 3, 2, 2, 2, 93, 94, 7, 46, 2, 2, 94, 8, 3, 2, 2, 2, 95, 96, 7, 93, 2, 2, 96, 10, 3, 2, 2, 2, 97, 98, 7, 95, 2, 2, 98, 12, 3, 2, 2, 2, 99, 100, 7, 107, 2, 2, 100, 101, 7, 104, 2, 2, 101, 14, 3, 2, 2, 2, 102, 103, 7, 103, 2, 2, 103, 104, 7, 110, 2, 2, 104, 105, 7, 117, 2, 2, 105, 106, 7, 103, 2, 2, 106, 16, 3, 2, 2, 2, 107, 108, 7, 110, 2, 2, 108, 109, 7, 103, 2, 2, 109, 110, 7, 118, 2, 2, 110, 18, 3, 2, 2, 2, 111, 112, 7, 63, 2, 2, 112, 20, 3, 2, 2, 2, 113, 114, 7, 61, 2, 2, 114, 22, 3, 2, 2, 2, 115, 116, 7, 125, 2, 2, 116, 24, 3, 2, 2, 2, 117, 118, 7, 127, 2, 2, 118, 26, 3, 2, 2, 2, 119, 120, 7, 104, 2, 2, 120, 121, 7, 112, 2, 2, 121, 28, 3, 2, 2, 2, 122, 123, 7, 47, 2, 2, 123, 124, 7, 64, 2, 2, 124, 30, 3, 2, 2, 2, 125, 126, 7, 102, 2, 2, 126, 127, 7, 103, 2, 2, 127, 128, 7, 104, 2, 2, 128, 32, 3, 2, 2, 2, 129, 130, 7, 60, 2, 2, 130, 34, 3, 2, 2, 2, 131, 132, 7, 86, 2, 2, 132, 133, 7, 103, 2, 2, 133, 134, 7, 112, 2, 2, 134, 135, 7, 117, 2, 2, 135, 136, 7, 113, 2, 2, 136, 137, 7, 116, 2, 2, 137, 36, 3, 2, 2, 2, 138, 139, 7, 97, 2, 2, 139, 38, 3, 2, 2, 2, 140, 141, 7, 120, 2, 2, 141, 142, 7, 50, 2, 2, 142, 143, 7, 48, 2, 2, 143, 144, 7, 50, 2, 2, 144, 145, 7, 48, 2, 2, 145, 146, 7, 52, 2, 2, 146, 40, 3, 2, 2, 2, 147, 149, 9, 2, 2, 2, 148, 147, 3, 2, 2, 2, 149, 150, 3, 2, 2, 2, 150, 148, 3, 2, 2, 2, 150, 151, 3, 2, 2, 2, 151, 152, 3, 2, 2, 2, 152, 153, 8, 21, 2, 2, 153, 42, 3, 2, 2, 2, 154, 155, 7, 49, 2, 2, 155, 156, 7, 49, 2, 2, 156, 160, 3, 2, 2, 2, 157, 159, 11, 2, 2, 2, 158, 157, 3, 2, 2, 2, 159, 162, 3, 2, 2, 2, 160, 161, 3, 2, 2, 2, 160, 158, 3, 2, 2, 2, 161, 163, 3, 2, 2, 2, 162, 160, 3, 2, 2, 2, 163, 164, 7, 12, 2, 2, 164, 165, 3, 2, 2, 2, 165, 166, 8, 22, 2, 2, 166, 44, 3, 2, 2, 2, 167, 168, 7, 49, 2, 2, 168, 169, 7, 44, 2, 2, 169, 173, 3, 2, 2, 2, 170, 172, 11, 2, 2, 2, 171, 170, 3, 2, 2, 2, 172, 175, 3, 2, 2, 2, 173, 174, 3, 2, 2, 2, 173, 171, 3, 2, 2, 2, 174, 176, 3, 2, 2, 2, 175, 173, 3, 2, 2, 2, 176, 177, 7, 44, 2, 2, 177, 178, 7, 49, 2, 2, 178, 179, 3, 2, 2, 2, 179, 180, 8, 23, 2, 2, 180, 46, 3, 2, 2, 2, 181, 182, 7, 44, 2, 2, 182, 48, 3, 2, 2, 2, 183, 184, 7, 49, 2, 2, 184, 50, 3, 2, 2, 2, 185, 186, 7, 45, 2, 2, 186, 52, 3, 2, 2, 2, 187, 188, 7, 47, 2, 2, 188, 54, 3, 2, 2, 2, 189, 190, 7, 62, 2, 2, 190, 56, 3, 2, 2, 2, 191, 192, 7, 64, 2, 2, 192, 58, 3, 2, 2, 2, 193, 194, 7, 62, 2, 2, 194, 195, 7, 63, 2, 2, 195, 60, 3, 2, 2, 2, 196, 197, 7, 64, 2, 2, 197, 198, 7, 63, 2, 2, 198, 62, 3, 2, 2, 2, 199, 200, 7, 63, 2, 2, 200, 201, 7, 63, 2, 2, 201, 64, 3, 2, 2, 2, 202, 203, 7, 35, 2, 2, 203, 204, 7, 63, 2, 2, 204, 66, 3, 2, 2, 2, 205, 206, 7, 66, 2, 2, 206, 207, 5, 83, 42, 2, 207, 68, 3, 2, 2, 2, 208, 209, 7, 39, 2, 2, 209, 210, 5, 83, 42, 2, 210, 70, 3, 2, 2, 2, 211, 212, 7, 39, 2, 2, 212, 213, 5, 79, 40, 2, 213, 72, 3, 2, 2, 2, 214, 215, 7, 111, 2, 2, 215, 216, 7, 119, 2, 2, 216, 217, 7, 118, 2, 2, 217, 74, 3, 2, 2, 2, 218, 219, 7, 86, 2, 2, 219, 220, 7, 116, 2, 2, 220, 221, 7, 119, 2, 2, 221, 228, 7, 103, 2, 2, 222, 223, 7, 72, 2, 2, 223, 224, 7, 99, 2, 2, 224, 225, 7, 110, 2, 2, 225, 226, 7, 117, 2, 2, 226, 228, 7, 103, 2, 2, 227, 218, 3, 2, 2, 2, 227, 222, 3, 2, 2, 2, 228, 76, 3, 2, 2, 2, 229, 230, 5, 79, 40, 2, 230, 231, 7, 48, 2, 2, 231, 233, 5, 79, 40, 2, 232, 234, 5, 81, 41, 2, 233, 232, 3, 2, 2, 2, 233, 234, 3, 2, 2, 2, 234, 239, 3, 2, 2, 2, 235, 236, 5, 79, 40, 2, 236, 237, 5, 81, 41, 2, 237, 239, 3, 2, 2, 2, 238, 229, 3, 2, 2, 2, 238, 235, 3, 2, 2, 2, 239, 78, 3, 2, 2, 2, 240, 242, 5, 87, 44, 2, 241, 240, 3, 2, 2, 2, 242, 243, 3, 2, 2, 2, 243, 241, 3, 2, 2, 2, 243, 244, 3, 2, 2, 2, 244, 80, 3, 2, 2, 2, 245, 247, 9, 3, 2, 2, 246, 248, 9, 4, 2, 2, 247, 246, 3, 2, 2, 2, 247, 248, 3, 2, 2, 2, 248, 249, 3, 2, 2, 2, 249, 250, 5, 79, 40, 2, 250, 82, 3, 2, 2, 2, 251, 254, 7, 97, 2, 2, 252, 254, 5, 85, 43, 2, 253, 251, 3, 2, 2, 2, 253, 252, 3, 2, 2, 2, 254, 260, 3, 2, 2, 2, 255, 259, 7, 97, 2, 2, 256, 259, 5, 85, 43, 2, 257, 259, 5, 87, 44, 2, 258, 255, 3, 2, 2, 2, 258, 256, 3, 2, 2, 2, 258, 257, 3, 2, 2, 2, 259, 262, 3, 2, 2, 2, 260, 258, 3, 2, 2, 2, 260, 261, 3, 2, 2, 2, 261, 84, 3, 2, 2, 2, 262, 260, 3, 2, 2, 2, 263, 264, 9, 5, 2, 2, 264, 86, 3, 2, 2, 2, 265, 266, 9, 6, 2, 2, 266, 88, 3, 2, 2, 2, 14, 2, 150, 160, 173, 227, 233, 238, 243, 247, 253, 258, 260, 3, 8, 2, 2] \ No newline at end of file diff --git a/python/tvm/relay/grammar/py2/RelayLexer.py b/python/tvm/relay/grammar/py2/RelayLexer.py deleted file mode 100644 index be87421..0000000 --- a/python/tvm/relay/grammar/py2/RelayLexer.py +++ /dev/null @@ -1,209 +0,0 @@ -# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2 -# encoding: utf-8 -from __future__ import print_function -from antlr4 import * -from io import StringIO -import sys - - - -def serializedATN(): - with StringIO() as buf: - buf.write(u"\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\2") - buf.write(u"*\u010b\b\1\4\2\t\2\4\3\t\3\4\4\t\4\4\5\t\5\4\6\t\6\4") - buf.write(u"\7\t\7\4\b\t\b\4\t\t\t\4\n\t\n\4\13\t\13\4\f\t\f\4\r") - buf.write(u"\t\r\4\16\t\16\4\17\t\17\4\20\t\20\4\21\t\21\4\22\t\22") - buf.write(u"\4\23\t\23\4\24\t\24\4\25\t\25\4\26\t\26\4\27\t\27\4") - buf.write(u"\30\t\30\4\31\t\31\4\32\t\32\4\33\t\33\4\34\t\34\4\35") - buf.write(u"\t\35\4\36\t\36\4\37\t\37\4 \t \4!\t!\4\"\t\"\4#\t#\4") - buf.write(u"$\t$\4%\t%\4&\t&\4\'\t\'\4(\t(\4)\t)\4*\t*\4+\t+\4,\t") - buf.write(u",\3\2\3\2\3\3\3\3\3\4\3\4\3\5\3\5\3\6\3\6\3\7\3\7\3\7") - buf.write(u"\3\b\3\b\3\b\3\b\3\b\3\t\3\t\3\t\3\t\3\n\3\n\3\13\3\13") - buf.write(u"\3\f\3\f\3\r\3\r\3\16\3\16\3\16\3\17\3\17\3\17\3\20\3") - buf.write(u"\20\3\20\3\20\3\21\3\21\3\22\3\22\3\22\3\22\3\22\3\22") - buf.write(u"\3\22\3\23\3\23\3\24\3\24\3\24\3\24\3\24\3\24\3\24\3") - buf.write(u"\25\6\25\u0095\n\25\r\25\16\25\u0096\3\25\3\25\3\26\3") - buf.write(u"\26\3\26\3\26\7\26\u009f\n\26\f\26\16\26\u00a2\13\26") - buf.write(u"\3\26\3\26\3\26\3\26\3\27\3\27\3\27\3\27\7\27\u00ac\n") - buf.write(u"\27\f\27\16\27\u00af\13\27\3\27\3\27\3\27\3\27\3\27\3") - buf.write(u"\30\3\30\3\31\3\31\3\32\3\32\3\33\3\33\3\34\3\34\3\35") - buf.write(u"\3\35\3\36\3\36\3\36\3\37\3\37\3\37\3 \3 \3 \3!\3!\3") - buf.write(u"!\3\"\3\"\3\"\3#\3#\3#\3$\3$\3$\3%\3%\3%\3%\3&\3&\3&") - buf.write(u"\3&\3&\3&\3&\3&\3&\5&\u00e4\n&\3\'\3\'\3\'\3\'\5\'\u00ea") - buf.write(u"\n\'\3\'\3\'\3\'\5\'\u00ef\n\'\3(\6(\u00f2\n(\r(\16(") - buf.write(u"\u00f3\3)\3)\5)\u00f8\n)\3)\3)\3*\3*\5*\u00fe\n*\3*\3") - buf.write(u"*\3*\7*\u0103\n*\f*\16*\u0106\13*\3+\3+\3,\3,\4\u00a0") - buf.write(u"\u00ad\2-\3\3\5\4\7\5\t\6\13\7\r\b\17\t\21\n\23\13\25") - buf.write(u"\f\27\r\31\16\33\17\35\20\37\21!\22#\23%\24\'\25)\26") - buf.write(u"+\27-\30/\31\61\32\63\33\65\34\67\359\36;\37= ?!A\"C") - buf.write(u"#E$G%I&K\'M(O)Q\2S*U\2W\2\3\2\7\5\2\13\f\17\17\"\"\4") - buf.write(u"\2GGgg\4\2--//\4\2C\\c|\3\2\62;\2\u0113\2\3\3\2\2\2\2") - buf.write(u"\5\3\2\2\2\2\7\3\2\2\2\2\t\3\2\2\2\2\13\3\2\2\2\2\r\3") - buf.write(u"\2\2\2\2\17\3\2\2\2\2\21\3\2\2\2\2\23\3\2\2\2\2\25\3") - buf.write(u"\2\2\2\2\27\3\2\2\2\2\31\3\2\2\2\2\33\3\2\2\2\2\35\3") - buf.write(u"\2\2\2\2\37\3\2\2\2\2!\3\2\2\2\2#\3\2\2\2\2%\3\2\2\2") - buf.write(u"\2\'\3\2\2\2\2)\3\2\2\2\2+\3\2\2\2\2-\3\2\2\2\2/\3\2") - buf.write(u"\2\2\2\61\3\2\2\2\2\63\3\2\2\2\2\65\3\2\2\2\2\67\3\2") - buf.write(u"\2\2\29\3\2\2\2\2;\3\2\2\2\2=\3\2\2\2\2?\3\2\2\2\2A\3") - buf.write(u"\2\2\2\2C\3\2\2\2\2E\3\2\2\2\2G\3\2\2\2\2I\3\2\2\2\2") - buf.write(u"K\3\2\2\2\2M\3\2\2\2\2O\3\2\2\2\2S\3\2\2\2\3Y\3\2\2\2") - buf.write(u"\5[\3\2\2\2\7]\3\2\2\2\t_\3\2\2\2\13a\3\2\2\2\rc\3\2") - buf.write(u"\2\2\17f\3\2\2\2\21k\3\2\2\2\23o\3\2\2\2\25q\3\2\2\2") - buf.write(u"\27s\3\2\2\2\31u\3\2\2\2\33w\3\2\2\2\35z\3\2\2\2\37}") - buf.write(u"\3\2\2\2!\u0081\3\2\2\2#\u0083\3\2\2\2%\u008a\3\2\2\2") - buf.write(u"\'\u008c\3\2\2\2)\u0094\3\2\2\2+\u009a\3\2\2\2-\u00a7") - buf.write(u"\3\2\2\2/\u00b5\3\2\2\2\61\u00b7\3\2\2\2\63\u00b9\3\2") - buf.write(u"\2\2\65\u00bb\3\2\2\2\67\u00bd\3\2\2\29\u00bf\3\2\2\2") - buf.write(u";\u00c1\3\2\2\2=\u00c4\3\2\2\2?\u00c7\3\2\2\2A\u00ca") - buf.write(u"\3\2\2\2C\u00cd\3\2\2\2E\u00d0\3\2\2\2G\u00d3\3\2\2\2") - buf.write(u"I\u00d6\3\2\2\2K\u00e3\3\2\2\2M\u00ee\3\2\2\2O\u00f1") - buf.write(u"\3\2\2\2Q\u00f5\3\2\2\2S\u00fd\3\2\2\2U\u0107\3\2\2\2") - buf.write(u"W\u0109\3\2\2\2YZ\7*\2\2Z\4\3\2\2\2[\\\7+\2\2\\\6\3\2") - buf.write(u"\2\2]^\7.\2\2^\b\3\2\2\2_`\7]\2\2`\n\3\2\2\2ab\7_\2\2") - buf.write(u"b\f\3\2\2\2cd\7k\2\2de\7h\2\2e\16\3\2\2\2fg\7g\2\2gh") - buf.write(u"\7n\2\2hi\7u\2\2ij\7g\2\2j\20\3\2\2\2kl\7n\2\2lm\7g\2") - buf.write(u"\2mn\7v\2\2n\22\3\2\2\2op\7?\2\2p\24\3\2\2\2qr\7=\2\2") - buf.write(u"r\26\3\2\2\2st\7}\2\2t\30\3\2\2\2uv\7\177\2\2v\32\3\2") - buf.write(u"\2\2wx\7h\2\2xy\7p\2\2y\34\3\2\2\2z{\7/\2\2{|\7@\2\2") - buf.write(u"|\36\3\2\2\2}~\7f\2\2~\177\7g\2\2\177\u0080\7h\2\2\u0080") - buf.write(u" \3\2\2\2\u0081\u0082\7<\2\2\u0082\"\3\2\2\2\u0083\u0084") - buf.write(u"\7V\2\2\u0084\u0085\7g\2\2\u0085\u0086\7p\2\2\u0086\u0087") - buf.write(u"\7u\2\2\u0087\u0088\7q\2\2\u0088\u0089\7t\2\2\u0089$") - buf.write(u"\3\2\2\2\u008a\u008b\7a\2\2\u008b&\3\2\2\2\u008c\u008d") - buf.write(u"\7x\2\2\u008d\u008e\7\62\2\2\u008e\u008f\7\60\2\2\u008f") - buf.write(u"\u0090\7\62\2\2\u0090\u0091\7\60\2\2\u0091\u0092\7\64") - buf.write(u"\2\2\u0092(\3\2\2\2\u0093\u0095\t\2\2\2\u0094\u0093\3") - buf.write(u"\2\2\2\u0095\u0096\3\2\2\2\u0096\u0094\3\2\2\2\u0096") - buf.write(u"\u0097\3\2\2\2\u0097\u0098\3\2\2\2\u0098\u0099\b\25\2") - buf.write(u"\2\u0099*\3\2\2\2\u009a\u009b\7\61\2\2\u009b\u009c\7") - buf.write(u"\61\2\2\u009c\u00a0\3\2\2\2\u009d\u009f\13\2\2\2\u009e") - buf.write(u"\u009d\3\2\2\2\u009f\u00a2\3\2\2\2\u00a0\u00a1\3\2\2") - buf.write(u"\2\u00a0\u009e\3\2\2\2\u00a1\u00a3\3\2\2\2\u00a2\u00a0") - buf.write(u"\3\2\2\2\u00a3\u00a4\7\f\2\2\u00a4\u00a5\3\2\2\2\u00a5") - buf.write(u"\u00a6\b\26\2\2\u00a6,\3\2\2\2\u00a7\u00a8\7\61\2\2\u00a8") - buf.write(u"\u00a9\7,\2\2\u00a9\u00ad\3\2\2\2\u00aa\u00ac\13\2\2") - buf.write(u"\2\u00ab\u00aa\3\2\2\2\u00ac\u00af\3\2\2\2\u00ad\u00ae") - buf.write(u"\3\2\2\2\u00ad\u00ab\3\2\2\2\u00ae\u00b0\3\2\2\2\u00af") - buf.write(u"\u00ad\3\2\2\2\u00b0\u00b1\7,\2\2\u00b1\u00b2\7\61\2") - buf.write(u"\2\u00b2\u00b3\3\2\2\2\u00b3\u00b4\b\27\2\2\u00b4.\3") - buf.write(u"\2\2\2\u00b5\u00b6\7,\2\2\u00b6\60\3\2\2\2\u00b7\u00b8") - buf.write(u"\7\61\2\2\u00b8\62\3\2\2\2\u00b9\u00ba\7-\2\2\u00ba\64") - buf.write(u"\3\2\2\2\u00bb\u00bc\7/\2\2\u00bc\66\3\2\2\2\u00bd\u00be") - buf.write(u"\7>\2\2\u00be8\3\2\2\2\u00bf\u00c0\7@\2\2\u00c0:\3\2") - buf.write(u"\2\2\u00c1\u00c2\7>\2\2\u00c2\u00c3\7?\2\2\u00c3<\3\2") - buf.write(u"\2\2\u00c4\u00c5\7@\2\2\u00c5\u00c6\7?\2\2\u00c6>\3\2") - buf.write(u"\2\2\u00c7\u00c8\7?\2\2\u00c8\u00c9\7?\2\2\u00c9@\3\2") - buf.write(u"\2\2\u00ca\u00cb\7#\2\2\u00cb\u00cc\7?\2\2\u00ccB\3\2") - buf.write(u"\2\2\u00cd\u00ce\7B\2\2\u00ce\u00cf\5S*\2\u00cfD\3\2") - buf.write(u"\2\2\u00d0\u00d1\7\'\2\2\u00d1\u00d2\5S*\2\u00d2F\3\2") - buf.write(u"\2\2\u00d3\u00d4\7\'\2\2\u00d4\u00d5\5O(\2\u00d5H\3\2") - buf.write(u"\2\2\u00d6\u00d7\7o\2\2\u00d7\u00d8\7w\2\2\u00d8\u00d9") - buf.write(u"\7v\2\2\u00d9J\3\2\2\2\u00da\u00db\7V\2\2\u00db\u00dc") - buf.write(u"\7t\2\2\u00dc\u00dd\7w\2\2\u00dd\u00e4\7g\2\2\u00de\u00df") - buf.write(u"\7H\2\2\u00df\u00e0\7c\2\2\u00e0\u00e1\7n\2\2\u00e1\u00e2") - buf.write(u"\7u\2\2\u00e2\u00e4\7g\2\2\u00e3\u00da\3\2\2\2\u00e3") - buf.write(u"\u00de\3\2\2\2\u00e4L\3\2\2\2\u00e5\u00e6\5O(\2\u00e6") - buf.write(u"\u00e7\7\60\2\2\u00e7\u00e9\5O(\2\u00e8\u00ea\5Q)\2\u00e9") - buf.write(u"\u00e8\3\2\2\2\u00e9\u00ea\3\2\2\2\u00ea\u00ef\3\2\2") - buf.write(u"\2\u00eb\u00ec\5O(\2\u00ec\u00ed\5Q)\2\u00ed\u00ef\3") - buf.write(u"\2\2\2\u00ee\u00e5\3\2\2\2\u00ee\u00eb\3\2\2\2\u00ef") - buf.write(u"N\3\2\2\2\u00f0\u00f2\5W,\2\u00f1\u00f0\3\2\2\2\u00f2") - buf.write(u"\u00f3\3\2\2\2\u00f3\u00f1\3\2\2\2\u00f3\u00f4\3\2\2") - buf.write(u"\2\u00f4P\3\2\2\2\u00f5\u00f7\t\3\2\2\u00f6\u00f8\t\4") - buf.write(u"\2\2\u00f7\u00f6\3\2\2\2\u00f7\u00f8\3\2\2\2\u00f8\u00f9") - buf.write(u"\3\2\2\2\u00f9\u00fa\5O(\2\u00faR\3\2\2\2\u00fb\u00fe") - buf.write(u"\7a\2\2\u00fc\u00fe\5U+\2\u00fd\u00fb\3\2\2\2\u00fd\u00fc") - buf.write(u"\3\2\2\2\u00fe\u0104\3\2\2\2\u00ff\u0103\7a\2\2\u0100") - buf.write(u"\u0103\5U+\2\u0101\u0103\5W,\2\u0102\u00ff\3\2\2\2\u0102") - buf.write(u"\u0100\3\2\2\2\u0102\u0101\3\2\2\2\u0103\u0106\3\2\2") - buf.write(u"\2\u0104\u0102\3\2\2\2\u0104\u0105\3\2\2\2\u0105T\3\2") - buf.write(u"\2\2\u0106\u0104\3\2\2\2\u0107\u0108\t\5\2\2\u0108V\3") - buf.write(u"\2\2\2\u0109\u010a\t\6\2\2\u010aX\3\2\2\2\16\2\u0096") - buf.write(u"\u00a0\u00ad\u00e3\u00e9\u00ee\u00f3\u00f7\u00fd\u0102") - buf.write(u"\u0104\3\b\2\2") - return buf.getvalue() - - -class RelayLexer(Lexer): - - atn = ATNDeserializer().deserialize(serializedATN()) - - decisionsToDFA = [ DFA(ds, i) for i, ds in enumerate(atn.decisionToState) ] - - T__0 = 1 - T__1 = 2 - T__2 = 3 - T__3 = 4 - T__4 = 5 - T__5 = 6 - T__6 = 7 - T__7 = 8 - T__8 = 9 - T__9 = 10 - T__10 = 11 - T__11 = 12 - T__12 = 13 - T__13 = 14 - T__14 = 15 - T__15 = 16 - T__16 = 17 - T__17 = 18 - SEMVER = 19 - WS = 20 - LINE_COMMENT = 21 - COMMENT = 22 - MUL = 23 - DIV = 24 - ADD = 25 - SUB = 26 - LT = 27 - GT = 28 - LE = 29 - GE = 30 - EQ = 31 - NE = 32 - GLOBAL_VAR = 33 - LOCAL_VAR = 34 - GRAPH_VAR = 35 - MUT = 36 - BOOL_LIT = 37 - FLOAT = 38 - NAT = 39 - CNAME = 40 - - channelNames = [ u"DEFAULT_TOKEN_CHANNEL", u"HIDDEN" ] - - modeNames = [ u"DEFAULT_MODE" ] - - literalNames = [ u"", - u"'('", u"')'", u"','", u"'['", u"']'", u"'if'", u"'else'", - u"'let'", u"'='", u"';'", u"'{'", u"'}'", u"'fn'", u"'->'", - u"'def'", u"':'", u"'Tensor'", u"'_'", u"'v0.0.2'", u"'*'", - u"'/'", u"'+'", u"'-'", u"'<'", u"'>'", u"'<='", u"'>='", u"'=='", - u"'!='", u"'mut'" ] - - symbolicNames = [ u"", - u"SEMVER", u"WS", u"LINE_COMMENT", u"COMMENT", u"MUL", u"DIV", - u"ADD", u"SUB", u"LT", u"GT", u"LE", u"GE", u"EQ", u"NE", u"GLOBAL_VAR", - u"LOCAL_VAR", u"GRAPH_VAR", u"MUT", u"BOOL_LIT", u"FLOAT", u"NAT", - u"CNAME" ] - - ruleNames = [ u"T__0", u"T__1", u"T__2", u"T__3", u"T__4", u"T__5", - u"T__6", u"T__7", u"T__8", u"T__9", u"T__10", u"T__11", - u"T__12", u"T__13", u"T__14", u"T__15", u"T__16", u"T__17", - u"SEMVER", u"WS", u"LINE_COMMENT", u"COMMENT", u"MUL", - u"DIV", u"ADD", u"SUB", u"LT", u"GT", u"LE", u"GE", u"EQ", - u"NE", u"GLOBAL_VAR", u"LOCAL_VAR", u"GRAPH_VAR", u"MUT", - u"BOOL_LIT", u"FLOAT", u"NAT", u"EXP", u"CNAME", u"LETTER", - u"DIGIT" ] - - grammarFileName = u"Relay.g4" - - def __init__(self, input=None, output=sys.stdout): - super(RelayLexer, self).__init__(input, output=output) - self.checkVersion("4.7.2") - self._interp = LexerATNSimulator(self, self.atn, self.decisionsToDFA, PredictionContextCache()) - self._actions = None - self._predicates = None - - diff --git a/python/tvm/relay/grammar/py2/RelayLexer.tokens b/python/tvm/relay/grammar/py2/RelayLexer.tokens deleted file mode 100644 index 41f3ee6..0000000 --- a/python/tvm/relay/grammar/py2/RelayLexer.tokens +++ /dev/null @@ -1,70 +0,0 @@ -T__0=1 -T__1=2 -T__2=3 -T__3=4 -T__4=5 -T__5=6 -T__6=7 -T__7=8 -T__8=9 -T__9=10 -T__10=11 -T__11=12 -T__12=13 -T__13=14 -T__14=15 -T__15=16 -T__16=17 -T__17=18 -SEMVER=19 -WS=20 -LINE_COMMENT=21 -COMMENT=22 -MUL=23 -DIV=24 -ADD=25 -SUB=26 -LT=27 -GT=28 -LE=29 -GE=30 -EQ=31 -NE=32 -GLOBAL_VAR=33 -LOCAL_VAR=34 -GRAPH_VAR=35 -MUT=36 -BOOL_LIT=37 -FLOAT=38 -NAT=39 -CNAME=40 -'('=1 -')'=2 -','=3 -'['=4 -']'=5 -'if'=6 -'else'=7 -'let'=8 -'='=9 -';'=10 -'{'=11 -'}'=12 -'fn'=13 -'->'=14 -'def'=15 -':'=16 -'Tensor'=17 -'_'=18 -'v0.0.2'=19 -'*'=23 -'/'=24 -'+'=25 -'-'=26 -'<'=27 -'>'=28 -'<='=29 -'>='=30 -'=='=31 -'!='=32 -'mut'=36 diff --git a/python/tvm/relay/grammar/py2/RelayParser.py b/python/tvm/relay/grammar/py2/RelayParser.py deleted file mode 100644 index 77f56bf..0000000 --- a/python/tvm/relay/grammar/py2/RelayParser.py +++ /dev/null @@ -1,2311 +0,0 @@ -# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2 -# encoding: utf-8 -from __future__ import print_function -from antlr4 import * -from io import StringIO -import sys - - -def serializedATN(): - with StringIO() as buf: - buf.write(u"\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\3") - buf.write(u"*\u014c\4\2\t\2\4\3\t\3\4\4\t\4\4\5\t\5\4\6\t\6\4\7\t") - buf.write(u"\7\4\b\t\b\4\t\t\t\4\n\t\n\4\13\t\13\4\f\t\f\4\r\t\r") - buf.write(u"\4\16\t\16\4\17\t\17\4\20\t\20\4\21\t\21\4\22\t\22\4") - buf.write(u"\23\t\23\3\2\3\2\3\3\3\3\7\3+\n\3\f\3\16\3.\13\3\3\3") - buf.write(u"\5\3\61\n\3\3\3\3\3\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3") - buf.write(u"\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\6\4H\n\4\r") - buf.write(u"\4\16\4I\3\4\3\4\3\4\3\4\3\4\3\4\7\4R\n\4\f\4\16\4U\13") - buf.write(u"\4\5\4W\n\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3") - buf.write(u"\4\5\4d\n\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\5\4n\n\4") - buf.write(u"\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4") - buf.write(u"\3\4\3\4\3\4\5\4\u0080\n\4\3\4\3\4\3\4\3\4\3\4\3\4\3") - buf.write(u"\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3") - buf.write(u"\4\7\4\u0096\n\4\f\4\16\4\u0099\13\4\5\4\u009b\n\4\3") - buf.write(u"\4\7\4\u009e\n\4\f\4\16\4\u00a1\13\4\3\5\3\5\5\5\u00a5") - buf.write(u"\n\5\3\5\3\5\3\5\3\5\3\5\5\5\u00ac\n\5\3\5\3\5\3\6\3") - buf.write(u"\6\3\6\5\6\u00b3\n\6\3\6\3\6\3\6\3\6\3\6\5\6\u00ba\n") - buf.write(u"\6\3\6\3\6\3\7\3\7\3\7\3\7\3\7\3\7\5\7\u00c4\n\7\3\b") - buf.write(u"\3\b\3\b\7\b\u00c9\n\b\f\b\16\b\u00cc\13\b\5\b\u00ce") - buf.write(u"\n\b\3\t\3\t\3\t\5\t\u00d3\n\t\3\n\3\n\3\n\7\n\u00d8") - buf.write(u"\n\n\f\n\16\n\u00db\13\n\5\n\u00dd\n\n\3\13\3\13\3\13") - buf.write(u"\3\13\3\f\3\f\3\f\3\f\3\f\3\f\7\f\u00e9\n\f\f\f\16\f") - buf.write(u"\u00ec\13\f\3\f\3\f\5\f\u00f0\n\f\3\r\3\r\3\r\3\r\3\r") - buf.write(u"\3\r\3\r\3\r\3\r\3\r\3\r\6\r\u00fd\n\r\r\r\16\r\u00fe") - buf.write(u"\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\5\r") - buf.write(u"\u010d\n\r\3\r\3\r\3\r\3\r\7\r\u0113\n\r\f\r\16\r\u0116") - buf.write(u"\13\r\5\r\u0118\n\r\3\r\3\r\3\r\3\r\3\r\5\r\u011f\n\r") - buf.write(u"\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3") - buf.write(u"\16\6\16\u012c\n\16\r\16\16\16\u012d\3\16\3\16\5\16\u0132") - buf.write(u"\n\16\3\17\3\17\3\17\3\17\3\17\5\17\u0139\n\17\3\20\3") - buf.write(u"\20\3\21\3\21\3\21\3\21\3\22\3\22\3\22\5\22\u0144\n\22") - buf.write(u"\3\23\3\23\3\23\3\23\5\23\u014a\n\23\3\23\2\3\6\24\2") - buf.write(u"\4\6\b\n\f\16\20\22\24\26\30\32\34\36 \"$\2\6\3\2\31") - buf.write(u"\32\3\2\33\34\3\2\35 \3\2!\"\2\u0175\2&\3\2\2\2\4(\3") - buf.write(u"\2\2\2\6\177\3\2\2\2\b\u00a2\3\2\2\2\n\u00af\3\2\2\2") - buf.write(u"\f\u00c3\3\2\2\2\16\u00cd\3\2\2\2\20\u00cf\3\2\2\2\22") - buf.write(u"\u00dc\3\2\2\2\24\u00de\3\2\2\2\26\u00ef\3\2\2\2\30\u011e") - buf.write(u"\3\2\2\2\32\u0131\3\2\2\2\34\u0138\3\2\2\2\36\u013a\3") - buf.write(u"\2\2\2 \u013c\3\2\2\2\"\u0143\3\2\2\2$\u0149\3\2\2\2") - buf.write(u"&\'\7*\2\2\'\3\3\2\2\2(\60\7\25\2\2)+\5\n\6\2*)\3\2\2") - buf.write(u"\2+.\3\2\2\2,*\3\2\2\2,-\3\2\2\2-\61\3\2\2\2.,\3\2\2") - buf.write(u"\2/\61\5\6\4\2\60,\3\2\2\2\60/\3\2\2\2\61\62\3\2\2\2") - buf.write(u"\62\63\7\2\2\3\63\5\3\2\2\2\64\65\b\4\1\2\65\66\7\3\2") - buf.write(u"\2\66\67\5\6\4\2\678\7\4\2\28\u0080\3\2\2\29:\7\34\2") - buf.write(u"\2:\u0080\5\6\4\23;\u0080\5\b\5\2<=\7\3\2\2=\u0080\7") - buf.write(u"\4\2\2>?\7\3\2\2?@\5\6\4\2@A\7\5\2\2AB\7\4\2\2B\u0080") - buf.write(u"\3\2\2\2CD\7\3\2\2DG\5\6\4\2EF\7\5\2\2FH\5\6\4\2GE\3") - buf.write(u"\2\2\2HI\3\2\2\2IG\3\2\2\2IJ\3\2\2\2JK\3\2\2\2KL\7\4") - buf.write(u"\2\2L\u0080\3\2\2\2MV\7\6\2\2NS\5\6\4\2OP\7\5\2\2PR\5") - buf.write(u"\6\4\2QO\3\2\2\2RU\3\2\2\2SQ\3\2\2\2ST\3\2\2\2TW\3\2") - buf.write(u"\2\2US\3\2\2\2VN\3\2\2\2VW\3\2\2\2WX\3\2\2\2X\u0080\7") - buf.write(u"\7\2\2YZ\7\b\2\2Z[\7\3\2\2[\\\5\6\4\2\\]\7\4\2\2]^\5") - buf.write(u" \21\2^_\7\t\2\2_`\5 \21\2`\u0080\3\2\2\2ac\7\n\2\2b") - buf.write(u"d\7&\2\2cb\3\2\2\2cd\3\2\2\2de\3\2\2\2ef\5\20\t\2fg\7") - buf.write(u"\13\2\2gh\5\6\4\2hi\7\f\2\2ij\5\6\4\bj\u0080\3\2\2\2") - buf.write(u"km\7\n\2\2ln\7&\2\2ml\3\2\2\2mn\3\2\2\2no\3\2\2\2op\5") - buf.write(u"\20\t\2pq\7\13\2\2qr\7\r\2\2rs\5\6\4\2st\7\16\2\2tu\7") - buf.write(u"\f\2\2uv\5\6\4\7v\u0080\3\2\2\2wx\5$\23\2xy\7\13\2\2") - buf.write(u"yz\5\6\4\2z{\7\f\2\2{|\5\6\4\5|\u0080\3\2\2\2}\u0080") - buf.write(u"\5$\23\2~\u0080\5\"\22\2\177\64\3\2\2\2\1779\3\2\2\2") - buf.write(u"\177;\3\2\2\2\177<\3\2\2\2\177>\3\2\2\2\177C\3\2\2\2") - buf.write(u"\177M\3\2\2\2\177Y\3\2\2\2\177a\3\2\2\2\177k\3\2\2\2") - buf.write(u"\177w\3\2\2\2\177}\3\2\2\2\177~\3\2\2\2\u0080\u009f\3") - buf.write(u"\2\2\2\u0081\u0082\f\22\2\2\u0082\u0083\t\2\2\2\u0083") - buf.write(u"\u009e\5\6\4\23\u0084\u0085\f\21\2\2\u0085\u0086\t\3") - buf.write(u"\2\2\u0086\u009e\5\6\4\22\u0087\u0088\f\20\2\2\u0088") - buf.write(u"\u0089\t\4\2\2\u0089\u009e\5\6\4\21\u008a\u008b\f\17") - buf.write(u"\2\2\u008b\u008c\t\5\2\2\u008c\u009e\5\6\4\20\u008d\u008e") - buf.write(u"\f\6\2\2\u008e\u008f\7\f\2\2\u008f\u009e\5\6\4\7\u0090") - buf.write(u"\u0091\f\24\2\2\u0091\u009a\7\3\2\2\u0092\u0097\5\6\4") - buf.write(u"\2\u0093\u0094\7\5\2\2\u0094\u0096\5\6\4\2\u0095\u0093") - buf.write(u"\3\2\2\2\u0096\u0099\3\2\2\2\u0097\u0095\3\2\2\2\u0097") - buf.write(u"\u0098\3\2\2\2\u0098\u009b\3\2\2\2\u0099\u0097\3\2\2") - buf.write(u"\2\u009a\u0092\3\2\2\2\u009a\u009b\3\2\2\2\u009b\u009c") - buf.write(u"\3\2\2\2\u009c\u009e\7\4\2\2\u009d\u0081\3\2\2\2\u009d") - buf.write(u"\u0084\3\2\2\2\u009d\u0087\3\2\2\2\u009d\u008a\3\2\2") - buf.write(u"\2\u009d\u008d\3\2\2\2\u009d\u0090\3\2\2\2\u009e\u00a1") - buf.write(u"\3\2\2\2\u009f\u009d\3\2\2\2\u009f\u00a0\3\2\2\2\u00a0") - buf.write(u"\7\3\2\2\2\u00a1\u009f\3\2\2\2\u00a2\u00a4\7\17\2\2\u00a3") - buf.write(u"\u00a5\5\26\f\2\u00a4\u00a3\3\2\2\2\u00a4\u00a5\3\2\2") - buf.write(u"\2\u00a5\u00a6\3\2\2\2\u00a6\u00a7\7\3\2\2\u00a7\u00a8") - buf.write(u"\5\f\7\2\u00a8\u00ab\7\4\2\2\u00a9\u00aa\7\20\2\2\u00aa") - buf.write(u"\u00ac\5\30\r\2\u00ab\u00a9\3\2\2\2\u00ab\u00ac\3\2\2") - buf.write(u"\2\u00ac\u00ad\3\2\2\2\u00ad\u00ae\5 \21\2\u00ae\t\3") - buf.write(u"\2\2\2\u00af\u00b0\7\21\2\2\u00b0\u00b2\5$\23\2\u00b1") - buf.write(u"\u00b3\5\26\f\2\u00b2\u00b1\3\2\2\2\u00b2\u00b3\3\2\2") - buf.write(u"\2\u00b3\u00b4\3\2\2\2\u00b4\u00b5\7\3\2\2\u00b5\u00b6") - buf.write(u"\5\f\7\2\u00b6\u00b9\7\4\2\2\u00b7\u00b8\7\20\2\2\u00b8") - buf.write(u"\u00ba\5\30\r\2\u00b9\u00b7\3\2\2\2\u00b9\u00ba\3\2\2") - buf.write(u"\2\u00ba\u00bb\3\2\2\2\u00bb\u00bc\5 \21\2\u00bc\13\3") - buf.write(u"\2\2\2\u00bd\u00c4\5\16\b\2\u00be\u00c4\5\22\n\2\u00bf") - buf.write(u"\u00c0\5\16\b\2\u00c0\u00c1\7\5\2\2\u00c1\u00c2\5\22") - buf.write(u"\n\2\u00c2\u00c4\3\2\2\2\u00c3\u00bd\3\2\2\2\u00c3\u00be") - buf.write(u"\3\2\2\2\u00c3\u00bf\3\2\2\2\u00c4\r\3\2\2\2\u00c5\u00ca") - buf.write(u"\5\20\t\2\u00c6\u00c7\7\5\2\2\u00c7\u00c9\5\20\t\2\u00c8") - buf.write(u"\u00c6\3\2\2\2\u00c9\u00cc\3\2\2\2\u00ca\u00c8\3\2\2") - buf.write(u"\2\u00ca\u00cb\3\2\2\2\u00cb\u00ce\3\2\2\2\u00cc\u00ca") - buf.write(u"\3\2\2\2\u00cd\u00c5\3\2\2\2\u00cd\u00ce\3\2\2\2\u00ce") - buf.write(u"\17\3\2\2\2\u00cf\u00d2\5$\23\2\u00d0\u00d1\7\22\2\2") - buf.write(u"\u00d1\u00d3\5\30\r\2\u00d2\u00d0\3\2\2\2\u00d2\u00d3") - buf.write(u"\3\2\2\2\u00d3\21\3\2\2\2\u00d4\u00d9\5\24\13\2\u00d5") - buf.write(u"\u00d6\7\5\2\2\u00d6\u00d8\5\24\13\2\u00d7\u00d5\3\2") - buf.write(u"\2\2\u00d8\u00db\3\2\2\2\u00d9\u00d7\3\2\2\2\u00d9\u00da") - buf.write(u"\3\2\2\2\u00da\u00dd\3\2\2\2\u00db\u00d9\3\2\2\2\u00dc") - buf.write(u"\u00d4\3\2\2\2\u00dc\u00dd\3\2\2\2\u00dd\23\3\2\2\2\u00de") - buf.write(u"\u00df\7*\2\2\u00df\u00e0\7\13\2\2\u00e0\u00e1\5\6\4") - buf.write(u"\2\u00e1\25\3\2\2\2\u00e2\u00e3\7\6\2\2\u00e3\u00f0\7") - buf.write(u"\7\2\2\u00e4\u00e5\7\6\2\2\u00e5\u00ea\5$\23\2\u00e6") - buf.write(u"\u00e7\7\5\2\2\u00e7\u00e9\5$\23\2\u00e8\u00e6\3\2\2") - buf.write(u"\2\u00e9\u00ec\3\2\2\2\u00ea\u00e8\3\2\2\2\u00ea\u00eb") - buf.write(u"\3\2\2\2\u00eb\u00ed\3\2\2\2\u00ec\u00ea\3\2\2\2\u00ed") - buf.write(u"\u00ee\7\7\2\2\u00ee\u00f0\3\2\2\2\u00ef\u00e2\3\2\2") - buf.write(u"\2\u00ef\u00e4\3\2\2\2\u00f0\27\3\2\2\2\u00f1\u00f2\7") - buf.write(u"\3\2\2\u00f2\u011f\7\4\2\2\u00f3\u00f4\7\3\2\2\u00f4") - buf.write(u"\u00f5\5\30\r\2\u00f5\u00f6\7\5\2\2\u00f6\u00f7\7\4\2") - buf.write(u"\2\u00f7\u011f\3\2\2\2\u00f8\u00f9\7\3\2\2\u00f9\u00fc") - buf.write(u"\5\30\r\2\u00fa\u00fb\7\5\2\2\u00fb\u00fd\5\30\r\2\u00fc") - buf.write(u"\u00fa\3\2\2\2\u00fd\u00fe\3\2\2\2\u00fe\u00fc\3\2\2") - buf.write(u"\2\u00fe\u00ff\3\2\2\2\u00ff\u0100\3\2\2\2\u0100\u0101") - buf.write(u"\7\4\2\2\u0101\u011f\3\2\2\2\u0102\u011f\5\36\20\2\u0103") - buf.write(u"\u0104\7\23\2\2\u0104\u0105\7\6\2\2\u0105\u0106\5\32") - buf.write(u"\16\2\u0106\u0107\7\5\2\2\u0107\u0108\5\30\r\2\u0108") - buf.write(u"\u0109\7\7\2\2\u0109\u011f\3\2\2\2\u010a\u010c\7\17\2") - buf.write(u"\2\u010b\u010d\5\26\f\2\u010c\u010b\3\2\2\2\u010c\u010d") - buf.write(u"\3\2\2\2\u010d\u010e\3\2\2\2\u010e\u0117\7\3\2\2\u010f") - buf.write(u"\u0114\5\30\r\2\u0110\u0111\7\5\2\2\u0111\u0113\5\30") - buf.write(u"\r\2\u0112\u0110\3\2\2\2\u0113\u0116\3\2\2\2\u0114\u0112") - buf.write(u"\3\2\2\2\u0114\u0115\3\2\2\2\u0115\u0118\3\2\2\2\u0116") - buf.write(u"\u0114\3\2\2\2\u0117\u010f\3\2\2\2\u0117\u0118\3\2\2") - buf.write(u"\2\u0118\u0119\3\2\2\2\u0119\u011a\7\4\2\2\u011a\u011b") - buf.write(u"\7\20\2\2\u011b\u011f\5\30\r\2\u011c\u011f\7\24\2\2\u011d") - buf.write(u"\u011f\7)\2\2\u011e\u00f1\3\2\2\2\u011e\u00f3\3\2\2\2") - buf.write(u"\u011e\u00f8\3\2\2\2\u011e\u0102\3\2\2\2\u011e\u0103") - buf.write(u"\3\2\2\2\u011e\u010a\3\2\2\2\u011e\u011c\3\2\2\2\u011e") - buf.write(u"\u011d\3\2\2\2\u011f\31\3\2\2\2\u0120\u0121\7\3\2\2\u0121") - buf.write(u"\u0132\7\4\2\2\u0122\u0123\7\3\2\2\u0123\u0124\5\34\17") - buf.write(u"\2\u0124\u0125\7\5\2\2\u0125\u0126\7\4\2\2\u0126\u0132") - buf.write(u"\3\2\2\2\u0127\u0128\7\3\2\2\u0128\u012b\5\34\17\2\u0129") - buf.write(u"\u012a\7\5\2\2\u012a\u012c\5\34\17\2\u012b\u0129\3\2") - buf.write(u"\2\2\u012c\u012d\3\2\2\2\u012d\u012b\3\2\2\2\u012d\u012e") - buf.write(u"\3\2\2\2\u012e\u012f\3\2\2\2\u012f\u0130\7\4\2\2\u0130") - buf.write(u"\u0132\3\2\2\2\u0131\u0120\3\2\2\2\u0131\u0122\3\2\2") - buf.write(u"\2\u0131\u0127\3\2\2\2\u0132\33\3\2\2\2\u0133\u0134\7") - buf.write(u"\3\2\2\u0134\u0135\5\34\17\2\u0135\u0136\7\4\2\2\u0136") - buf.write(u"\u0139\3\2\2\2\u0137\u0139\7)\2\2\u0138\u0133\3\2\2\2") - buf.write(u"\u0138\u0137\3\2\2\2\u0139\35\3\2\2\2\u013a\u013b\7*") - buf.write(u"\2\2\u013b\37\3\2\2\2\u013c\u013d\7\r\2\2\u013d\u013e") - buf.write(u"\5\6\4\2\u013e\u013f\7\16\2\2\u013f!\3\2\2\2\u0140\u0144") - buf.write(u"\7(\2\2\u0141\u0144\7)\2\2\u0142\u0144\7\'\2\2\u0143") - buf.write(u"\u0140\3\2\2\2\u0143\u0141\3\2\2\2\u0143\u0142\3\2\2") - buf.write(u"\2\u0144#\3\2\2\2\u0145\u014a\5\2\2\2\u0146\u014a\7#") - buf.write(u"\2\2\u0147\u014a\7$\2\2\u0148\u014a\7%\2\2\u0149\u0145") - buf.write(u"\3\2\2\2\u0149\u0146\3\2\2\2\u0149\u0147\3\2\2\2\u0149") - buf.write(u"\u0148\3\2\2\2\u014a%\3\2\2\2$,\60ISVcm\177\u0097\u009a") - buf.write(u"\u009d\u009f\u00a4\u00ab\u00b2\u00b9\u00c3\u00ca\u00cd") - buf.write(u"\u00d2\u00d9\u00dc\u00ea\u00ef\u00fe\u010c\u0114\u0117") - buf.write(u"\u011e\u012d\u0131\u0138\u0143\u0149") - return buf.getvalue() - - -class RelayParser ( Parser ): - - grammarFileName = "Relay.g4" - - atn = ATNDeserializer().deserialize(serializedATN()) - - decisionsToDFA = [ DFA(ds, i) for i, ds in enumerate(atn.decisionToState) ] - - sharedContextCache = PredictionContextCache() - - literalNames = [ u"", u"'('", u"')'", u"','", u"'['", u"']'", - u"'if'", u"'else'", u"'let'", u"'='", u"';'", u"'{'", - u"'}'", u"'fn'", u"'->'", u"'def'", u"':'", u"'Tensor'", - u"'_'", u"'v0.0.2'", u"", u"", u"", - u"'*'", u"'/'", u"'+'", u"'-'", u"'<'", u"'>'", u"'<='", - u"'>='", u"'=='", u"'!='", u"", u"", - u"", u"'mut'" ] - - symbolicNames = [ u"", u"", u"", u"", - u"", u"", u"", u"", - u"", u"", u"", u"", - u"", u"", u"", u"", - u"", u"", u"", u"SEMVER", - u"WS", u"LINE_COMMENT", u"COMMENT", u"MUL", u"DIV", - u"ADD", u"SUB", u"LT", u"GT", u"LE", u"GE", u"EQ", - u"NE", u"GLOBAL_VAR", u"LOCAL_VAR", u"GRAPH_VAR", - u"MUT", u"BOOL_LIT", u"FLOAT", u"NAT", u"CNAME" ] - - RULE_opIdent = 0 - RULE_prog = 1 - RULE_expr = 2 - RULE_func = 3 - RULE_defn = 4 - RULE_argList = 5 - RULE_varList = 6 - RULE_var = 7 - RULE_attrList = 8 - RULE_attr = 9 - RULE_typeParamSeq = 10 - RULE_type_ = 11 - RULE_shapeSeq = 12 - RULE_shape = 13 - RULE_typeIdent = 14 - RULE_body = 15 - RULE_scalar = 16 - RULE_ident = 17 - - ruleNames = [ u"opIdent", u"prog", u"expr", u"func", u"defn", u"argList", - u"varList", u"var", u"attrList", u"attr", u"typeParamSeq", - u"type_", u"shapeSeq", u"shape", u"typeIdent", u"body", - u"scalar", u"ident" ] - - EOF = Token.EOF - T__0=1 - T__1=2 - T__2=3 - T__3=4 - T__4=5 - T__5=6 - T__6=7 - T__7=8 - T__8=9 - T__9=10 - T__10=11 - T__11=12 - T__12=13 - T__13=14 - T__14=15 - T__15=16 - T__16=17 - T__17=18 - SEMVER=19 - WS=20 - LINE_COMMENT=21 - COMMENT=22 - MUL=23 - DIV=24 - ADD=25 - SUB=26 - LT=27 - GT=28 - LE=29 - GE=30 - EQ=31 - NE=32 - GLOBAL_VAR=33 - LOCAL_VAR=34 - GRAPH_VAR=35 - MUT=36 - BOOL_LIT=37 - FLOAT=38 - NAT=39 - CNAME=40 - - def __init__(self, input, output=sys.stdout): - super(RelayParser, self).__init__(input, output=output) - self.checkVersion("4.7.2") - self._interp = ParserATNSimulator(self, self.atn, self.decisionsToDFA, self.sharedContextCache) - self._predicates = None - - - - - class OpIdentContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.OpIdentContext, self).__init__(parent, invokingState) - self.parser = parser - - def CNAME(self): - return self.getToken(RelayParser.CNAME, 0) - - def getRuleIndex(self): - return RelayParser.RULE_opIdent - - def accept(self, visitor): - if hasattr(visitor, "visitOpIdent"): - return visitor.visitOpIdent(self) - else: - return visitor.visitChildren(self) - - - - - def opIdent(self): - - localctx = RelayParser.OpIdentContext(self, self._ctx, self.state) - self.enterRule(localctx, 0, self.RULE_opIdent) - try: - self.enterOuterAlt(localctx, 1) - self.state = 36 - self.match(RelayParser.CNAME) - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class ProgContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.ProgContext, self).__init__(parent, invokingState) - self.parser = parser - - def SEMVER(self): - return self.getToken(RelayParser.SEMVER, 0) - - def EOF(self): - return self.getToken(RelayParser.EOF, 0) - - def expr(self): - return self.getTypedRuleContext(RelayParser.ExprContext,0) - - - def defn(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.DefnContext) - else: - return self.getTypedRuleContext(RelayParser.DefnContext,i) - - - def getRuleIndex(self): - return RelayParser.RULE_prog - - def accept(self, visitor): - if hasattr(visitor, "visitProg"): - return visitor.visitProg(self) - else: - return visitor.visitChildren(self) - - - - - def prog(self): - - localctx = RelayParser.ProgContext(self, self._ctx, self.state) - self.enterRule(localctx, 2, self.RULE_prog) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 38 - self.match(RelayParser.SEMVER) - self.state = 46 - self._errHandler.sync(self) - token = self._input.LA(1) - if token in [RelayParser.EOF, RelayParser.T__14]: - self.state = 42 - self._errHandler.sync(self) - _la = self._input.LA(1) - while _la==RelayParser.T__14: - self.state = 39 - self.defn() - self.state = 44 - self._errHandler.sync(self) - _la = self._input.LA(1) - - pass - elif token in [RelayParser.T__0, RelayParser.T__3, RelayParser.T__5, RelayParser.T__7, RelayParser.T__12, RelayParser.SUB, RelayParser.GLOBAL_VAR, RelayParser.LOCAL_VAR, RelayParser.GRAPH_VAR, RelayParser.BOOL_LIT, RelayParser.FLOAT, RelayParser.NAT, RelayParser.CNAME]: - self.state = 45 - self.expr(0) - pass - else: - raise NoViableAltException(self) - - self.state = 48 - self.match(RelayParser.EOF) - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class ExprContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.ExprContext, self).__init__(parent, invokingState) - self.parser = parser - - - def getRuleIndex(self): - return RelayParser.RULE_expr - - - def copyFrom(self, ctx): - super(RelayParser.ExprContext, self).copyFrom(ctx) - - - class IdentExprContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.IdentExprContext, self).__init__(parser) - self.copyFrom(ctx) - - def ident(self): - return self.getTypedRuleContext(RelayParser.IdentContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitIdentExpr"): - return visitor.visitIdentExpr(self) - else: - return visitor.visitChildren(self) - - - class CallContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.CallContext, self).__init__(parser) - self.copyFrom(ctx) - - def expr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ExprContext) - else: - return self.getTypedRuleContext(RelayParser.ExprContext,i) - - - def accept(self, visitor): - if hasattr(visitor, "visitCall"): - return visitor.visitCall(self) - else: - return visitor.visitChildren(self) - - - class NegContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.NegContext, self).__init__(parser) - self.copyFrom(ctx) - - def SUB(self): - return self.getToken(RelayParser.SUB, 0) - def expr(self): - return self.getTypedRuleContext(RelayParser.ExprContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitNeg"): - return visitor.visitNeg(self) - else: - return visitor.visitChildren(self) - - - class TupleContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.TupleContext, self).__init__(parser) - self.copyFrom(ctx) - - def expr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ExprContext) - else: - return self.getTypedRuleContext(RelayParser.ExprContext,i) - - - def accept(self, visitor): - if hasattr(visitor, "visitTuple"): - return visitor.visitTuple(self) - else: - return visitor.visitChildren(self) - - - class ParensContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.ParensContext, self).__init__(parser) - self.copyFrom(ctx) - - def expr(self): - return self.getTypedRuleContext(RelayParser.ExprContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitParens"): - return visitor.visitParens(self) - else: - return visitor.visitChildren(self) - - - class FuncExprContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.FuncExprContext, self).__init__(parser) - self.copyFrom(ctx) - - def func(self): - return self.getTypedRuleContext(RelayParser.FuncContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitFuncExpr"): - return visitor.visitFuncExpr(self) - else: - return visitor.visitChildren(self) - - - class ScalarExprContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.ScalarExprContext, self).__init__(parser) - self.copyFrom(ctx) - - def scalar(self): - return self.getTypedRuleContext(RelayParser.ScalarContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitScalarExpr"): - return visitor.visitScalarExpr(self) - else: - return visitor.visitChildren(self) - - - class LetContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.LetContext, self).__init__(parser) - self.copyFrom(ctx) - - def var(self): - return self.getTypedRuleContext(RelayParser.VarContext,0) - - def expr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ExprContext) - else: - return self.getTypedRuleContext(RelayParser.ExprContext,i) - - def MUT(self): - return self.getToken(RelayParser.MUT, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitLet"): - return visitor.visitLet(self) - else: - return visitor.visitChildren(self) - - - class TensorContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.TensorContext, self).__init__(parser) - self.copyFrom(ctx) - - def expr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ExprContext) - else: - return self.getTypedRuleContext(RelayParser.ExprContext,i) - - - def accept(self, visitor): - if hasattr(visitor, "visitTensor"): - return visitor.visitTensor(self) - else: - return visitor.visitChildren(self) - - - class IfElseContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.IfElseContext, self).__init__(parser) - self.copyFrom(ctx) - - def expr(self): - return self.getTypedRuleContext(RelayParser.ExprContext,0) - - def body(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.BodyContext) - else: - return self.getTypedRuleContext(RelayParser.BodyContext,i) - - - def accept(self, visitor): - if hasattr(visitor, "visitIfElse"): - return visitor.visitIfElse(self) - else: - return visitor.visitChildren(self) - - - class GraphContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.GraphContext, self).__init__(parser) - self.copyFrom(ctx) - - def ident(self): - return self.getTypedRuleContext(RelayParser.IdentContext,0) - - def expr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ExprContext) - else: - return self.getTypedRuleContext(RelayParser.ExprContext,i) - - - def accept(self, visitor): - if hasattr(visitor, "visitGraph"): - return visitor.visitGraph(self) - else: - return visitor.visitChildren(self) - - - class BinOpContext(ExprContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ExprContext) - super(RelayParser.BinOpContext, self).__init__(parser) - self.op = None # Token - self.copyFrom(ctx) - - def expr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ExprContext) - else: - return self.getTypedRuleContext(RelayParser.ExprContext,i) - - def MUL(self): - return self.getToken(RelayParser.MUL, 0) - def DIV(self): - return self.getToken(RelayParser.DIV, 0) - def ADD(self): - return self.getToken(RelayParser.ADD, 0) - def SUB(self): - return self.getToken(RelayParser.SUB, 0) - def LT(self): - return self.getToken(RelayParser.LT, 0) - def GT(self): - return self.getToken(RelayParser.GT, 0) - def LE(self): - return self.getToken(RelayParser.LE, 0) - def GE(self): - return self.getToken(RelayParser.GE, 0) - def EQ(self): - return self.getToken(RelayParser.EQ, 0) - def NE(self): - return self.getToken(RelayParser.NE, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitBinOp"): - return visitor.visitBinOp(self) - else: - return visitor.visitChildren(self) - - - - def expr(self, _p=0): - _parentctx = self._ctx - _parentState = self.state - localctx = RelayParser.ExprContext(self, self._ctx, _parentState) - _prevctx = localctx - _startState = 4 - self.enterRecursionRule(localctx, 4, self.RULE_expr, _p) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 125 - self._errHandler.sync(self) - la_ = self._interp.adaptivePredict(self._input,7,self._ctx) - if la_ == 1: - localctx = RelayParser.ParensContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - - self.state = 51 - self.match(RelayParser.T__0) - self.state = 52 - self.expr(0) - self.state = 53 - self.match(RelayParser.T__1) - pass - - elif la_ == 2: - localctx = RelayParser.NegContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 55 - self.match(RelayParser.SUB) - self.state = 56 - self.expr(17) - pass - - elif la_ == 3: - localctx = RelayParser.FuncExprContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 57 - self.func() - pass - - elif la_ == 4: - localctx = RelayParser.TupleContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 58 - self.match(RelayParser.T__0) - self.state = 59 - self.match(RelayParser.T__1) - pass - - elif la_ == 5: - localctx = RelayParser.TupleContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 60 - self.match(RelayParser.T__0) - self.state = 61 - self.expr(0) - self.state = 62 - self.match(RelayParser.T__2) - self.state = 63 - self.match(RelayParser.T__1) - pass - - elif la_ == 6: - localctx = RelayParser.TupleContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 65 - self.match(RelayParser.T__0) - self.state = 66 - self.expr(0) - self.state = 69 - self._errHandler.sync(self) - _la = self._input.LA(1) - while True: - self.state = 67 - self.match(RelayParser.T__2) - self.state = 68 - self.expr(0) - self.state = 71 - self._errHandler.sync(self) - _la = self._input.LA(1) - if not (_la==RelayParser.T__2): - break - - self.state = 73 - self.match(RelayParser.T__1) - pass - - elif la_ == 7: - localctx = RelayParser.TensorContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 75 - self.match(RelayParser.T__3) - self.state = 84 - self._errHandler.sync(self) - _la = self._input.LA(1) - if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.T__0) | (1 << RelayParser.T__3) | (1 << RelayParser.T__5) | (1 << RelayParser.T__7) | (1 << RelayParser.T__12) | (1 << RelayParser.SUB) | (1 << RelayParser.GLOBAL_VAR) | (1 << RelayParser.LOCAL_VAR) | (1 << RelayParser.GRAPH_VAR) | (1 << RelayParser.BOOL_LIT) | (1 << RelayParser.FLOAT) | (1 << RelayParser.NAT) | (1 << RelayParser.CNAME))) != 0): - self.state = 76 - self.expr(0) - self.state = 81 - self._errHandler.sync(self) - _la = self._input.LA(1) - while _la==RelayParser.T__2: - self.state = 77 - self.match(RelayParser.T__2) - self.state = 78 - self.expr(0) - self.state = 83 - self._errHandler.sync(self) - _la = self._input.LA(1) - - - - self.state = 86 - self.match(RelayParser.T__4) - pass - - elif la_ == 8: - localctx = RelayParser.IfElseContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 87 - self.match(RelayParser.T__5) - self.state = 88 - self.match(RelayParser.T__0) - self.state = 89 - self.expr(0) - self.state = 90 - self.match(RelayParser.T__1) - self.state = 91 - self.body() - self.state = 92 - self.match(RelayParser.T__6) - self.state = 93 - self.body() - pass - - elif la_ == 9: - localctx = RelayParser.LetContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 95 - self.match(RelayParser.T__7) - self.state = 97 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.MUT: - self.state = 96 - self.match(RelayParser.MUT) - - - self.state = 99 - self.var() - self.state = 100 - self.match(RelayParser.T__8) - self.state = 101 - self.expr(0) - self.state = 102 - self.match(RelayParser.T__9) - self.state = 103 - self.expr(6) - pass - - elif la_ == 10: - localctx = RelayParser.LetContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 105 - self.match(RelayParser.T__7) - self.state = 107 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.MUT: - self.state = 106 - self.match(RelayParser.MUT) - - - self.state = 109 - self.var() - self.state = 110 - self.match(RelayParser.T__8) - self.state = 111 - self.match(RelayParser.T__10) - self.state = 112 - self.expr(0) - self.state = 113 - self.match(RelayParser.T__11) - self.state = 114 - self.match(RelayParser.T__9) - self.state = 115 - self.expr(5) - pass - - elif la_ == 11: - localctx = RelayParser.GraphContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 117 - self.ident() - self.state = 118 - self.match(RelayParser.T__8) - self.state = 119 - self.expr(0) - self.state = 120 - self.match(RelayParser.T__9) - self.state = 121 - self.expr(3) - pass - - elif la_ == 12: - localctx = RelayParser.IdentExprContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 123 - self.ident() - pass - - elif la_ == 13: - localctx = RelayParser.ScalarExprContext(self, localctx) - self._ctx = localctx - _prevctx = localctx - self.state = 124 - self.scalar() - pass - - - self._ctx.stop = self._input.LT(-1) - self.state = 157 - self._errHandler.sync(self) - _alt = self._interp.adaptivePredict(self._input,11,self._ctx) - while _alt!=2 and _alt!=ATN.INVALID_ALT_NUMBER: - if _alt==1: - if self._parseListeners is not None: - self.triggerExitRuleEvent() - _prevctx = localctx - self.state = 155 - self._errHandler.sync(self) - la_ = self._interp.adaptivePredict(self._input,10,self._ctx) - if la_ == 1: - localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState)) - self.pushNewRecursionContext(localctx, _startState, self.RULE_expr) - self.state = 127 - if not self.precpred(self._ctx, 16): - from antlr4.error.Errors import FailedPredicateException - raise FailedPredicateException(self, "self.precpred(self._ctx, 16)") - self.state = 128 - localctx.op = self._input.LT(1) - _la = self._input.LA(1) - if not(_la==RelayParser.MUL or _la==RelayParser.DIV): - localctx.op = self._errHandler.recoverInline(self) - else: - self._errHandler.reportMatch(self) - self.consume() - self.state = 129 - self.expr(17) - pass - - elif la_ == 2: - localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState)) - self.pushNewRecursionContext(localctx, _startState, self.RULE_expr) - self.state = 130 - if not self.precpred(self._ctx, 15): - from antlr4.error.Errors import FailedPredicateException - raise FailedPredicateException(self, "self.precpred(self._ctx, 15)") - self.state = 131 - localctx.op = self._input.LT(1) - _la = self._input.LA(1) - if not(_la==RelayParser.ADD or _la==RelayParser.SUB): - localctx.op = self._errHandler.recoverInline(self) - else: - self._errHandler.reportMatch(self) - self.consume() - self.state = 132 - self.expr(16) - pass - - elif la_ == 3: - localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState)) - self.pushNewRecursionContext(localctx, _startState, self.RULE_expr) - self.state = 133 - if not self.precpred(self._ctx, 14): - from antlr4.error.Errors import FailedPredicateException - raise FailedPredicateException(self, "self.precpred(self._ctx, 14)") - self.state = 134 - localctx.op = self._input.LT(1) - _la = self._input.LA(1) - if not((((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.LT) | (1 << RelayParser.GT) | (1 << RelayParser.LE) | (1 << RelayParser.GE))) != 0)): - localctx.op = self._errHandler.recoverInline(self) - else: - self._errHandler.reportMatch(self) - self.consume() - self.state = 135 - self.expr(15) - pass - - elif la_ == 4: - localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState)) - self.pushNewRecursionContext(localctx, _startState, self.RULE_expr) - self.state = 136 - if not self.precpred(self._ctx, 13): - from antlr4.error.Errors import FailedPredicateException - raise FailedPredicateException(self, "self.precpred(self._ctx, 13)") - self.state = 137 - localctx.op = self._input.LT(1) - _la = self._input.LA(1) - if not(_la==RelayParser.EQ or _la==RelayParser.NE): - localctx.op = self._errHandler.recoverInline(self) - else: - self._errHandler.reportMatch(self) - self.consume() - self.state = 138 - self.expr(14) - pass - - elif la_ == 5: - localctx = RelayParser.LetContext(self, RelayParser.ExprContext(self, _parentctx, _parentState)) - self.pushNewRecursionContext(localctx, _startState, self.RULE_expr) - self.state = 139 - if not self.precpred(self._ctx, 4): - from antlr4.error.Errors import FailedPredicateException - raise FailedPredicateException(self, "self.precpred(self._ctx, 4)") - self.state = 140 - self.match(RelayParser.T__9) - self.state = 141 - self.expr(5) - pass - - elif la_ == 6: - localctx = RelayParser.CallContext(self, RelayParser.ExprContext(self, _parentctx, _parentState)) - self.pushNewRecursionContext(localctx, _startState, self.RULE_expr) - self.state = 142 - if not self.precpred(self._ctx, 18): - from antlr4.error.Errors import FailedPredicateException - raise FailedPredicateException(self, "self.precpred(self._ctx, 18)") - self.state = 143 - self.match(RelayParser.T__0) - self.state = 152 - self._errHandler.sync(self) - _la = self._input.LA(1) - if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.T__0) | (1 << RelayParser.T__3) | (1 << RelayParser.T__5) | (1 << RelayParser.T__7) | (1 << RelayParser.T__12) | (1 << RelayParser.SUB) | (1 << RelayParser.GLOBAL_VAR) | (1 << RelayParser.LOCAL_VAR) | (1 << RelayParser.GRAPH_VAR) | (1 << RelayParser.BOOL_LIT) | (1 << RelayParser.FLOAT) | (1 << RelayParser.NAT) | (1 << RelayParser.CNAME))) != 0): - self.state = 144 - self.expr(0) - self.state = 149 - self._errHandler.sync(self) - _la = self._input.LA(1) - while _la==RelayParser.T__2: - self.state = 145 - self.match(RelayParser.T__2) - self.state = 146 - self.expr(0) - self.state = 151 - self._errHandler.sync(self) - _la = self._input.LA(1) - - - - self.state = 154 - self.match(RelayParser.T__1) - pass - - - self.state = 159 - self._errHandler.sync(self) - _alt = self._interp.adaptivePredict(self._input,11,self._ctx) - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.unrollRecursionContexts(_parentctx) - return localctx - - - class FuncContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.FuncContext, self).__init__(parent, invokingState) - self.parser = parser - - def argList(self): - return self.getTypedRuleContext(RelayParser.ArgListContext,0) - - - def body(self): - return self.getTypedRuleContext(RelayParser.BodyContext,0) - - - def typeParamSeq(self): - return self.getTypedRuleContext(RelayParser.TypeParamSeqContext,0) - - - def type_(self): - return self.getTypedRuleContext(RelayParser.Type_Context,0) - - - def getRuleIndex(self): - return RelayParser.RULE_func - - def accept(self, visitor): - if hasattr(visitor, "visitFunc"): - return visitor.visitFunc(self) - else: - return visitor.visitChildren(self) - - - - - def func(self): - - localctx = RelayParser.FuncContext(self, self._ctx, self.state) - self.enterRule(localctx, 6, self.RULE_func) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 160 - self.match(RelayParser.T__12) - self.state = 162 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.T__3: - self.state = 161 - self.typeParamSeq() - - - self.state = 164 - self.match(RelayParser.T__0) - self.state = 165 - self.argList() - self.state = 166 - self.match(RelayParser.T__1) - self.state = 169 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.T__13: - self.state = 167 - self.match(RelayParser.T__13) - self.state = 168 - self.type_() - - - self.state = 171 - self.body() - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class DefnContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.DefnContext, self).__init__(parent, invokingState) - self.parser = parser - - def ident(self): - return self.getTypedRuleContext(RelayParser.IdentContext,0) - - - def argList(self): - return self.getTypedRuleContext(RelayParser.ArgListContext,0) - - - def body(self): - return self.getTypedRuleContext(RelayParser.BodyContext,0) - - - def typeParamSeq(self): - return self.getTypedRuleContext(RelayParser.TypeParamSeqContext,0) - - - def type_(self): - return self.getTypedRuleContext(RelayParser.Type_Context,0) - - - def getRuleIndex(self): - return RelayParser.RULE_defn - - def accept(self, visitor): - if hasattr(visitor, "visitDefn"): - return visitor.visitDefn(self) - else: - return visitor.visitChildren(self) - - - - - def defn(self): - - localctx = RelayParser.DefnContext(self, self._ctx, self.state) - self.enterRule(localctx, 8, self.RULE_defn) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 173 - self.match(RelayParser.T__14) - self.state = 174 - self.ident() - self.state = 176 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.T__3: - self.state = 175 - self.typeParamSeq() - - - self.state = 178 - self.match(RelayParser.T__0) - self.state = 179 - self.argList() - self.state = 180 - self.match(RelayParser.T__1) - self.state = 183 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.T__13: - self.state = 181 - self.match(RelayParser.T__13) - self.state = 182 - self.type_() - - - self.state = 185 - self.body() - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class ArgListContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.ArgListContext, self).__init__(parent, invokingState) - self.parser = parser - - def varList(self): - return self.getTypedRuleContext(RelayParser.VarListContext,0) - - - def attrList(self): - return self.getTypedRuleContext(RelayParser.AttrListContext,0) - - - def getRuleIndex(self): - return RelayParser.RULE_argList - - def accept(self, visitor): - if hasattr(visitor, "visitArgList"): - return visitor.visitArgList(self) - else: - return visitor.visitChildren(self) - - - - - def argList(self): - - localctx = RelayParser.ArgListContext(self, self._ctx, self.state) - self.enterRule(localctx, 10, self.RULE_argList) - try: - self.state = 193 - self._errHandler.sync(self) - la_ = self._interp.adaptivePredict(self._input,16,self._ctx) - if la_ == 1: - self.enterOuterAlt(localctx, 1) - self.state = 187 - self.varList() - pass - - elif la_ == 2: - self.enterOuterAlt(localctx, 2) - self.state = 188 - self.attrList() - pass - - elif la_ == 3: - self.enterOuterAlt(localctx, 3) - self.state = 189 - self.varList() - self.state = 190 - self.match(RelayParser.T__2) - self.state = 191 - self.attrList() - pass - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class VarListContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.VarListContext, self).__init__(parent, invokingState) - self.parser = parser - - def var(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.VarContext) - else: - return self.getTypedRuleContext(RelayParser.VarContext,i) - - - def getRuleIndex(self): - return RelayParser.RULE_varList - - def accept(self, visitor): - if hasattr(visitor, "visitVarList"): - return visitor.visitVarList(self) - else: - return visitor.visitChildren(self) - - - - - def varList(self): - - localctx = RelayParser.VarListContext(self, self._ctx, self.state) - self.enterRule(localctx, 12, self.RULE_varList) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 203 - self._errHandler.sync(self) - _la = self._input.LA(1) - if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.GLOBAL_VAR) | (1 << RelayParser.LOCAL_VAR) | (1 << RelayParser.GRAPH_VAR) | (1 << RelayParser.CNAME))) != 0): - self.state = 195 - self.var() - self.state = 200 - self._errHandler.sync(self) - _alt = self._interp.adaptivePredict(self._input,17,self._ctx) - while _alt!=2 and _alt!=ATN.INVALID_ALT_NUMBER: - if _alt==1: - self.state = 196 - self.match(RelayParser.T__2) - self.state = 197 - self.var() - self.state = 202 - self._errHandler.sync(self) - _alt = self._interp.adaptivePredict(self._input,17,self._ctx) - - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class VarContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.VarContext, self).__init__(parent, invokingState) - self.parser = parser - - def ident(self): - return self.getTypedRuleContext(RelayParser.IdentContext,0) - - - def type_(self): - return self.getTypedRuleContext(RelayParser.Type_Context,0) - - - def getRuleIndex(self): - return RelayParser.RULE_var - - def accept(self, visitor): - if hasattr(visitor, "visitVar"): - return visitor.visitVar(self) - else: - return visitor.visitChildren(self) - - - - - def var(self): - - localctx = RelayParser.VarContext(self, self._ctx, self.state) - self.enterRule(localctx, 14, self.RULE_var) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 205 - self.ident() - self.state = 208 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.T__15: - self.state = 206 - self.match(RelayParser.T__15) - self.state = 207 - self.type_() - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class AttrListContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.AttrListContext, self).__init__(parent, invokingState) - self.parser = parser - - def attr(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.AttrContext) - else: - return self.getTypedRuleContext(RelayParser.AttrContext,i) - - - def getRuleIndex(self): - return RelayParser.RULE_attrList - - def accept(self, visitor): - if hasattr(visitor, "visitAttrList"): - return visitor.visitAttrList(self) - else: - return visitor.visitChildren(self) - - - - - def attrList(self): - - localctx = RelayParser.AttrListContext(self, self._ctx, self.state) - self.enterRule(localctx, 16, self.RULE_attrList) - self._la = 0 # Token type - try: - self.enterOuterAlt(localctx, 1) - self.state = 218 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.CNAME: - self.state = 210 - self.attr() - self.state = 215 - self._errHandler.sync(self) - _la = self._input.LA(1) - while _la==RelayParser.T__2: - self.state = 211 - self.match(RelayParser.T__2) - self.state = 212 - self.attr() - self.state = 217 - self._errHandler.sync(self) - _la = self._input.LA(1) - - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class AttrContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.AttrContext, self).__init__(parent, invokingState) - self.parser = parser - - def CNAME(self): - return self.getToken(RelayParser.CNAME, 0) - - def expr(self): - return self.getTypedRuleContext(RelayParser.ExprContext,0) - - - def getRuleIndex(self): - return RelayParser.RULE_attr - - def accept(self, visitor): - if hasattr(visitor, "visitAttr"): - return visitor.visitAttr(self) - else: - return visitor.visitChildren(self) - - - - - def attr(self): - - localctx = RelayParser.AttrContext(self, self._ctx, self.state) - self.enterRule(localctx, 18, self.RULE_attr) - try: - self.enterOuterAlt(localctx, 1) - self.state = 220 - self.match(RelayParser.CNAME) - self.state = 221 - self.match(RelayParser.T__8) - self.state = 222 - self.expr(0) - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class TypeParamSeqContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.TypeParamSeqContext, self).__init__(parent, invokingState) - self.parser = parser - - def ident(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.IdentContext) - else: - return self.getTypedRuleContext(RelayParser.IdentContext,i) - - - def getRuleIndex(self): - return RelayParser.RULE_typeParamSeq - - def accept(self, visitor): - if hasattr(visitor, "visitTypeParamSeq"): - return visitor.visitTypeParamSeq(self) - else: - return visitor.visitChildren(self) - - - - - def typeParamSeq(self): - - localctx = RelayParser.TypeParamSeqContext(self, self._ctx, self.state) - self.enterRule(localctx, 20, self.RULE_typeParamSeq) - self._la = 0 # Token type - try: - self.state = 237 - self._errHandler.sync(self) - la_ = self._interp.adaptivePredict(self._input,23,self._ctx) - if la_ == 1: - self.enterOuterAlt(localctx, 1) - self.state = 224 - self.match(RelayParser.T__3) - self.state = 225 - self.match(RelayParser.T__4) - pass - - elif la_ == 2: - self.enterOuterAlt(localctx, 2) - self.state = 226 - self.match(RelayParser.T__3) - self.state = 227 - self.ident() - self.state = 232 - self._errHandler.sync(self) - _la = self._input.LA(1) - while _la==RelayParser.T__2: - self.state = 228 - self.match(RelayParser.T__2) - self.state = 229 - self.ident() - self.state = 234 - self._errHandler.sync(self) - _la = self._input.LA(1) - - self.state = 235 - self.match(RelayParser.T__4) - pass - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class Type_Context(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.Type_Context, self).__init__(parent, invokingState) - self.parser = parser - - - def getRuleIndex(self): - return RelayParser.RULE_type_ - - - def copyFrom(self, ctx): - super(RelayParser.Type_Context, self).copyFrom(ctx) - - - - class IntTypeContext(Type_Context): - - def __init__(self, parser, ctx): # actually a RelayParser.Type_Context) - super(RelayParser.IntTypeContext, self).__init__(parser) - self.copyFrom(ctx) - - def NAT(self): - return self.getToken(RelayParser.NAT, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitIntType"): - return visitor.visitIntType(self) - else: - return visitor.visitChildren(self) - - - class TupleTypeContext(Type_Context): - - def __init__(self, parser, ctx): # actually a RelayParser.Type_Context) - super(RelayParser.TupleTypeContext, self).__init__(parser) - self.copyFrom(ctx) - - def type_(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.Type_Context) - else: - return self.getTypedRuleContext(RelayParser.Type_Context,i) - - - def accept(self, visitor): - if hasattr(visitor, "visitTupleType"): - return visitor.visitTupleType(self) - else: - return visitor.visitChildren(self) - - - class TypeIdentTypeContext(Type_Context): - - def __init__(self, parser, ctx): # actually a RelayParser.Type_Context) - super(RelayParser.TypeIdentTypeContext, self).__init__(parser) - self.copyFrom(ctx) - - def typeIdent(self): - return self.getTypedRuleContext(RelayParser.TypeIdentContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitTypeIdentType"): - return visitor.visitTypeIdentType(self) - else: - return visitor.visitChildren(self) - - - class IncompleteTypeContext(Type_Context): - - def __init__(self, parser, ctx): # actually a RelayParser.Type_Context) - super(RelayParser.IncompleteTypeContext, self).__init__(parser) - self.copyFrom(ctx) - - - def accept(self, visitor): - if hasattr(visitor, "visitIncompleteType"): - return visitor.visitIncompleteType(self) - else: - return visitor.visitChildren(self) - - - class TensorTypeContext(Type_Context): - - def __init__(self, parser, ctx): # actually a RelayParser.Type_Context) - super(RelayParser.TensorTypeContext, self).__init__(parser) - self.copyFrom(ctx) - - def shapeSeq(self): - return self.getTypedRuleContext(RelayParser.ShapeSeqContext,0) - - def type_(self): - return self.getTypedRuleContext(RelayParser.Type_Context,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitTensorType"): - return visitor.visitTensorType(self) - else: - return visitor.visitChildren(self) - - - class FuncTypeContext(Type_Context): - - def __init__(self, parser, ctx): # actually a RelayParser.Type_Context) - super(RelayParser.FuncTypeContext, self).__init__(parser) - self.copyFrom(ctx) - - def type_(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.Type_Context) - else: - return self.getTypedRuleContext(RelayParser.Type_Context,i) - - def typeParamSeq(self): - return self.getTypedRuleContext(RelayParser.TypeParamSeqContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitFuncType"): - return visitor.visitFuncType(self) - else: - return visitor.visitChildren(self) - - - - def type_(self): - - localctx = RelayParser.Type_Context(self, self._ctx, self.state) - self.enterRule(localctx, 22, self.RULE_type_) - self._la = 0 # Token type - try: - self.state = 284 - self._errHandler.sync(self) - la_ = self._interp.adaptivePredict(self._input,28,self._ctx) - if la_ == 1: - localctx = RelayParser.TupleTypeContext(self, localctx) - self.enterOuterAlt(localctx, 1) - self.state = 239 - self.match(RelayParser.T__0) - self.state = 240 - self.match(RelayParser.T__1) - pass - - elif la_ == 2: - localctx = RelayParser.TupleTypeContext(self, localctx) - self.enterOuterAlt(localctx, 2) - self.state = 241 - self.match(RelayParser.T__0) - self.state = 242 - self.type_() - self.state = 243 - self.match(RelayParser.T__2) - self.state = 244 - self.match(RelayParser.T__1) - pass - - elif la_ == 3: - localctx = RelayParser.TupleTypeContext(self, localctx) - self.enterOuterAlt(localctx, 3) - self.state = 246 - self.match(RelayParser.T__0) - self.state = 247 - self.type_() - self.state = 250 - self._errHandler.sync(self) - _la = self._input.LA(1) - while True: - self.state = 248 - self.match(RelayParser.T__2) - self.state = 249 - self.type_() - self.state = 252 - self._errHandler.sync(self) - _la = self._input.LA(1) - if not (_la==RelayParser.T__2): - break - - self.state = 254 - self.match(RelayParser.T__1) - pass - - elif la_ == 4: - localctx = RelayParser.TypeIdentTypeContext(self, localctx) - self.enterOuterAlt(localctx, 4) - self.state = 256 - self.typeIdent() - pass - - elif la_ == 5: - localctx = RelayParser.TensorTypeContext(self, localctx) - self.enterOuterAlt(localctx, 5) - self.state = 257 - self.match(RelayParser.T__16) - self.state = 258 - self.match(RelayParser.T__3) - self.state = 259 - self.shapeSeq() - self.state = 260 - self.match(RelayParser.T__2) - self.state = 261 - self.type_() - self.state = 262 - self.match(RelayParser.T__4) - pass - - elif la_ == 6: - localctx = RelayParser.FuncTypeContext(self, localctx) - self.enterOuterAlt(localctx, 6) - self.state = 264 - self.match(RelayParser.T__12) - self.state = 266 - self._errHandler.sync(self) - _la = self._input.LA(1) - if _la==RelayParser.T__3: - self.state = 265 - self.typeParamSeq() - - - self.state = 268 - self.match(RelayParser.T__0) - self.state = 277 - self._errHandler.sync(self) - _la = self._input.LA(1) - if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.T__0) | (1 << RelayParser.T__12) | (1 << RelayParser.T__16) | (1 << RelayParser.T__17) | (1 << RelayParser.NAT) | (1 << RelayParser.CNAME))) != 0): - self.state = 269 - self.type_() - self.state = 274 - self._errHandler.sync(self) - _la = self._input.LA(1) - while _la==RelayParser.T__2: - self.state = 270 - self.match(RelayParser.T__2) - self.state = 271 - self.type_() - self.state = 276 - self._errHandler.sync(self) - _la = self._input.LA(1) - - - - self.state = 279 - self.match(RelayParser.T__1) - self.state = 280 - self.match(RelayParser.T__13) - self.state = 281 - self.type_() - pass - - elif la_ == 7: - localctx = RelayParser.IncompleteTypeContext(self, localctx) - self.enterOuterAlt(localctx, 7) - self.state = 282 - self.match(RelayParser.T__17) - pass - - elif la_ == 8: - localctx = RelayParser.IntTypeContext(self, localctx) - self.enterOuterAlt(localctx, 8) - self.state = 283 - self.match(RelayParser.NAT) - pass - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class ShapeSeqContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.ShapeSeqContext, self).__init__(parent, invokingState) - self.parser = parser - - def shape(self, i=None): - if i is None: - return self.getTypedRuleContexts(RelayParser.ShapeContext) - else: - return self.getTypedRuleContext(RelayParser.ShapeContext,i) - - - def getRuleIndex(self): - return RelayParser.RULE_shapeSeq - - def accept(self, visitor): - if hasattr(visitor, "visitShapeSeq"): - return visitor.visitShapeSeq(self) - else: - return visitor.visitChildren(self) - - - - - def shapeSeq(self): - - localctx = RelayParser.ShapeSeqContext(self, self._ctx, self.state) - self.enterRule(localctx, 24, self.RULE_shapeSeq) - self._la = 0 # Token type - try: - self.state = 303 - self._errHandler.sync(self) - la_ = self._interp.adaptivePredict(self._input,30,self._ctx) - if la_ == 1: - self.enterOuterAlt(localctx, 1) - self.state = 286 - self.match(RelayParser.T__0) - self.state = 287 - self.match(RelayParser.T__1) - pass - - elif la_ == 2: - self.enterOuterAlt(localctx, 2) - self.state = 288 - self.match(RelayParser.T__0) - self.state = 289 - self.shape() - self.state = 290 - self.match(RelayParser.T__2) - self.state = 291 - self.match(RelayParser.T__1) - pass - - elif la_ == 3: - self.enterOuterAlt(localctx, 3) - self.state = 293 - self.match(RelayParser.T__0) - self.state = 294 - self.shape() - self.state = 297 - self._errHandler.sync(self) - _la = self._input.LA(1) - while True: - self.state = 295 - self.match(RelayParser.T__2) - self.state = 296 - self.shape() - self.state = 299 - self._errHandler.sync(self) - _la = self._input.LA(1) - if not (_la==RelayParser.T__2): - break - - self.state = 301 - self.match(RelayParser.T__1) - pass - - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class ShapeContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.ShapeContext, self).__init__(parent, invokingState) - self.parser = parser - - - def getRuleIndex(self): - return RelayParser.RULE_shape - - - def copyFrom(self, ctx): - super(RelayParser.ShapeContext, self).copyFrom(ctx) - - - - class ParensShapeContext(ShapeContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ShapeContext) - super(RelayParser.ParensShapeContext, self).__init__(parser) - self.copyFrom(ctx) - - def shape(self): - return self.getTypedRuleContext(RelayParser.ShapeContext,0) - - - def accept(self, visitor): - if hasattr(visitor, "visitParensShape"): - return visitor.visitParensShape(self) - else: - return visitor.visitChildren(self) - - - class IntShapeContext(ShapeContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ShapeContext) - super(RelayParser.IntShapeContext, self).__init__(parser) - self.copyFrom(ctx) - - def NAT(self): - return self.getToken(RelayParser.NAT, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitIntShape"): - return visitor.visitIntShape(self) - else: - return visitor.visitChildren(self) - - - - def shape(self): - - localctx = RelayParser.ShapeContext(self, self._ctx, self.state) - self.enterRule(localctx, 26, self.RULE_shape) - try: - self.state = 310 - self._errHandler.sync(self) - token = self._input.LA(1) - if token in [RelayParser.T__0]: - localctx = RelayParser.ParensShapeContext(self, localctx) - self.enterOuterAlt(localctx, 1) - self.state = 305 - self.match(RelayParser.T__0) - self.state = 306 - self.shape() - self.state = 307 - self.match(RelayParser.T__1) - pass - elif token in [RelayParser.NAT]: - localctx = RelayParser.IntShapeContext(self, localctx) - self.enterOuterAlt(localctx, 2) - self.state = 309 - self.match(RelayParser.NAT) - pass - else: - raise NoViableAltException(self) - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class TypeIdentContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.TypeIdentContext, self).__init__(parent, invokingState) - self.parser = parser - - def CNAME(self): - return self.getToken(RelayParser.CNAME, 0) - - def getRuleIndex(self): - return RelayParser.RULE_typeIdent - - def accept(self, visitor): - if hasattr(visitor, "visitTypeIdent"): - return visitor.visitTypeIdent(self) - else: - return visitor.visitChildren(self) - - - - - def typeIdent(self): - - localctx = RelayParser.TypeIdentContext(self, self._ctx, self.state) - self.enterRule(localctx, 28, self.RULE_typeIdent) - try: - self.enterOuterAlt(localctx, 1) - self.state = 312 - self.match(RelayParser.CNAME) - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class BodyContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.BodyContext, self).__init__(parent, invokingState) - self.parser = parser - - def expr(self): - return self.getTypedRuleContext(RelayParser.ExprContext,0) - - - def getRuleIndex(self): - return RelayParser.RULE_body - - def accept(self, visitor): - if hasattr(visitor, "visitBody"): - return visitor.visitBody(self) - else: - return visitor.visitChildren(self) - - - - - def body(self): - - localctx = RelayParser.BodyContext(self, self._ctx, self.state) - self.enterRule(localctx, 30, self.RULE_body) - try: - self.enterOuterAlt(localctx, 1) - self.state = 314 - self.match(RelayParser.T__10) - self.state = 315 - self.expr(0) - self.state = 316 - self.match(RelayParser.T__11) - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class ScalarContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.ScalarContext, self).__init__(parent, invokingState) - self.parser = parser - - - def getRuleIndex(self): - return RelayParser.RULE_scalar - - - def copyFrom(self, ctx): - super(RelayParser.ScalarContext, self).copyFrom(ctx) - - - - class ScalarFloatContext(ScalarContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ScalarContext) - super(RelayParser.ScalarFloatContext, self).__init__(parser) - self.copyFrom(ctx) - - def FLOAT(self): - return self.getToken(RelayParser.FLOAT, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitScalarFloat"): - return visitor.visitScalarFloat(self) - else: - return visitor.visitChildren(self) - - - class ScalarBoolContext(ScalarContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ScalarContext) - super(RelayParser.ScalarBoolContext, self).__init__(parser) - self.copyFrom(ctx) - - def BOOL_LIT(self): - return self.getToken(RelayParser.BOOL_LIT, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitScalarBool"): - return visitor.visitScalarBool(self) - else: - return visitor.visitChildren(self) - - - class ScalarIntContext(ScalarContext): - - def __init__(self, parser, ctx): # actually a RelayParser.ScalarContext) - super(RelayParser.ScalarIntContext, self).__init__(parser) - self.copyFrom(ctx) - - def NAT(self): - return self.getToken(RelayParser.NAT, 0) - - def accept(self, visitor): - if hasattr(visitor, "visitScalarInt"): - return visitor.visitScalarInt(self) - else: - return visitor.visitChildren(self) - - - - def scalar(self): - - localctx = RelayParser.ScalarContext(self, self._ctx, self.state) - self.enterRule(localctx, 32, self.RULE_scalar) - try: - self.state = 321 - self._errHandler.sync(self) - token = self._input.LA(1) - if token in [RelayParser.FLOAT]: - localctx = RelayParser.ScalarFloatContext(self, localctx) - self.enterOuterAlt(localctx, 1) - self.state = 318 - self.match(RelayParser.FLOAT) - pass - elif token in [RelayParser.NAT]: - localctx = RelayParser.ScalarIntContext(self, localctx) - self.enterOuterAlt(localctx, 2) - self.state = 319 - self.match(RelayParser.NAT) - pass - elif token in [RelayParser.BOOL_LIT]: - localctx = RelayParser.ScalarBoolContext(self, localctx) - self.enterOuterAlt(localctx, 3) - self.state = 320 - self.match(RelayParser.BOOL_LIT) - pass - else: - raise NoViableAltException(self) - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - class IdentContext(ParserRuleContext): - - def __init__(self, parser, parent=None, invokingState=-1): - super(RelayParser.IdentContext, self).__init__(parent, invokingState) - self.parser = parser - - def opIdent(self): - return self.getTypedRuleContext(RelayParser.OpIdentContext,0) - - - def GLOBAL_VAR(self): - return self.getToken(RelayParser.GLOBAL_VAR, 0) - - def LOCAL_VAR(self): - return self.getToken(RelayParser.LOCAL_VAR, 0) - - def GRAPH_VAR(self): - return self.getToken(RelayParser.GRAPH_VAR, 0) - - def getRuleIndex(self): - return RelayParser.RULE_ident - - def accept(self, visitor): - if hasattr(visitor, "visitIdent"): - return visitor.visitIdent(self) - else: - return visitor.visitChildren(self) - - - - - def ident(self): - - localctx = RelayParser.IdentContext(self, self._ctx, self.state) - self.enterRule(localctx, 34, self.RULE_ident) - try: - self.state = 327 - self._errHandler.sync(self) - token = self._input.LA(1) - if token in [RelayParser.CNAME]: - self.enterOuterAlt(localctx, 1) - self.state = 323 - self.opIdent() - pass - elif token in [RelayParser.GLOBAL_VAR]: - self.enterOuterAlt(localctx, 2) - self.state = 324 - self.match(RelayParser.GLOBAL_VAR) - pass - elif token in [RelayParser.LOCAL_VAR]: - self.enterOuterAlt(localctx, 3) - self.state = 325 - self.match(RelayParser.LOCAL_VAR) - pass - elif token in [RelayParser.GRAPH_VAR]: - self.enterOuterAlt(localctx, 4) - self.state = 326 - self.match(RelayParser.GRAPH_VAR) - pass - else: - raise NoViableAltException(self) - - except RecognitionException as re: - localctx.exception = re - self._errHandler.reportError(self, re) - self._errHandler.recover(self, re) - finally: - self.exitRule() - return localctx - - - - def sempred(self, localctx, ruleIndex, predIndex): - if self._predicates == None: - self._predicates = dict() - self._predicates[2] = self.expr_sempred - pred = self._predicates.get(ruleIndex, None) - if pred is None: - raise Exception("No predicate with index:" + str(ruleIndex)) - else: - return pred(localctx, predIndex) - - def expr_sempred(self, localctx, predIndex): - if predIndex == 0: - return self.precpred(self._ctx, 16) - - - if predIndex == 1: - return self.precpred(self._ctx, 15) - - - if predIndex == 2: - return self.precpred(self._ctx, 14) - - - if predIndex == 3: - return self.precpred(self._ctx, 13) - - - if predIndex == 4: - return self.precpred(self._ctx, 4) - - - if predIndex == 5: - return self.precpred(self._ctx, 18) - - - - - diff --git a/python/tvm/relay/grammar/py2/RelayVisitor.py b/python/tvm/relay/grammar/py2/RelayVisitor.py deleted file mode 100644 index eae67d8..0000000 --- a/python/tvm/relay/grammar/py2/RelayVisitor.py +++ /dev/null @@ -1,192 +0,0 @@ -# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2 -from antlr4 import * - -# This class defines a complete generic visitor for a parse tree produced by RelayParser. - -class RelayVisitor(ParseTreeVisitor): - - # Visit a parse tree produced by RelayParser#opIdent. - def visitOpIdent(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#prog. - def visitProg(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#identExpr. - def visitIdentExpr(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#call. - def visitCall(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#neg. - def visitNeg(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#tuple. - def visitTuple(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#parens. - def visitParens(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#funcExpr. - def visitFuncExpr(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#scalarExpr. - def visitScalarExpr(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#let. - def visitLet(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#tensor. - def visitTensor(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#ifElse. - def visitIfElse(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#graph. - def visitGraph(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#binOp. - def visitBinOp(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#func. - def visitFunc(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#defn. - def visitDefn(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#argList. - def visitArgList(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#varList. - def visitVarList(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#var. - def visitVar(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#attrList. - def visitAttrList(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#attr. - def visitAttr(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#typeParamSeq. - def visitTypeParamSeq(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#tupleType. - def visitTupleType(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#typeIdentType. - def visitTypeIdentType(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#tensorType. - def visitTensorType(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#funcType. - def visitFuncType(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#incompleteType. - def visitIncompleteType(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#intType. - def visitIntType(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#shapeSeq. - def visitShapeSeq(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#parensShape. - def visitParensShape(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#intShape. - def visitIntShape(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#typeIdent. - def visitTypeIdent(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#body. - def visitBody(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#scalarFloat. - def visitScalarFloat(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#scalarInt. - def visitScalarInt(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#scalarBool. - def visitScalarBool(self, ctx): - return self.visitChildren(ctx) - - - # Visit a parse tree produced by RelayParser#ident. - def visitIdent(self, ctx): - return self.visitChildren(ctx) - - diff --git a/python/tvm/relay/grammar/py2/__init__.py b/python/tvm/relay/grammar/py2/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/python/tvm/relay/grammar/py3/.gitattributes b/python/tvm/relay/grammar/py3/.gitattributes deleted file mode 100644 index 4adf65f..0000000 --- a/python/tvm/relay/grammar/py3/.gitattributes +++ /dev/null @@ -1,3 +0,0 @@ -Relay* binary -Relay* linguist-generated=true -Relay* linguist-detectable=false \ No newline at end of file diff --git a/python/tvm/relay/grammar/py3/RelayLexer.py b/python/tvm/relay/grammar/py3/RelayLexer.py index fbf74bf..53bfbad 100644 --- a/python/tvm/relay/grammar/py3/RelayLexer.py +++ b/python/tvm/relay/grammar/py3/RelayLexer.py @@ -1,11 +1,10 @@ -# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2 +# Generated from /home/marisa/Work/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.1 from antlr4 import * from io import StringIO from typing.io import TextIO import sys - def serializedATN(): with StringIO() as buf: buf.write("\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\2*") @@ -195,7 +194,7 @@ class RelayLexer(Lexer): def __init__(self, input=None, output:TextIO = sys.stdout): super().__init__(input, output) - self.checkVersion("4.7.2") + self.checkVersion("4.7.1") self._interp = LexerATNSimulator(self, self.atn, self.decisionsToDFA, PredictionContextCache()) self._actions = None self._predicates = None diff --git a/python/tvm/relay/grammar/py3/RelayParser.py b/python/tvm/relay/grammar/py3/RelayParser.py index ff5cffc..bf8332c 100644 --- a/python/tvm/relay/grammar/py3/RelayParser.py +++ b/python/tvm/relay/grammar/py3/RelayParser.py @@ -1,11 +1,10 @@ -# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2 +# Generated from /home/marisa/Work/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.1 # encoding: utf-8 from antlr4 import * from io import StringIO from typing.io import TextIO import sys - def serializedATN(): with StringIO() as buf: buf.write("\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\3*") @@ -260,13 +259,12 @@ class RelayParser ( Parser ): def __init__(self, input:TokenStream, output:TextIO = sys.stdout): super().__init__(input, output) - self.checkVersion("4.7.2") + self.checkVersion("4.7.1") self._interp = ParserATNSimulator(self, self.atn, self.decisionsToDFA, self.sharedContextCache) self._predicates = None - class OpIdentContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -304,7 +302,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class ProgContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -381,7 +378,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class ExprContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -440,8 +436,6 @@ class RelayParser ( Parser ): super().__init__(parser) self.copyFrom(ctx) - def SUB(self): - return self.getToken(RelayParser.SUB, 0) def expr(self): return self.getTypedRuleContext(RelayParser.ExprContext,0) @@ -628,26 +622,6 @@ class RelayParser ( Parser ): else: return self.getTypedRuleContext(RelayParser.ExprContext,i) - def MUL(self): - return self.getToken(RelayParser.MUL, 0) - def DIV(self): - return self.getToken(RelayParser.DIV, 0) - def ADD(self): - return self.getToken(RelayParser.ADD, 0) - def SUB(self): - return self.getToken(RelayParser.SUB, 0) - def LT(self): - return self.getToken(RelayParser.LT, 0) - def GT(self): - return self.getToken(RelayParser.GT, 0) - def LE(self): - return self.getToken(RelayParser.LE, 0) - def GE(self): - return self.getToken(RelayParser.GE, 0) - def EQ(self): - return self.getToken(RelayParser.EQ, 0) - def NE(self): - return self.getToken(RelayParser.NE, 0) def accept(self, visitor:ParseTreeVisitor): if hasattr( visitor, "visitBinOp" ): @@ -1037,7 +1011,6 @@ class RelayParser ( Parser ): self.unrollRecursionContexts(_parentctx) return localctx - class FuncContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1115,7 +1088,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class DefnContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1199,7 +1171,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class ArgListContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1265,7 +1236,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class VarListContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1327,7 +1297,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class VarContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1381,7 +1350,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class AttrListContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1442,7 +1410,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class AttrContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1488,7 +1455,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class TypeParamSeqContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1562,7 +1528,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class Type_Context(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1836,7 +1801,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class ShapeSeqContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -1924,7 +1888,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class ShapeContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -2010,7 +1973,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class TypeIdentContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -2048,7 +2010,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class BodyContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -2091,7 +2052,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class ScalarContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): @@ -2194,7 +2154,6 @@ class RelayParser ( Parser ): self.exitRule() return localctx - class IdentContext(ParserRuleContext): def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1): diff --git a/python/tvm/relay/grammar/py3/RelayVisitor.py b/python/tvm/relay/grammar/py3/RelayVisitor.py index 64308dc..9e3631f 100644 --- a/python/tvm/relay/grammar/py3/RelayVisitor.py +++ b/python/tvm/relay/grammar/py3/RelayVisitor.py @@ -1,4 +1,4 @@ -# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2 +# Generated from /home/marisa/Work/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.1 from antlr4 import * if __name__ is not None and "." in __name__: from .RelayParser import RelayParser diff --git a/python/tvm/relay/grammar/py3/__init__.py b/python/tvm/relay/grammar/py3/__init__.py deleted file mode 100644 index e69de29..0000000