-
Notifications
You must be signed in to change notification settings - Fork 22
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implement gesv_batch via gesv call (#1877)
* First working version with transpose and C contig * Second working version with moveaxis, transpose and F contig * Add more shape checks * Pass sycl::queue by reference for gesv/gesv_batch * Update _batched_solve implementation * Remove old impl in _batched_solve * Use py::gil_scoped_release before gesv call * Move gesv_batch to gesv_batch.cpp * Improve gesv_batch with independent linear streams * Extend checks for gesv/gesv_batch * Add common_gesv_checks * Release GIL in gesv_batch_impl * Remove host_task_events from gesv * Use check_zeros_shape in gesv and gesv_batch * Add additional checks for gesv_impl * Move alloc_scratchpad to common_helpers.hpp * Use helper::alloc_scratchpad in gesv_batch_impl * Remove current_scratch_gesv check * Remove lda, ldb pass to gesv_batch_impl, gesv_impl * Use const and constexpr in gesv/gesv_batch * Use dpnp.reshape in _batched_solve * Implement alloc_ipiv in common_helpers.hpp * Add gesv_common_utils.hpp * Implement handle_lapack_exc function * Use try/catch for scratchpad/ipiv allocation * Update alloc_scratchpad/alloc_ipiv * gesv_scratchpad_size can be 0 * Implement help functions alloc_ipiv/alloc_scratchpad * Reuse alloc_scratchpad/ipiv in batch versions 1e5ba88
- Loading branch information
1 parent
c2916b8
commit 1021e2a
Showing
515 changed files
with
1,980 additions
and
1,611 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,4 +1,4 @@ | ||
# Sphinx build info version 1 | ||
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done. | ||
config: 82899eb693d56f85faa6f668528177f3 | ||
config: bcf56222450b83e69ebbdf8346235d8b | ||
tags: 645f666f9bcd5a90fca523b33c5a78b7 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.