()
| 1412 | ################################################################################ |
| 1413 | |
| 1414 | def manager_path(): |
| 1415 | if _running_with_deploy() or platform.system() == 'Windows': |
| 1416 | return b"" |
| 1417 | path = get_file_path('torch', 'bin', 'torch_shm_manager') |
| 1418 | prepare_multiprocessing_environment(get_file_path('torch')) |
| 1419 | if not os.path.exists(path): |
| 1420 | raise RuntimeError("Unable to find torch_shm_manager at " + path) |
| 1421 | return path.encode('utf-8') |
| 1422 | |
| 1423 | from torch.amp import autocast |
| 1424 |
no test coverage detected
searching dependent graphs…