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

[functorch] gelu skips to cpu only

Browse Source
This commit is contained in:
Sam Andow
2022-02-17 22:02:16 +00:00
committed by Jon Janzen
parent ad35082061
commit 08449c2a7c
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
functorch/test/test_ops.py
View File

@ -569,7 +569,7 @@ class TestOperators(TestCase):
xfail('index_put', ''), xfail('index_put', ''),
xfail('lu_solve'), xfail('lu_solve'),
xfail('index_copy'), xfail('index_copy'),
xfail('nn.functional.gelu'), xfail('nn.functional.gelu', device_type='cpu'),
}) })
@ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,)) @ops(functorch_lagging_op_db + additional_op_db, allowed_dtypes=(torch.float,))
@ -644,7 +644,7 @@ class TestOperators(TestCase):
xfail('quantile'), xfail('quantile'),
# RuntimeError: vmap: inplace arithmetic(self, *extra_args) # RuntimeError: vmap: inplace arithmetic(self, *extra_args)
xfail('nn.functional.gelu'), xfail('nn.functional.gelu', device_type='cpu'),
# Not implemented # Not implemented
xfail('scatter'), xfail('scatter'),
@ -725,7 +725,7 @@ class TestOperators(TestCase):
xfail('maximum'), xfail('maximum'),
xfail('linalg.householder_product'), xfail('linalg.householder_product'),
xfail('tensor_split'), xfail('tensor_split'),
xfail('nn.functional.gelu'), xfail('nn.functional.gelu', device_type='cpu'),
xfail('quantile'), xfail('quantile'),
xfail('var_mean'), xfail('var_mean'),
xfail('as_strided'), xfail('as_strided'),
@ -745,7 +745,6 @@ class TestOperators(TestCase):
xfail('nn.functional.linear'), xfail('nn.functional.linear'),
xfail('view_as_complex'), xfail('view_as_complex'),
xfail('prod'), xfail('prod'),
xfail('nn.functional.gelu'),
# Some kind of issue with unsymmetric tangent type # Some kind of issue with unsymmetric tangent type
# Runtime Error: The tangent part of the matrix A should also be symmetric. # Runtime Error: The tangent part of the matrix A should also be symmetric.
@ -820,7 +819,6 @@ class TestOperators(TestCase):
xfail('masked_select'), xfail('masked_select'),
xfail('matrix_exp'), xfail('matrix_exp'),
xfail('nanquantile'), xfail('nanquantile'),
xfail('nn.functional.gelu'),
xfail('norm', 'nuc'), xfail('norm', 'nuc'),
xfail('pinverse'), xfail('pinverse'),
xfail('prod'), xfail('prod'),