Ahead-of-time build and PyTorch Stable ABI #127

	name: OEQ C++ Extension Build Verification

	on:
	push:
	branches: [ "main" ]
	pull_request:
	branches: [ "main" ]
	types: [ labeled ]

	permissions:
	contents: read

	jobs:
	verify_cuda_extension:
	if: ${{ github.event.label.name == 'ci-ready' \|\| github.event_name != 'pull_request' }}
	runs-on: ubuntu-latest

	steps:
	- uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: "3.10"
	cache: 'pip'
	cache-dependency-path: '**/requirements_cuda_ci.txt'
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	sudo apt-get update
	sudo apt install nvidia-cuda-toolkit
	pip install -r .github/workflows/requirements_cuda_ci.txt
	pip install -e "./openequivariance"

	- name: Test CUDA extension build via import
	run: \|
	pytest \
	tests/import_test.py::test_extension_built \
	tests/import_test.py::test_torch_extension_built

	export OEQ_JIT_EXTENSION=1

	pytest \
	tests/import_test.py::test_extension_built \
	tests/import_test.py::test_torch_extension_built

	- name: Test JAX extension build
	run: \|
	XLA_DIRECT_DOWNLOAD=1 pip install -e "./openequivariance_extjax" --no-build-isolation

Provide feedback