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

Remove remaining global set_default_dtype calls from tests (#107246)

Browse Source 
Fixes #68972

Pull Request resolved: https://github.com/pytorch/pytorch/pull/107246
Approved by: https://github.com/ezyang
This commit is contained in:
Kurt Mohler
2023-08-24 16:10:48 +00:00
committed by PyTorch MergeBot
parent 918df10198
commit aa8ea1d787
17 changed files with 899 additions and 873 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_ops_gradients.py
View File

@ -10,9 +10,6 @@ from torch.testing._internal.custom_op_db import custom_op_db
from torch.testing._internal.common_device_type import \
(instantiate_device_type_tests, ops, OpDTypes)
# TODO: fixme https://github.com/pytorch/pytorch/issues/68972
torch.set_default_dtype(torch.float32)
# gradcheck requires double precision
_gradcheck_ops = partial(ops, dtypes=OpDTypes.supported,
allowed_dtypes=[torch.double, torch.cdouble])