| 238 | }; |
| 239 | |
| 240 | void copyTensor(ft::Tensor& dst, ft::Tensor& src) |
| 241 | { |
| 242 | FT_CHECK_WITH_INFO( |
| 243 | src.sizeBytes() == dst.sizeBytes(), |
| 244 | ft::fmtstr("src and dst has different size (%ld != %ld)", src.sizeBytes(), dst.sizeBytes())); |
| 245 | ft::cudaAutoCpy(dst.getPtr<char>(), src.getPtr<char>(), src.sizeBytes(), stream); |
| 246 | cudaStreamSynchronize(stream); |
| 247 | } |
| 248 | |
| 249 | }; |
| 250 |