frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle size/etc accessors in FakeTensor, support accessing symbolic types from toInt/etc in IValue (#124760)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/122772

Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/124760
Approved by: https://github.com/albanD, https://github.com/eellison
This commit is contained in:
Edward Z. Yang
2024-04-24 16:29:37 -04:00
committed by PyTorch MergeBot
parent 9bd6e93a04
commit 0d58aeb73a
10 changed files with 37 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/test_proxy_tensor.py
View File

@ -1905,9 +1905,7 @@ symbolic_tensor_failures = {
xfail('nn.functional.binary_cross_entropy', ''), # aten.new_empty.default - couldn't find symbolic meta function/decom...
xfail('nn.functional.cross_entropy', ''), # aten.size.default - couldn't find symbolic meta function/decomposition
xfail('nn.functional.ctc_loss'), # aten._ctc_loss.Tensor - couldn't find symbolic meta function/decomposition
xfail('nn.functional.fractional_max_pool3d', ''), # argument 'size' must be tuple of ints, but found element of t...
xfail('quantile', ''), # Could not run 'aten::equal' with arguments from the 'Meta' backend.
xfail('resize_as_', ''), # aten.clone.default - couldn't find symbolic meta function/decomposition
xfail('unique_consecutive', ''), # aten.unique_consecutive.default - couldn't find symbolic meta function/decomposition
xfail('unique', ''), # aten._unique2.default - couldn't find symbolic meta function/decomposition
@ -1958,29 +1956,17 @@ out_symbolic_tensor_failures = {
xfail('angle', ''),
xfail('argmax', ''),
xfail('argmin', ''),
xfail('bmm', ''),
xfail('fft.fft2', ''),
xfail('fft.fftn', ''),
xfail('fft.ifft2', ''),
xfail('fft.ifftn', ''),
xfail('gather', ''),
xfail('linalg.cholesky', ''),
xfail('linalg.cholesky_ex', ''),
xfail('linalg.det', ''),
xfail('linalg.det', 'singular'),
xfail('linalg.inv', ''),
xfail('linalg.inv_ex', ''),
xfail('linalg.pinv', ''),
xfail('linalg.pinv', 'hermitian'),
xfail('linalg.svdvals', ''),
xfail('lu', ''),
xfail('max', 'reduction_with_dim'),
xfail('min', 'reduction_with_dim'),
xfail('nn.functional.avg_pool2d', ''),
xfail('scatter_add', ''),
xfail('scatter', ''),
xfail('take_along_dim', ''),
xfail('topk', ''),
xfail('triangular_solve', ''),
xfail('view_copy', ''),