(x, stream)
| 471 | |
| 472 | |
| 473 | def _record_stream(x, stream): |
| 474 | if stream is None: |
| 475 | return x |
| 476 | if hasattr(x, "record_stream"): |
| 477 | x.record_stream(stream) |
| 478 | return x |
| 479 | elif isinstance(x, _PrefetchedGraphFeatures): |
| 480 | node_feats = recursive_apply(x.node_feats, _record_stream, stream) |
| 481 | edge_feats = recursive_apply(x.edge_feats, _record_stream, stream) |
| 482 | return _PrefetchedGraphFeatures(node_feats, edge_feats) |
| 483 | else: |
| 484 | return x |
| 485 | |
| 486 | |
| 487 | def _prefetch(batch, dataloader, stream): |
nothing calls this directly
no test coverage detected