frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add shim fallback for narrow (#156496)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156496
Approved by: https://github.com/albanD
This commit is contained in:
Jane Xu
2025-06-20 08:35:08 -07:00
committed by PyTorch MergeBot
parent 6ed85bfe6a
commit eb331b59fe
6 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torchgen/aoti/fallback_ops.py
View File

@ -123,6 +123,7 @@ inductor_fallback_ops: dict[str, dict[str, list[str]]] = {
"aten.mul.Scalar": {},
"aten.mul.Tensor": {},
"aten.nanmedian.default": {},
"aten.narrow.default": {},
"aten.native_dropout.default": {},
"aten.nonzero.default": {},
"aten.normal_functional.default": {},