Skip to content

In gradient clipping, if DTensors are used, need to first convert them to local tensors #4000

In gradient clipping, if DTensors are used, need to first convert them to local tensors

In gradient clipping, if DTensors are used, need to first convert them to local tensors #4000

Workflow file for this run

name: Pyre Check
on:
push:
branches: [main]
pull_request:
jobs:
pyre-check:
runs-on: ubuntu-22.04
defaults:
run:
shell: bash -el {0}
steps:
- uses: conda-incubator/setup-miniconda@v2
with:
python-version: 3.8
- name: Checkout Torchrec
uses: actions/checkout@v2
- name: Install dependencies
run: >
conda install --yes pytorch cpuonly -c pytorch-nightly &&
pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu &&
pip install -r requirements.txt &&
pip install pyre-check-nightly==$(cat .pyre_configuration | grep version | awk '{print $2}' | sed 's/\"//g')
- name: Pyre check
run: pyre check