Skip to content

Move softmax_csr autograd module to C++ #1909

Move softmax_csr autograd module to C++

Move softmax_csr autograd module to C++ #1909

Workflow file for this run

name: Minimal Installation
on: # yamllint disable-line rule:truthy
push:
branches:
- master
pull_request:
jobs:
import:
runs-on: ubuntu-latest
strategy:
matrix:
cuda-version: ['cpu', 'cu121']
steps:
- name: Checkout repository
uses: actions/checkout@v4
with:
submodules: recursive
- name: Setup packages
uses: ./.github/actions/setup
with:
cuda-version: ${{ matrix.cuda-version }}
- name: Install package
run: |
source ./.github/workflows/cuda/${{ runner.os }}-env.sh ${{ matrix.cuda-version }}
pip install --verbose -e .
- name: Test imports
run: |
python -c 'import pyg_lib;print(pyg_lib.cuda_version())'
python -c "import pyg_lib.ops"
python -c "import pyg_lib.sampler"