| 36 | } |
| 37 | |
| 38 | ~DeviceStream() { |
| 39 | CHECK_CUDA( cudaSetDevice(this->gpu_id) ); |
| 40 | if (this->stream != NULL) { |
| 41 | CHECK_CUDA( cudaStreamDestroy(this->stream) ); |
| 42 | this->stream = NULL; |
| 43 | } |
| 44 | // std::cout << "DeviceStream destructor called!" << std::endl; |
| 45 | } |
| 46 | }; |
| 47 | |
| 48 | // cublas handle wrapper |
nothing calls this directly
no outgoing calls
no test coverage detected