Skip to content

add dense kernel for all sharding types #3721

add dense kernel for all sharding types

add dense kernel for all sharding types #3721

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