metaopt · XuehaiPan · Jul 10, 2022 · Jul 7, 2022 · Jul 7, 2022 · Jul 7, 2022
diff --git a/.editorconfig b/.editorconfig
@@ -14,6 +14,9 @@ insert_final_newline = true
 indent_size = 4
 src_paths=torchopt,tests,examples
 
+[*.{yaml,yml}]
+indent_size = 2
+
 [*.md]
 indent_size = 2
 x-soft-wrap-text = true
@@ -25,11 +28,5 @@ x-soft-wrap-text = true
 [Makefile]
 indent_style = tab
 
-[*.cpp]
-indent_size = 2
-
-[*.h]
-indent_size = 2
-
-[*.cuh?]
+[*.{cpp,h,cu,cuh}]
 indent_size = 2
diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
@@ -1,44 +1,78 @@
 name: Lint
 
-on: [push, pull_request]
+on:
+  push:
+  pull_request:
+
+permissions:
+  contents: read
 
 jobs:
   lint:
     runs-on: ubuntu-latest
+    timeout-minutes: 30
     steps:
-    - name: Cancel previous run
-      uses: styfle/cancel-workflow-action@0.9.1
-      with:
-        access_token: ${{ github.token }}
-    - uses: actions/checkout@v2
-    - name: Set up Python 3.8
-      uses: actions/setup-python@v2
-      with:
-        python-version: 3.8
-    - name: Upgrade pip
-      run: |
-        python -m pip install --upgrade pip
-    - name: Install requirements
-      run: |
-        python -m pip install -r requirements.txt
-    - name: Install dependencies
-      run: |
-        python -m pip install -e .
-    - name: flake8
-      run: |
-        make flake8
-    - name: isort and yapf
-      run: |
-        make py-format
-    - name: addlicense
-      run: |
-        make addlicense
-    - name: mypy
-      run: |
-        make mypy
+      - name: Cancel previous run
+        uses: styfle/cancel-workflow-action@0.10.0
+        with:
+          access_token: ${{ github.token }}
+
+      - name: Checkout
+        uses: actions/checkout@v3
+        with:
+          submodules: "recursive"
+          fetch-depth: 1
+
+      - name: Set up Python 3.7  # the lowest version we support
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.7"
+          update-environment: true
+
+      - name: Setup CUDA Toolkit
+        uses: Jimver/cuda-toolkit@v0.2.7
+        id: cuda-toolkit
+        with:
+          cuda: "11.3.1"
+          method: network
+          sub-packages: '["nvcc"]'
+      - run: |
+          echo "Installed CUDA version is: ${{steps.cuda-toolkit.outputs.cuda}}"
+          echo "CUDA install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
+          nvcc -V
+
+      - name: Upgrade pip
+        run: |
+          python -m pip install --upgrade pip setuptools
+
+      - name: Install dependencies
+        run: |
+          python -m pip install -r tests/requirements.txt
+
+      - name: Install TorchOpt
+        run: |
+          python -m pip install -e .
+
+      - name: flake8
+        run: |
+          make flake8
+
+      - name: isort and yapf
+        run: |
+          make py-format
+
+      - name: addlicense
+        run: |
+          make addlicense
+
+      - name: mypy
+        run: |
+          make mypy
+
     # - name: docstyle
     #   run: |
     #     make docstyle
+
     # - name: spelling
     #   run: |
-    #     make spelling
+    #     make spelling
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -7,28 +7,91 @@ on:
 jobs:
   deploy:
     runs-on: ubuntu-latest
+    timeout-minutes: 30
     steps:
