Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add general block sparse tensor support #135

Open
wants to merge 27 commits into
base: main
Choose a base branch
from
Open

Conversation

erdalmutlu
Copy link
Contributor

Description
Adds support for general block sparse tensor

  • Support for lambda based block-wise is_non_zero check for constructing tensors
  • Adds new TensorInfo struct for describing block sparse tensors
  • Support for TensorInfo based block sparse tensor construction

erdalmutlu and others added 27 commits November 8, 2024 13:59
- 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
@erdalmutlu erdalmutlu self-assigned this Jan 31, 2025
@CLAassistant
Copy link

CLAassistant commented Jan 31, 2025

CLA assistant check
All committers have signed the CLA.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants