-
### 🐛 Describe the bug
torch.argmax is slow on CPU relative to the numpy implementation. I would expect these two to be approximately equally fast. Here is a small script to reproduce this:
```pyt…
-
when i run "rdagent fin_model", it works well on my cpu to train a GRU. How to use gpu device such as "cuda:0" to run this demo?
Some outputs of my terminal when running this script are as follows:
[…
-
### 🐛 Describe the bug
```
python -u -m pytest -s -v test/inductor/test_cpu_select_algorithm.py -k test_int8_woq_mm_batch_size_32_in_features_128_out_features_65_cpu_bfloat16
python -u -m pytes…
-
***tl;dr PyTorch will stop publishing Anaconda packages that depend on Anaconda’s default packages due to the high maintenance costs for conda builds which are not justifiable with the ROI we observe …
-
### Your current environment
The output of `python collect_env.py`
```text
Collecting environment information...
WARNING 10-25 14:14:20 _custom_ops.py:19] Failed to import from vllm._C with Im…
-
### Problem Description
Hi,
We are working on MLPerf SDXL Inference benchmark, and we are trying to use lower level API like MIGraphX to speed up the inference. We are running the barebone implem…
-
### 🐛 Describe the bug
datasets.PhotoTour(root="./data/phototour", name="notredame", train=False, download=True)
when i run this code, it comes up with this bug.
Traceback (most recent call last):
…
-
### Your current environment
```text
PyTorch version: N/A
Is debug build: N/A
CUDA used to build PyTorch: N/A
ROCM used to build PyTorch: N/A
OS: Fedora Linux 38 (Thirty Eight) (riscv64)
GCC …
-
Follow the guide here: https://github.com/intel/ai-reference-models/tree/main/models_v2/pytorch/llama/training/cpu, faced several issues:
1. https://github.com/intel/ai-reference-models/blob/main/m…
-
### 🐛 Describe the bug
FAILED: caffe2/CMakeFiles/torch_hip.dir/__/aten/src/ATen/native/hip/torch_hip_generated_CumprodKernel.hip.o /home/hmsjwzb/work/pytorch/build/caffe2/CMakeFiles/torch_hip.dir/__/…