| 135 | |
| 136 | template <typename T> |
| 137 | void Mat<T>::write(T* dst) const { |
| 138 | megdnn_assert(is_continuous()); |
| 139 | cuda_check(cudaMemcpy( |
| 140 | dst, m_data.get(), sizeof(T) * this->total_nr_elem(), |
| 141 | cudaMemcpyDeviceToHost)); |
| 142 | } |
| 143 | |
| 144 | template class Mat<uchar>; |
| 145 | template class Mat<float>; |