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

Update base for Update on "[WIP] Add scalar support for eager-through-torch.compile"

Browse Source 
cc voznesenskym penguinwu jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler amjames desertfire chauhang

[ghstack-poisoned]
This commit is contained in:
Wang, Eikan
2024-04-15 15:55:59 +00:00
parent aa7a947cd2
commit ab3aa0d6b2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/library.py
View File

@ -140,7 +140,7 @@ class Library:
>>> my_lib = Library("aten", "IMPL")
>>> def div_cpu(self, other):
>>> return self * (1 / other)
>>> my_lib._impl_with_aoti_compile("div", "Tensor", "CPU", div_cpu)
>>> my_lib._impl_with_aoti_compile("aten::div", "Tensor", "CPU", div_cpu)
'''
impl_fn_name = "impl_with_aoti_compile"