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

Revert "refactor lazy init to device-agnostic (#118846)"

Browse Source 
This reverts commit 520771d7b35034c96c5b4604ecf8960e6aab856f.

Reverted https://github.com/pytorch/pytorch/pull/118846 on behalf of https://github.com/atalman due to Failing, tests https://github.com/pytorch/torchdistx/blob/main/src/python/torchdistx/_C/fake.cc#L11  ([comment](https://github.com/pytorch/pytorch/pull/118846#issuecomment-1927651305))
This commit is contained in:
PyTorch MergeBot
2024-02-05 18:06:30 +00:00
parent 124a54ef16
commit ab613a4019
16 changed files with 116 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torchgen/api/python.py
View File

@ -1443,7 +1443,7 @@ const auto options = TensorOptions()
.layout({arg_parser_outputs['layout'].expr})
.requires_grad({arg_parser_outputs['requires_grad'].expr})
.pinned_memory({arg_parser_outputs['pin_memory'].expr});
torch::utils::maybe_initialize_device(options);
torch::utils::maybe_initialize_cuda(options);
"""
)
lambda_args_exprs["options"] = "options"