File tree Expand file tree Collapse file tree 2 files changed +2
-0
lines changed
Expand file tree Collapse file tree 2 files changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -111,6 +111,7 @@ def thread_police_duuu_daaa_duuu_daaa():
111111 sys .version_info >= (3 , 9 )
112112 and isinstance (thread , _ExecutorManagerThread )
113113 or "ThreadPoolExecutor-" in thread .name
114+ or thread .name == "InductorSubproc" # torch.compile
114115 ):
115116 # probably `torch.compile`, can't narrow it down further
116117 continue
Original file line number Diff line number Diff line change @@ -170,6 +170,7 @@ def thread_police_duuu_daaa_duuu_daaa():
170170 sys .version_info >= (3 , 9 )
171171 and isinstance (thread , _ExecutorManagerThread )
172172 or "ThreadPoolExecutor-" in thread .name
173+ or thread .name == "InductorSubproc" # torch.compile
173174 ):
174175 # probably `torch.compile`, can't narrow it down further
175176 continue
You can’t perform that action at this time.
0 commit comments