Record the stream that is using this tensor. Note ---- This API is more for testing. Users should call ``record_stream`` on torch.Tensor or dgl.graph directly. Parameters ---------- stream : DGLStreamHandle
(self, stream)
| 368 | check_call(_LIB.DGLArrayUnpinData(self.handle)) |
| 369 | |
| 370 | def record_stream(self, stream): |
| 371 | """Record the stream that is using this tensor. |
| 372 | |
| 373 | Note |
| 374 | ---- |
| 375 | This API is more for testing. Users should call ``record_stream`` |
| 376 | on torch.Tensor or dgl.graph directly. |
| 377 | |
| 378 | Parameters |
| 379 | ---------- |
| 380 | stream : DGLStreamHandle |
| 381 | """ |
| 382 | check_call(_LIB.DGLArrayRecordStream(self.handle, stream)) |
| 383 | |
| 384 | |
| 385 | def free_extension_handle(handle, type_code): |
no test coverage detected