Yuanhao Ji
Yuanhao Ji
Hi, there! Thank you for your contributions first! I just have an question. How to push the docker image with different tags to multiple registries when deploying? I just want...
Fix #124327 `device_type`, the first arg of [init_device_mesh()](https://github.com/pytorch/pytorch/blob/a0466061e17358fb621cfde3f85e0bd6d13cfc55/torch/distributed/device_mesh.py#L503), does not support types with indexes, such as `cuda:0`. If `cuda:0` is used as a parameter, `_get_device_handle()` will not correctly return `torch.cuda`....
## Description Fix the duplicated line in `advanced_source/privateuseone.rst` ```diff 1. Register new backend module to Pytorch. -2. Generate methods and properties related to the new backend. +2. Rename PrivateUse1 to...
- add usage of `word_language_model/generate.py`
## Standards checklist: - [x] The PR title is descriptive. - [x] The PR doesn't replicate another PR which is already open. - [x] I have read the contribution guide...
Fixes #3039 This PR adds a tutorial about the out-of-tree extension autoloading mechanism introduced in https://github.com/pytorch/pytorch/issues/122468 and implemented in https://github.com/pytorch/pytorch/pull/127074 cc: @jgong5 @Yikun @hipudding @FFFrog @albanD
Fixes https://github.com/pytorch/pytorch/issues/134293 cc @NmomoN @mengpenghui @fwenguang @cdzhan @1274085042 @PHLens @ailzhang @albanD
### 🚀 Describe the improvement or the new tutorial This tutorial will help the third party extension writers better to improve seamless integration within PyTorch. ### Existing tutorials on this...
All examples can only be run on CUDA or MPS using `--cuda` or `--mps`. This change adds an argument `--device` to make our examples run on a specific device, e.g.,...