Skip to content

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper #755

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper

[FSDPv1] Only perform cat() during last microbatch backward() within FlattenParamsWrapper #755

Workflow file for this run

name: pre-commit
on:
pull_request:
push:
branches: [main]
jobs:
pre-commit:
runs-on: ubuntu-latest
strategy:
matrix:
# make sure python versions are consistent with those used in .circleci/config.yml
python-version: ['3.7.12', '3.8.12', '3.9.7']
steps:
- uses: actions/checkout@v2
- uses: actions/setup-python@v2
with:
python-version: ${{ matrix.python-version }}
- uses: pre-commit/[email protected]