Skip to content

Commit 4f7b02b

Browse files
authored
fix jit script error on adaptive_avg_pool2d (#99)
1 parent 2324119 commit 4f7b02b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intel_pytorch_extension_py/ops/pooling.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,6 @@ def max_pool2d(input, kernel_size: Vector, stride: Vector, padding: Vector, dila
4444
stride = kernel_size
4545
return torch.ops.torch_ipex.max_pool2d(input, _pair(kernel_size), _pair(stride), _pair(padding), _pair(dilation), ceil_mode)
4646

47-
F.adaptive_avg_pool2d = adaptive_avg_pool2d
47+
torch.adaptive_avg_pool2d = adaptive_avg_pool2d
4848
torch.max_pool2d = max_pool2d
4949
torch.max_pool3d = max_pool3d

0 commit comments

Comments
 (0)