Get the magic method try reverse protocol correct (#88030)
Signed-off-by: Edward Z. Yang <[email protected]>
cc @mlazos @soumith @voznesenskym @yanboliang @penguinwu @anijain2305 @EikanWang @jgong5 @Guobing-Chen @chunyuan-w @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88030
Approved by: https://github.com/anjali411, https://github.com/albanD
diff --git a/test/test_proxy_tensor.py b/test/test_proxy_tensor.py
index 4f4265b..a5de034 100644
--- a/test/test_proxy_tensor.py
+++ b/test/test_proxy_tensor.py
@@ -975,6 +975,11 @@
return x.shape
self._test_dynamic(f, [(5, 3)], [[(4, 6)]])
+ def test_rmethod(self):
+ def f(x):
+ return x.size(0) + x
+ self._test_dynamic(f, [(5,)], [[(4,)], [(12,)]])
+
def test_mega_guard(self):
def f(a, b):
assert a.shape[0] == b.shape[0] * 2
