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

Revert "[CUDA12] set_device change (#94864)"

Browse Source 
This reverts commit c18be2b2ec00133abe28efcdd0462e50ddd45a1a.

Reverted https://github.com/pytorch/pytorch/pull/94864 on behalf of https://github.com/ezyang due to avoid affecting cuda 11
This commit is contained in:
PyTorch MergeBot
2023-04-05 14:53:00 +00:00
parent 981f9f0408
commit 279ca5f9db
25 changed files with 60 additions and 276 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/csrc/jit/python/python_sugared_value.cpp
View File

@ -229,7 +229,6 @@ std::shared_ptr<SugaredValue> CUDAPythonModuleValue::attr(
"default_stream",
"current_device",
"_exchange_device",
"_maybe_exchange_device",
"set_device",
"device_index",
"device_count",