| 177 | } |
| 178 | |
| 179 | void IndexingMultiAxisVecImpl::exec( |
| 180 | _megdnn_tensor_in src, const IndexDesc& index, _megdnn_tensor_out dst, |
| 181 | _megdnn_workspace workspace) { |
| 182 | auto info = check_exec(src.layout, index, dst.layout, workspace.size); |
| 183 | info.error_tracker = m_error_tracker; |
| 184 | info.error_info = async_error_info(handle()); |
| 185 | ExecImpl<indexing_multi_axis_vec_kdef::OprFwd>{ |
| 186 | src, dst, index, workspace, info, hip_stream(handle())}(); |
| 187 | } |
| 188 | |
| 189 | size_t IndexingSetMultiAxisVecImpl::get_workspace_in_bytes(size_t value_idx_size) { |
| 190 | return value_idx_size * sizeof(int); |
nothing calls this directly
no test coverage detected