install.yml

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 .
        shell: bash
        env:
          TORCH_CUDA_ARCH_LIST: "5.0+PTX;6.0;7.0;7.5;8.0;8.6"

      - 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"