| 272 | } |
| 273 | |
| 274 | WorkspaceBundle SVDForwardImpl::get_workspace_bundle( |
| 275 | size_t m, size_t n, size_t dtype_size, void* raw_ptr) { |
| 276 | // Scratchpads for u and v. |
| 277 | size_t dim0 = std::max(m, n); |
| 278 | size_t dim1 = std::min(m, n); |
| 279 | return {raw_ptr, |
| 280 | {m * n * dtype_size, dim0 * dim0 * dtype_size, dim1 * dim1 * dtype_size}, |
| 281 | handle()->alignment_requirement()}; |
| 282 | } |
| 283 | |
| 284 | template <typename T> |
| 285 | void exec_internal( |
nothing calls this directly
no test coverage detected