-
Platforms: linux
This test was disabled because it is failing in CI. See [recent examples](https://hud.pytorch.org/flakytest?name=test_comprehensive_float_cpu_int64&suite=TestInductorOpInfoCPU&limit=…
-
### Your current environment
The output of `python collect_env.py`
```text
Collecting environment information...
PyTorch version: 2.4.0+cu121
Is debug build: False
CUDA used to build PyTor…
-
I'm trying to deploy a Lambda function for a PyTorch model performing inference on CPU.
In order to keep the 512 MB size limit for the /tmp folder, I want to build PyTorch from source,
so I'm using…
-
This should reduce disk space usage as well as image build times.
-
### 🐛 Describe the bug
When processing complex data type, torch.linalg.vector_norm raises an overflow error.
```python
import torch
>>> torch.linalg.vector_norm(torch.randn(3, 3), torch.tensor(2…
-
### 🐛 Describe the bug
torch.library.custom_op does not document how it can be used with torch.amp. I made it work, but not without implementing a new `custom_setup_context` decorator for the operato…
-
### Your current environment
```text
PyTorch version: 2.1.2+cu121
Is debug build: False
CUDA used to build PyTorch: 12.1
ROCM used to build PyTorch: N/A
OS: Debian GNU/Linux 11 (bullseye) (x…
-
I have a Pytorch model, so i created a docker environment with ubuntu installed Golang , and try to run the simple code example:
`package main
import (
"fmt"
"github.com/sugarme/gotch"
"…
-
### 🐛 Describe the bug
It appears that creating a deepcopy of a [DualGraphModule](https://github.com/pytorch/vision/blob/838ad6ccf6ef7032485671a54c1570876e366dc5/torchvision/models/feature_extraction…
-
```
_ TestOnnxModelOutputConsistency_opset18CPU.test_output_match_nn_functional_embedding_bag_cpu_float16 (inputs='(tensor([[ 3.4629, 4.7383, -5.2305, -7.2500, 3.7188],\n [ 0.5361, -6.0039, …