dialects: (onnx) remove ONNX-related code

.github/workflows/ci-mlir.yml

@@ -109,13 +109,6 @@ jobs:
         export PATH=$PATH:${GITHUB_WORKSPACE}/llvm-project/build/bin/
         uv run pytest --nbval docs/mlir_interoperation.ipynb --maxfail 1 -vv
 
-    - name: Test ONNX-dependent marimo notebooks
-      run: |
-        cd xdsl
-        # Add mlir-opt to the path
-        export PATH=$PATH:${GITHUB_WORKSPACE}/llvm-project/build/bin/
-        uv run make tests-marimo-onnx
-
     - name: Combine coverage data
       run: |
         cd xdsl

Makefile

@@ -98,30 +98,10 @@ tests-marimo: uv-installed
 	done
 	@echo "All marimo tests passed successfully."
 
-.PHONY: tests-marimo-onnx
-tests-marimo-onnx: uv-installed
-	@if uv run python -c "import onnx" > /dev/null 2>&1; then \
-		echo "onnx is installed, running tests."; \
-		if ! command -v mlir-opt > /dev/null 2>&1; then \
-			echo "MLIR is not installed, skipping tests."; \
-			exit 0; \
-		fi; \
-		for file in docs/marimo/onnx/*.py; do \
-			echo "Running $$file"; \
-			error_message=$$(uv run python3 "$$file" 2>&1) || { \
-				echo "Error running $$file"; \
-				echo "$$error_message"; \
-				exit 1; \
-			}; \
-		done; \
-		echo "All marimo onnx tests passed successfully."; \
-	else \
-		echo "onnx is not installed, skipping tests."; \
-	fi
 
 # run all tests
 .PHONY: tests-functional
-tests-functional: pytest tests-toy filecheck pytest-nb tests-marimo tests-marimo-onnx
+tests-functional: pytest tests-toy filecheck pytest-nb tests-marimo
 	@echo All functional tests done.
 
 # run all tests

README.md

@@ -53,7 +53,7 @@ In order to keep the set of dependencies ot a minimum, these extra dependencies
 specified explicitly. To install these, use:
 
 ``` bash
-pip install xdsl[gui,jax,riscv,onnx]
+pip install xdsl[gui,jax,riscv]
 ```
 
 To install the testing/development dependencies, use:

docs/marimo/__marimo__/linalg_snitch.ipynb

@@ -591,7 +591,7 @@
     "riscv_op_counter = OpCounter()\n",
     "riscv_interpreter = Interpreter(riscv_module, listeners=(riscv_op_counter,))\n",
     "\n",
-    "register_implementations(riscv_interpreter, riscv_ctx, include_onnx=False)\n",
+    "register_implementations(riscv_interpreter, riscv_ctx)\n",
     "\n",
     "riscv_interpreter.call_op(\"matmul\", (a_shaped.data_ptr.raw, b_shaped.data_ptr.raw, riscv_c_shaped.data_ptr.raw))\n",
     "\n",
@@ -633,7 +633,7 @@
     "\n",
     "snitch_c_shaped = ShapedArray(TypedPtr.new_float64([0.0] * c_len), c_shape)\n",
     "\n",
-    "register_implementations(snitch_interpreter, snitch_stream_ctx, include_onnx=False)\n",
+    "register_implementations(snitch_interpreter, snitch_stream_ctx)\n",
     "\n",
     "snitch_interpreter.call_op(\n",
     "    \"matmul\",\n",

docs/marimo/linalg_snitch.py

@@ -513,7 +513,7 @@ def _(TypedPtr, a_shape, b_shape, c_shape, mo, riscv_ctx, riscv_module):
     riscv_op_counter = OpCounter()
     riscv_interpreter = Interpreter(riscv_module, listeners=(riscv_op_counter,))
 
-    register_implementations(riscv_interpreter, riscv_ctx, include_onnx=False)
+    register_implementations(riscv_interpreter, riscv_ctx)
 
     riscv_interpreter.call_op("matmul", (a_shaped.data_ptr.raw, b_shaped.data_ptr.raw, riscv_c_shaped.data_ptr.raw))
 
@@ -566,7 +566,7 @@ def _(
 
     snitch_c_shaped = ShapedArray(TypedPtr.new_float64([0.0] * c_len), c_shape)
 
-    register_implementations(snitch_interpreter, snitch_stream_ctx, include_onnx=False)
+    register_implementations(snitch_interpreter, snitch_stream_ctx)
 
     snitch_interpreter.call_op(
         "matmul",