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

use DTRACE_ENV_VAR as the trace logs directory of set (#146412)

Browse Source 
```
(/home/bobren/local/a/pytorch-env) [7:47] devgpu035:/home/bobren/local/a/pytorch TORCH_DTRACE=/tmp/bb python r1.py
```
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146412
Approved by: https://github.com/angelayi
ghstack dependencies: #145848
This commit is contained in:
bobrenjc93
2025-02-04 07:57:12 -08:00
committed by PyTorch MergeBot
parent 7f65a20884
commit d23e4f8109
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/_logging/_internal.py
View File

@ -1001,8 +1001,9 @@ def _init_logs(log_file_name=None):
# configuration
trace_dir_name = os.environ.get(TRACE_ENV_VAR, None)
if os.environ.get(DTRACE_ENV_VAR, None):
if dtrace_dir_name := os.environ.get(DTRACE_ENV_VAR, None):
GET_DTRACE_STRUCTURED = True
trace_dir_name = dtrace_dir_name
# This handler may remove itself if trace_dir_name is None and we are not
# actually in an FB environment. This allows us to defer actually