blksprs 1.9.1__tar.gz → 1.9.2__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (30) hide show
  1. {blksprs-1.9.1 → blksprs-1.9.2}/PKG-INFO +4 -2
  2. {blksprs-1.9.1 → blksprs-1.9.2}/README.md +2 -1
  3. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/processing.py +4 -4
  4. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/PKG-INFO +4 -2
  5. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/requires.txt +1 -0
  6. {blksprs-1.9.1 → blksprs-1.9.2}/pyproject.toml +3 -2
  7. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/__init__.py +0 -0
  8. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/layouting/distribution_layout.py +0 -0
  9. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/layouting/sparsity_layout.py +0 -0
  10. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/conversion.py +0 -0
  11. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/distribution.py +0 -0
  12. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/experimental/distribution_mdi.py +0 -0
  13. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/flow.py +0 -0
  14. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/matmul.py +0 -0
  15. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/misc/broadcast_ops.py +0 -0
  16. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/misc/exp.py +0 -0
  17. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/misc/row_wise.py +0 -0
  18. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/partitioning.py +0 -0
  19. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/repeat.py +0 -0
  20. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/softmax.py +0 -0
  21. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/transpose.py +0 -0
  22. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/benchmarking.py +0 -0
  23. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/blksprs_tensor.py +0 -0
  24. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/layout_utils.py +0 -0
  25. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/tools.py +0 -0
  26. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/validation.py +0 -0
  27. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/SOURCES.txt +0 -0
  28. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/dependency_links.txt +0 -0
  29. {blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/top_level.txt +0 -0
  30. {blksprs-1.9.1 → blksprs-1.9.2}/setup.cfg +0 -0
{blksprs-1.9.1 → blksprs-1.9.2}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: blksprs
3
- Version: 1.9.1
3
+ Version: 1.9.2
4
4
  Summary: A lightweight library for operations on blocksparse matrices in PyTorch.
5
5
  Author-email: Felix Schön <schoen@kr.tuwien.ac.at>
6
6
  Project-URL: Homepage, https://github.com/FelixSchoen/blksprs
@@ -8,6 +8,7 @@ Project-URL: Bugtracker, https://github.com/FelixSchoen/blksprs/issues
8
8
  Requires-Python: >=3.11
9
9
  Description-Content-Type: text/markdown
10
10
  Requires-Dist: torch
11
+ Requires-Dist: numpy
11
12
  Provides-Extra: test
12
13
  Requires-Dist: pytest; extra == "test"
13
14
  Requires-Dist: pytest-xdist; extra == "test"
@@ -83,7 +84,8 @@ We recommend installing blksprs from [PyPI](https://pypi.org/project/blksprs/) u
83
84
 
84
85
  ### Dependencies
85
86
 
86
- - [PyTorch](https://pytorch.org/) (built with v2.5.0)
87
+ - [PyTorch](https://pytorch.org/) (built with v2.5.1)
88
+ - _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.2.0)_
87
89
  - _[Triton](https://github.com/triton-lang/triton) (included with PyTorch)_
88
90
 
89
91
  ## Changelog
{blksprs-1.9.1 → blksprs-1.9.2}/README.md RENAMED
@@ -64,7 +64,8 @@ We recommend installing blksprs from [PyPI](https://pypi.org/project/blksprs/) u
64
64
 
65
65
  ### Dependencies
66
66
 
67
- - [PyTorch](https://pytorch.org/) (built with v2.5.0)
67
+ - [PyTorch](https://pytorch.org/) (built with v2.5.1)
68
+ - _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.2.0)_
68
69
  - _[Triton](https://github.com/triton-lang/triton) (included with PyTorch)_
69
70
 
70
71
  ## Changelog
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/processing.py RENAMED
@@ -66,9 +66,9 @@ def apply_function_applicable_row_wise(x: BlksprsTensor, sparsity_layout: Tensor
66
66
 
67
67
 
68
68
  def _pack_layout(sparsity_layout: Tensor) -> BlksprsTensor:
69
- sparsity_layout_resized = sparsity_layout.resize(1, sparsity_layout.size(0) * sparsity_layout.size(1),
70
- sparsity_layout.size(2))
71
- non_zero_rows = torch.any(sparsity_layout_resized, dim=-1)
72
- sparsity_layout_filtered = sparsity_layout_resized[non_zero_rows].unsqueeze(0)
69
+ sparsity_layout_reshaped = sparsity_layout.reshape(1, sparsity_layout.size(0) * sparsity_layout.size(1),
70
+ sparsity_layout.size(2))
71
+ non_zero_rows = torch.any(sparsity_layout_reshaped, dim=-1)
72
+ sparsity_layout_filtered = sparsity_layout_reshaped[non_zero_rows].unsqueeze(0)
73
73
 
74
74
  return sparsity_layout_filtered
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: blksprs
3
- Version: 1.9.1
3
+ Version: 1.9.2
4
4
  Summary: A lightweight library for operations on blocksparse matrices in PyTorch.
5
5
  Author-email: Felix Schön <schoen@kr.tuwien.ac.at>
6
6
  Project-URL: Homepage, https://github.com/FelixSchoen/blksprs
@@ -8,6 +8,7 @@ Project-URL: Bugtracker, https://github.com/FelixSchoen/blksprs/issues
8
8
  Requires-Python: >=3.11
9
9
  Description-Content-Type: text/markdown
10
10
  Requires-Dist: torch
11
+ Requires-Dist: numpy
11
12
  Provides-Extra: test
12
13
  Requires-Dist: pytest; extra == "test"
13
14
  Requires-Dist: pytest-xdist; extra == "test"
@@ -83,7 +84,8 @@ We recommend installing blksprs from [PyPI](https://pypi.org/project/blksprs/) u
83
84
 
84
85
  ### Dependencies
85
86
 
86
- - [PyTorch](https://pytorch.org/) (built with v2.5.0)
87
+ - [PyTorch](https://pytorch.org/) (built with v2.5.1)
88
+ - _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.2.0)_
87
89
  - _[Triton](https://github.com/triton-lang/triton) (included with PyTorch)_
88
90
 
89
91
  ## Changelog
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs.egg-info/requires.txt RENAMED
@@ -1,4 +1,5 @@
1
1
  torch
2
+ numpy
2
3
 
3
4
  [build]
4
5
  build
{blksprs-1.9.1 → blksprs-1.9.2}/pyproject.toml RENAMED
@@ -1,13 +1,14 @@
1
1
  [project]
2
2
  name = "blksprs"
3
- version = "1.9.1"
3
+ version = "1.9.2"
4
4
  authors = [{ name = "Felix Schön", email = "schoen@kr.tuwien.ac.at" }]
5
5
  description = "A lightweight library for operations on blocksparse matrices in PyTorch."
6
6
  readme = "README.md"
7
7
  requires-python = ">=3.11"
8
8
  license = { file = "LICENSE.md" }
9
9
  dependencies = [
10
- "torch"
10
+ "torch",
11
+ "numpy"
11
12
  ]
12
13
 
13
14
  [project.urls]
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/__init__.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/flow.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/matmul.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/misc/exp.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/repeat.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/ops/softmax.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/blksprs/utils/tools.py RENAMED
File without changes
{blksprs-1.9.1 → blksprs-1.9.2}/setup.cfg RENAMED
File without changes