| 76 | CUMAT_DISALLOW_COPY_AND_ASSIGN(CublasApi); |
| 77 | private: |
| 78 | CublasApi(Context& ctx) |
| 79 | : handle_(nullptr) |
| 80 | , stream_(ctx.stream()) |
| 81 | { |
| 82 | CUBLAS_SAFE_CALL(cublasCreate_v2(&handle_)); |
| 83 | CUBLAS_SAFE_CALL(cublasSetPointerMode(handle_, CUBLAS_POINTER_MODE_HOST)); |
| 84 | CUBLAS_SAFE_CALL(cublasSetStream(handle_, stream_)); |
| 85 | } |
| 86 | public: |
| 87 | ~CublasApi() |
| 88 | { |