Yuanhao Ji

Results 17 issues of 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`....

oncall: distributed
triaged
open source
ciflow/trunk
ciflow/periodic
merging
release notes: DeviceMesh

## 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...

grammar

- add usage of `word_language_model/generate.py`

cla signed

## 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...

Area: plugin
Type: documentation

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

cla signed
new tutorial
2.5

Fixes https://github.com/pytorch/pytorch/issues/134293 cc @NmomoN @mengpenghui @fwenguang @cdzhan @1274085042 @PHLens @ailzhang @albanD

cla signed
module: PrivateUse1

### 🚀 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.,...

cla signed