name: Build Xformers

on: [push]

jobs:
  build:
    runs-on: ubuntu-latest
    # container:
    #   image: pytorch/manylinux-cuda113
    container:
      image: python:3.10-slim
      env:
        DEBIAN_FRONTEND: noninteractive
        XFORMERS_DISABLE_FLASH_ATTN: 1
        FORCE_CUDA: 1
        # TORCH_CUDA_ARCH_LIST: "3.7;5.0;5.2;6.0;6.1+PTX;7.0;7.5+PTX;8.0;8.6+PTX"
        TORCH_CUDA_ARCH_LIST: "6.1"
        NVCC_FLAGS: "--use_fast_math -DXFORMERS_MEM_EFF_ATTENTION_DISABLE_BACKWARD"
    steps:
      - run: |
          apt-get update
          apt-get install gpg wget git -y
          wget https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64/cuda-keyring_1.0-1_all.deb
          dpkg -i cuda-keyring_1.0-1_all.deb
          apt-get update
          apt-get install cuda-nvcc-11-8 cuda-libraries-dev-11-8 -y

          export PIP_CACHE_DIR=$(pwd)
          pip install ninja torch==1.12.1+cu113
          pip wheel --wheel-dir=data git+https://github.com/facebookresearch/xformers.git@ba93c5012d00bd1b010514a7bc9bd938c1ad6149#egg=xformers
      - name: Archive production artifacts
        uses: actions/upload-artifact@v3
        with:
          name: dist-without-markdown
          path: data