8000 Make fx.node.map_arg() and .map_aggregate() generic by rec · Pull Request #146248 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

Make fx.node.map_arg() and .map_aggregate() generic #146248

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 27 commits into from
Closed

Make fx.node.map_arg() and .map_aggregate() generic #146248

Changes from 1 commit
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
ba4ee7f
Update
rec Feb 1, 2025
afd81cd
Update
rec Feb 1, 2025
7674025
Update
rec Feb 2, 2025
f15da1f
Update
rec Feb 2, 2025
ae10f6c
Update
rec Feb 2, 2025
1230b89
Update
rec Feb 3, 2025
c63c17d
Update
rec Feb 3, 2025
9dc40b5
Update
rec Feb 4, 2025
5918536
Update
rec Feb 4, 2025
e4aa732
Update
rec Feb 5, 2025
441c089
Update
rec Feb 5, 2025
dd5dc46
Update
rec Feb 6, 2025
b52a69d
Update
rec Feb 7, 2025
1160968
Update
rec Feb 7, 2025
f248065
Update
rec Feb 10, 2025
bbad690
Update
rec Feb 11, 2025
c62e68b
Update
rec Feb 11, 2025
e421e13
Update
rec Feb 12, 2025
d13e435
Update
rec Feb 12, 2025
41a2dec
Update
rec Feb 12, 2025
cfc5f8b
Update
rec Feb 12, 2025
8a2b953
Update
rec Feb 12, 2025
dab69d3
Update
rec Feb 13, 2025
e1bf862
Update
rec Feb 13, 2025
2d5de70
Update
rec Feb 13, 2025
1e18ebd
Update
rec Feb 14, 2025
d0d7327
Update
rec Feb 14, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update 
[ghstack-poisoned]
  • Loading branch information
@rec
rec committed Feb 13, 2025
commit e1bf862981f69afca4a7f8ce29accffd3b1c9f9c
12 changes: 9 additions & 3 deletions torch/_tensor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from collections import OrderedDict
from copy import deepcopy
from numbers import Number
from typing import Any, Optional, Union
from typing import Any, Callable, cast, Optional, Union

import torch
import torch._C as _C
Expand Down Expand Up @@ -1104,8 +1104,14 @@ def __rdiv__(self, other):
__rtruediv__ = __rdiv__
__itruediv__ = _C.TensorBase.__idiv__

__pow__ = _handle_torch_function_and_wrap_type_error_to_not_implemented(
_C.TensorBase.pow
__pow__ = cast(
Callable[
["torch._C.TensorBase", Union["Tensor", int, float, bool, complex]],
"Tensor",
],
_handle_torch_function_and_wrap_type_error_to_not_implemented(
_C.TensorBase.pow
),
)
__ipow__ = _handle_torch_function_and_wrap_type_error_to_not_implemented(
_C.TensorBase.pow_
Expand Down
Loading
0