Skip to content

Commit 01f8654

Browse files
PaliCfacebook-github-bot
authored andcommitted
Remove torch._running_with_deploy() from fbcode and add RuntimeError handling for load_library (#3207)
Summary: Pull Request resolved: #3207 As per https://fb.workplace.com/groups/pytorch.dev/permalink/1828123831099422 we can now safely remove “torch.is_deploy_running”. This commit does this! Additionally we are handling RuntimeErrors while using load_library in torchrec Reviewed By: nipung90 Differential Revision: D78525065 fbshipit-source-id: 85e76dc6ced60c5b286153c2156def2c21f7d302
1 parent 83f8114 commit 01f8654

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

torchrec/distributed/train_pipeline/tracing.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
from typing import Any, Dict, List, Optional, Set, Tuple, Union
1313

1414
import torch
15-
1615
from torch.distributed._composable.fsdp.fully_shard import FSDPModule as FSDP2
1716

1817
from torch.distributed.fsdp import FullyShardedDataParallel as FSDP

0 commit comments

Comments
 (0)