Returns true if this device is part of the GPUTracer logging.
(self, device_name)
| 626 | return tensor |
| 627 | |
| 628 | def _is_gputrace_device(self, device_name): |
| 629 | """Returns true if this device is part of the GPUTracer logging.""" |
| 630 | return '/stream:' in device_name or '/memcpy' in device_name or \ |
| 631 | '/stream#' in device_name or 'stream:MemcpyHtoD' in device_name or \ |
| 632 | 'stream:MemcpyDtoH' in device_name |
| 633 | |
| 634 | def _is_hosttrace_dvice(self, device_name): |
| 635 | """ Retruns true if this device is part of the HostTracer logging. """ |
no outgoing calls
no test coverage detected