Add general block sparse tensor support#135
Merged
ajaypanyala merged 28 commits intomainfrom Jun 12, 2025
Merged
Conversation
…ces in TiledIndexSpaces
- adds BlockSparseTensor implementation - adds new tensor constructors for BlockSparseTensor - adds test cases for new tensor type - adds CCSD_V4_BlockSparse implementation using new tensor type - updates map op execution to allow exact copy on the same tensor - adds block overlap check to mapop
* adds `from_distributed_tensor` and `to_distributed_tensor` functions * adds test case to `Test_LocalTensor` * adds documentation for the new functionality
- Move block sparse tensor construction to default tensor construction - Remove BlockSparseTensor implementation - Move BlockSparseInfo struct into a new file - Move non zero check function construction into BlockSparseInfo struct - Update test cases with the new interface - Update tensor construction documentation
- adds a new boolean parameter for constructing map op specifically for exact copy - adds new logic for copying same shaped blocks in block sparse tensors
- Adds new constructors that uses TiledIndexLabels, TiledIndexSpaces, and strings representing the sub-space names for representing allowed blocks - Updates unit tests and Test_CCSD_V4_BlockSparse with the new constructors - Adds documentation for the new constructors
|
|
ajaypanyala
force-pushed
the
main
branch
2 times, most recently
from
f376056 to
ffac6f5
Compare
ajaypanyala
force-pushed
the
main
branch
2 times, most recently
from
a85dea1 to
3b542d3
Compare
This file contains hidden or 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
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
4 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
Adds support for general block sparse tensor
is_non_zerocheck for constructing tensorsTensorInfostruct for describing block sparse tensorsTensorInfobased block sparse tensor construction