Skip to content

Commit f076eaf

Browse files
authored
Removing the hack to fix the avg_pool2d backward (#619)
1 parent 7a7545e commit f076eaf

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

functorch/_src/compilers.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,6 @@ def ts_compile(fx_g: fx.GraphModule, _) -> Callable:
4040
args = list(node.args)
4141
args[1] = [1]
4242
node.args = tuple(args)
43-
elif node.target == torch.ops.aten.avg_pool2d_backward:
44-
# Handle empty strides
45-
if node.args[3] == []:
46-
args = list(node.args)
47-
args[3] = [1, 1]
48-
node.args = tuple(args)
4943

5044
for node in fx_g.graph.nodes:
5145
new_kwargs = {}

0 commit comments

Comments
 (0)