add ci

.github/workflows/README.md

@@ -0,0 +1,46 @@
+# GitHub Workflows
+
+This directory contains GitHub Actions workflows for automated testing and continuous integration.
+
+## Workflows
+
+### e2e_gcn.yml
+
+**Purpose**: End-to-end testing of GCN model on Cora dataset
+
+**Trigger**: Runs on push to main branch or pull requests targeting main branch
+
+**Environment**: Ubuntu 22.04 LTS (CPU-only)
+
+**Steps**:
+1. Checkout code
+2. Set up Python 3.10
+3. Install system dependencies (build-essential, python3-dev)
+4. Install Python dependencies from requirements_ci.txt
+5. Run GCN example on Cora dataset with JITTOR_USE_CUDA=0
+
+## Requirements Files
+
+### requirements_ci.txt
+
+CPU-only requirements file for GitHub Actions environment.
+
+**Key differences from main requirements.txt**:
+- Removed CUDA-dependent packages (cupy)
+- All other dependencies remain the same
+- Ensures compatibility with GitHub's CPU-only runners
+
+## Usage
+
+Workflows run automatically on push/PR events. To manually trigger:
+
+1. Go to GitHub repository
+2. Click "Actions" tab
+3. Select the workflow
+4. Click "Run workflow"
+
+## Notes
+
+- All workflows run in CPU-only mode
+- JITTOR_USE_CUDA=0 is enforced for all tests
+- Results are visible in GitHub Actions interface

.github/workflows/e2e_gcn.yml

@@ -0,0 +1,32 @@
+name: E2E GCN Testing
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  e2e_gcn:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.10'
+    - name: Install system dependencies
+      run: |
+        sudo apt-get update
+        sudo apt-get install -y build-essential python3-dev
+    - name: Install Python dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r .github/workflows/requirements_ci.txt
+        pip install -e .
+    - name: Run GCN example on Cora dataset
+      run: |
+        export JITTOR_USE_CUDA=0
+        python examples/gcn_example.py --dataset cora
+      env:
+        JITTOR_USE_CUDA: 0

.github/workflows/requirements_ci.txt

@@ -0,0 +1,21 @@
+ase==3.24.0
+astunparse==1.6.3
+autograd==1.7.0
+Flask==3.1.0
+huggingface_hub==0.27.1
+jittor==1.3.9.14
+numpy==1.24.0
+pandas==2.2.3
+Pillow==11.1.0
+PyMetis==2023.1.1
+pyparsing==3.2.1
+pywebio==1.8.3
+recommonmark==0.7.1
+schnetpack==2.0.0
+scikit_learn==1.6.1
+scipy==1.15.1
+setuptools==69.5.1
+six==1.16.0
+sphinx_rtd_theme==3.0.2
+sympy==1.13.3
+tqdm==4.66.4

jittor_geometric/ops/cpp/edgesoftmax_op.cc

@@ -5,9 +5,9 @@
  */
 
 #include  "var.h"
-#include  <cuda.h>
 #include  "edgesoftmax_op.h"
 #ifdef JIT_cuda
+#include  <cuda.h>
 // #include <cub/device/device_segmented_radix_sort.cuh>
 #include <cub/cub.cuh>
 #endif