-    - uses: actions/checkout@v2
-    - name: Set up Python
-      uses: actions/setup-python@v1
-      with:
-        python-version: '3.x'
-    - name: Check consistency between the package version and release tag
-      run: |
-        RELEASE_VER=${GITHUB_REF#refs/*/}
-        PACKAGE_VER="v`python setup.py --version`"
-        if [ $RELEASE_VER != $PACKAGE_VER ]
-        then
-          echo "package ver. ($PACKAGE_VER) != release ver. ($RELEASE_VER)"; exit 1
-        fi
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install setuptools wheel twine
-    - name: Build and publish
-      env:
-        TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
-        TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
-      run: |
-        python setup.py sdist bdist_wheel
-        twine upload dist/*
+      - name: Checkout
+        uses: actions/checkout@v3
+        with:
+          submodules: "recursive"
+          fetch-depth: 1
+
+      - name: Set up Python 3.7
+        id: py37
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.7"
+          update-environment: false
+
+      - name: Set up Python 3.8
+        id: py38
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.8"
+          update-environment: false
+
+      - name: Set up Python 3.9
+        id: py39
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.9"
+          update-environment: false
+
+      - name: Set up Python 3.10
+        id: py310
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+          update-environment: false
+
+      - name: Set up Python executable paths
+        run: |
+          DEFAULT_PYTHON="${{ steps.py37.outputs.python-path }}"
+          echo "DEFAULT_PYTHON='${DEFAULT_PYTHON}'" >> "${GITHUB_ENV}"
+
+          PYTHON_EXECUTABLES="${{ steps.py37.outputs.python-path }}"
+          PYTHON_EXECUTABLES="${PYTHON_EXECUTABLES}:${{ steps.py38.outputs.python-path }}"
+          PYTHON_EXECUTABLES="${PYTHON_EXECUTABLES}:${{ steps.py39.outputs.python-path }}"
+          PYTHON_EXECUTABLES="${PYTHON_EXECUTABLES}:${{ steps.py310.outputs.python-path }}"
+          echo "PYTHON_EXECUTABLES='${PYTHON_EXECUTABLES}'" >> "${GITHUB_ENV}"
+
+      - name: Check consistency between the package version and release tag
+        run: |
+          RELEASE_VER="${GITHUB_REF#refs/*/}"
+          PACKAGE_VER="v$(python setup.py --version)"
+          if [[ "${RELEASE_VER}" != "${PACKAGE_VER}" ]]; then
+            echo "package ver. (${PACKAGE_VER}) != release ver. (${RELEASE_VER})"
+            exit 1
+          fi
+
+      - name: Setup CUDA Toolkit
+        uses: Jimver/cuda-toolkit@v0.2.7
+        id: cuda-toolkit
+        with:
+          cuda: "11.3.1"
+          method: network
+          sub-packages: '["nvcc"]'
+      - run: |
+          echo "Installed CUDA version is: ${{steps.cuda-toolkit.outputs.cuda}}"
+          echo "CUDA install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
+          nvcc -V
+
+      - name: Build sdist and wheels
+        run: |
+          while IFS='' read -rd':' PYTHON || [[ -n "${PYTHON}" ]]; do
+            [[ -z "${PYTHON}" ]] && continue
+            echo "Building wheel with Python: ${PYTHON} ($("${PYTHON}" --version))"
+            "${PYTHON}" -m pip install --upgrade pip setuptools wheel build
+            if [[ "${PYTHON}" == "${DEFAULT_PYTHON}" ]]; then
+              "${PYTHON}" -m build
+            else
+              "${PYTHON}" -m build --wheel
+            fi
+          done <<< "${PYTHON_EXECUTABLES}"
+
+      - name: Publish to PyPI
+        env:
+          TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
+          TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
+        run: |
+          "${DEFAULT_PYTHON}" -m pip install --upgrade twine
+          "${DEFAULT_PYTHON}" -m twine upload dist/*
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -0,0 +1,60 @@
+name: Tests
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+
+permissions:
+  contents: read
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
+    steps:
+      - name: Cancel previous run
+        uses: styfle/cancel-workflow-action@0.10.0
+        with:
+          access_token: ${{ github.token }}
+
+      - name: Checkout
+        uses: actions/checkout@v3
+        with:
+          submodules: "recursive"
+          fetch-depth: 1
+
+      - name: Set up Python 3.7  # the lowest version we support
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.7"
+          update-environment: true
+
+      - name: Setup CUDA Toolkit
+        uses: Jimver/cuda-toolkit@v0.2.7
+        id: cuda-toolkit
+        with:
+          cuda: "11.3.1"
+          method: network
+          sub-packages: '["nvcc"]'
+      - run: |
+          echo "Installed CUDA version is: ${{steps.cuda-toolkit.outputs.cuda}}"
+          echo "CUDA install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
+          nvcc -V
+
+      - name: Upgrade pip
+        run: |
+          python -m pip install --upgrade pip setuptools
+
+      - name: Install dependencies
+        run: |
+          python -m pip install -r tests/requirements.txt
+
+      - name: Install TorchOpt
+        run: |
+          python -m pip install -e .
+
+      - name: Test with pytest
+        run: |
+          make pytest
diff --git a/.gitmodules b/.gitmodules
diff --git a/CITATION.cff b/CITATION.cff
@@ -27,4 +27,4 @@ authors:
 version: 0.4.1
 date-released: "2022-04-09"
 license: Apache-2.0
-repository-code: "https://github.com/metaopt/torchopt"
+repository-code: "https://github.com/metaopt/TorchOpt"