-
### 🐛 Describe the bug
```
import torch
import time
import torch.nn as nn
import torch.utils.data as Data
torch.manual_seed(10)
class ATT(nn.Module):
def __init__(self, ):
super…
-
### 🐛 Describe the bug
This is only a minor issue, but Tensors created with `torch.normal` or `torch.from_numpy` creates a tensor on CPU and not on the `default_device`. At least if only float inpu…
-
### Your current environment
The output of `python collect_env.py`
```text
PyTorch version: 2.6.0.dev20240918+rocm6.2
Is debug build: False
CUDA used to build PyTorch: N/A
ROCM used to bui…
-
### 🐛 Describe the bug
here a 'simple' way to reproduce the issue
```
import torch
@torch.jit.script
def snake(x, alpha):
x = x + alpha + 1e-9
return x
class Snake1d(torch.nn.Modul…
-
### 🐛 Describe the bug
Hi,
I recently noticed a weird behavior in Pytorch. The import order of torch and numpy can have a significant impact on the dataloader performance. In short (and see the c…
-
# Expected Behavior
I wanted to load two models at once to swap between them based on speed or quality.
Here's the related code:
```
from llama_cpp import Llama
total_threads = 16
n_threads …
-
```
$ python complex_tensor.py
ComplexTensor(real=tensor([[-5]]), imag=tensor([[10]]))
$
/home/albandes/local/pytorch/3.11_debug_source/torch/_inductor/compile_worker/__main__.py:45: ResourceWar…
-
### 🐛 Describe the bug
Output shape in the ONNX file exported for the following dummy model using with ceil_mode AvgPooling2d is wrong size.
This issue was reported in #71549 and fixed in pytorc…
-
### Your current environment
```text
The output of `python collect_env.py`
Collecting environment information...
PyTorch version: 2.3.1+cu121
Is debug build: False
CUDA used to build PyTorch: 12…
-
### Your current environment
```
PyTorch version: 2.3.0+cu121
Is debug build: False
CUDA used to build PyTorch: 12.1
ROCM used to build PyTorch: N/A
OS: Ubuntu 22.04.3 LTS (x86_64)
GCC versio…
ebi64 updated
1 month ago