build: work around GCC optimizer bug for aarch64

.github/workflows/pr.yml

@@ -14,6 +14,10 @@ jobs:
   pull-request:
     runs-on: ubuntu-latest
     steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
       - name: Create app token
         uses: actions/create-github-app-token@v2
         id: app-token
@@ -25,6 +29,5 @@ jobs:
           gh pr create -d -B master -H "$GITHUB_REF_NAME"
           "${PR_TITLE:+--title=}${PR_TITLE:---fill-first}"
         env:
-          GH_REPO: ${{github.repository}}
           GH_TOKEN: ${{steps.app-token.outputs.token}}
           PR_TITLE: ${{github.event.inputs.title}}

.github/workflows/pypi.yml

@@ -49,7 +49,7 @@ jobs:
           CIBW_ARCHS: ${{matrix.cibw_arch}}
           CIBW_SKIP: "*-musllinux_aarch64"
           # FIXME: stop skipping when the tests stop crashing
-          CIBW_TEST_SKIP: "*-win_* *linux_aarch64"
+          CIBW_TEST_SKIP: "*-win_*"
       - name: Upload wheels
         uses: actions/upload-artifact@v4
         with:

setup.py

@@ -1,7 +1,21 @@
-from sys import platform
+from os import name as os_name
+from platform import machine
 
 from setuptools import Extension, setup  # type: ignore
 
+if os_name != "nt":
+    cflags = [
+        "-std=c11",
+        "-fvisibility=hidden",
+        "-Wno-cast-function-type",
+        "-Werror=implicit-function-declaration",
+    ]
+    # FIXME: GCC optimizer bug workaround for #330 & #386
+    if machine().startswith("aarch64"):
+        cflags.append("--param=early-inlining-insns=9")
+else:
+    cflags = ["/std:c11", "/wd4244"]
+
 setup(
     packages=["tree_sitter"],
     include_package_data=False,
@@ -36,18 +50,7 @@
                 ("PY_SSIZE_T_CLEAN", None),
                 ("TREE_SITTER_HIDE_SYMBOLS", None),
             ],
-            undef_macros=[
-                "TREE_SITTER_FEATURE_WASM",
-            ],
-            extra_compile_args=[
-                "-std=c11",
-                "-fvisibility=hidden",
-                "-Wno-cast-function-type",
-                "-Werror=implicit-function-declaration",
-            ] if platform != "win32" else [
-                "/std:c11",
-                "/wd4244",
-            ],
+            extra_compile_args=cflags,
         )
     ],
 )

tree_sitter/binding/parser.c

@@ -101,10 +101,7 @@ static bool parser_progress_callback(TSParseState *state) {
 PyObject *parser_parse(Parser *self, PyObject *args, PyObject *kwargs) {
     ModuleState *state = GET_MODULE_STATE(self);
     PyObject *source_or_callback;
-    PyObject *old_tree_obj = NULL;
-    PyObject *encoding_obj = NULL;
-    PyObject *progress_callback_obj = NULL;
-    bool keep_text = true;
+    PyObject *old_tree_obj = NULL, *encoding_obj = NULL, *progress_callback_obj = NULL;
     char *keywords[] = {"", "old_tree", "encoding", "progress_callback", NULL};
     if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O|O!OO:parse", keywords, &source_or_callback,
                                      state->tree_type, &old_tree_obj, &encoding_obj,
@@ -206,7 +203,7 @@ PyObject *parser_parse(Parser *self, PyObject *args, PyObject *kwargs) {
     }
     tree->tree = new_tree;
     tree->language = self->language;
-    tree->source = keep_text ? source_or_callback : Py_None;
+    tree->source = source_or_callback;
     Py_INCREF(tree->source);
     Py_INCREF(tree->language);
     return PyObject_Init((PyObject *)tree, state->tree_type);