mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Fixes #94841 This fixes the error messages in the following files, the same as those referenced in the linked issue. I was not able to find any additional examples, but am happy to add commits for any that I may have missed! ``` aten/src/ATen/native/Blas.cpp: "size mismatch, got ", self.size(0), ", ", mat.size(0), "x", mat.size(1), ",", vec.size(0)); torch/_decomp/decompositions.py: lambda: f"size mismatch, got {self.size(0)}x{self.size(1)},{vec.size(0)}", ``` Example output for `Blas.cpp` before: ``` size mismatch, got 3, 3x4,1 ``` The new error messages have the following format: ``` aten/src/ATen/native/Blas.cpp: "size mismatch, got bias (", self.size(0), "), matrix (", mat.size(0), "x", mat.size(1), "), vector (", vec.size(0), ")"); torch/_decomp/decompositions.py: lambda: f"size mismatch, got matrix ({self.size(0)}x{self.size(1)}), vector ({vec.size(0)})", ``` Example output for `Blas.cpp` after: ``` size mismatch, got bias (3), matrix (3x4), vector (1) ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/96863 Approved by: https://github.com/albanD |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| decompositions.py | ||
| decompositions_for_jvp.py | ||