Skip to content
This repository was archived by the owner on Aug 1, 2025. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion CITATION.cff
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,4 @@ authors:
given-names: Jason
title: "TorchDynamo"
date-released: 2022-01-01
url: https://github.com/facebookresearch/torchdynamot
url: https://github.com/pytorch/torchdynamo
12 changes: 6 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -345,7 +345,7 @@ conda activate torchdynamo
# for CUDA version, replace `cpuonly` with `cudatoolkit=11.3`
conda install pytorch torchvision torchaudio torchtext cpuonly -c pytorch-nightly

git clone git@github.com:facebookresearch/torchdynamo.git
git clone git@github.com:pytorch/torchdynamo.git
cd torchdynamo
pip install -r requirements.txt

Expand Down Expand Up @@ -381,10 +381,10 @@ make lint-deps

## Tests

[![Test Python 3.7](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py37.yml/badge.svg)](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py37.yml)
[![Test Python 3.8](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py38.yml/badge.svg)](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py38.yml)
[![Test Python 3.9](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py39.yml/badge.svg)](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py39.yml)
[![Test Python 3.10](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py310.yml/badge.svg)](https://github.com/facebookresearch/torchdynamo/actions/workflows/test-py39.yml)
[![Test Python 3.7](https://github.com/pytorch/torchdynamo/actions/workflows/test-py37.yml/badge.svg)](https://github.com/pytorch/torchdynamo/actions/workflows/test-py37.yml)
[![Test Python 3.8](https://github.com/pytorch/torchdynamo/actions/workflows/test-py38.yml/badge.svg)](https://github.com/pytorch/torchdynamo/actions/workflows/test-py38.yml)
[![Test Python 3.9](https://github.com/pytorch/torchdynamo/actions/workflows/test-py39.yml/badge.svg)](https://github.com/pytorch/torchdynamo/actions/workflows/test-py39.yml)
[![Test Python 3.10](https://github.com/pytorch/torchdynamo/actions/workflows/test-py310.yml/badge.svg)](https://github.com/pytorch/torchdynamo/actions/workflows/test-py39.yml)

Run tests with
```shell
Expand Down Expand Up @@ -430,7 +430,7 @@ generate `*.csv` files. See `./torchbench.py --help` for more options.

## Linting and Automatic Code Formatting

[![Lint](https://github.com/facebookresearch/torchdynamo/actions/workflows/lint.yml/badge.svg)](https://github.com/facebookresearch/torchdynamo/actions/workflows/lint.yml)
[![Lint](https://github.com/pytorch/torchdynamo/actions/workflows/lint.yml/badge.svg)](https://github.com/pytorch/torchdynamo/actions/workflows/lint.yml)
[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
[![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort/)

Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
setup(
name="torchdynamo",
version="0.2.0",
url="https://github.com/facebookresearch/torchdynamo",
url="https://github.com/pytorch/torchdynamo",
description="A Python-level JIT compiler designed to make unmodified PyTorch programs faster.",
long_description=long_description,
long_description_content_type="text/markdown",
Expand Down
2 changes: 1 addition & 1 deletion tests/test_repros.py
Original file line number Diff line number Diff line change
Expand Up @@ -1030,7 +1030,7 @@ def test_reformer_sorting(self):
self.assertEqual(cnt.op_count, ifdyn(28, 14))

def test_recursive_map(self):
# https://github.com/facebookresearch/torchdynamo/issues/132
# https://github.com/pytorch/torchdynamo/issues/132
def _recursive_map(struct, batch_dim=0):
for k, v in struct.items():
if v is not None:
Expand Down
4 changes: 2 additions & 2 deletions torchbench.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,9 +61,9 @@
SKIP = {
# non-deterministic output / cant check correctness
"pyhpc_turbulent_kinetic_energy",
# https://github.com/facebookresearch/torchdynamo/issues/101
# https://github.com/pytorch/torchdynamo/issues/101
"detectron2_maskrcnn",
# https://github.com/facebookresearch/torchdynamo/issues/145
# https://github.com/pytorch/torchdynamo/issues/145
"fambench_xlmr",
}

Expand Down