(device_type: str | None = None)
| 310 | |
| 311 | |
| 312 | def device_synchronize(device_type: str | None = None): |
| 313 | if device_type is None: |
| 314 | device_type = get_device() |
| 315 | device_mod = getattr(torch, device_type, torch.cuda) |
| 316 | device_mod.synchronize() |
| 317 | |
| 318 | |
| 319 | def enable_full_determinism(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…