blksprs 2.1.3__py3-none-any.whl → 2.1.4__py3-none-any.whl

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.
blksprs/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  from blksprs.utils.blksprs_tensor import BlksprsTensor
2
2
 
3
- __version__ = "2.1.2"
3
+ __version__ = "2.1.4"
4
4
 
5
5
 
6
6
  class ops:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: blksprs
3
- Version: 2.1.3
3
+ Version: 2.1.4
4
4
  Summary: A lightweight library for operations on block-sparse 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
@@ -20,7 +20,8 @@ Requires-Dist: matplotlib; extra == "test"
20
20
  # blksprs
21
21
 
22
22
  [![GitHub Release](https://img.shields.io/github/v/release/FelixSchoen/blksprs?include_prereleases&label=Latest%20Release)](https://github.com/FelixSchoen/blksprs/releases)
23
- [![Python Version](https://img.shields.io/badge/Python%20Version-3.11-blue)](https://www.python.org/downloads/release/python-3119/)
23
+ [![Python 3.11](https://img.shields.io/badge/Python%20Version-3.11-blue)](https://www.python.org/downloads/release/python-3119/)
24
+ [![Python 3.12](https://img.shields.io/badge/Python%20Version-3.12-blue)](https://www.python.org/downloads/release/python-31210/)
24
25
 
25
26
  ## Overview
26
27
 
@@ -75,9 +76,7 @@ _* see the [Roadmap](#roadmap) section for more information_
75
76
 
76
77
  ## Installation
77
78
 
78
- Note that due to the dependency on [Triton](https://github.com/triton-lang/triton) this library is **only compatible
79
- with
80
- the Linux platform**.
79
+ Note that due to the dependency on [Triton](https://github.com/triton-lang/triton) this library is **only compatible with the Linux platform**.
81
80
  Keep track of this [issue](https://github.com/triton-lang/triton/issues/1640) for updates.
82
81
 
83
82
  We recommend installing blksprs from [PyPI](https://pypi.org/project/blksprs/) using pip:
@@ -86,8 +85,8 @@ We recommend installing blksprs from [PyPI](https://pypi.org/project/blksprs/) u
86
85
 
87
86
  ### Dependencies
88
87
 
89
- - [PyTorch](https://pytorch.org/) (built with v2.6)
90
- - _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.2.4)_
88
+ - [PyTorch](https://pytorch.org/) (built with v2.7.1)
89
+ - _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.3.1)_
91
90
  - _[Triton](https://github.com/triton-lang/triton) (included with PyTorch)_
92
91
 
93
92
  ## Changelog
@@ -103,7 +102,7 @@ We will continue to maintain the library and fix any issues that arise.
103
102
  Should you find any bugs please open an [issue](https://github.com/FelixSchoen/blksprs/issues).
104
103
  We also encourage [pull requests](https://github.com/FelixSchoen/blksprs/pulls).
105
104
 
106
- It might be that this changes with future projects, but as of March 2025, we are content with the current state of the
105
+ It might be that this changes with future projects, but as of June 2025, we are content with the current state of the
107
106
  library.
108
107
 
109
108
  ## Known Limitations and Issues
@@ -112,9 +111,6 @@ library.
112
111
  In order to work around this bug a manual conversion of some values is needed, (slightly) negatively impacting
113
112
  performance.
114
113
  Watch the [issue](https://github.com/triton-lang/triton/issues/6376) on Triton's issue tracker for more information.
115
- - PyTorch's `wrap_triton()` currently does not support config pruning. It thus cannot be used for some of the kernels,
116
- which could impact graph compilation.
117
- - There seem to be some issues with autocasting, forcing some operations to manually cast.
118
114
  - There will be some slight numerical differences between vanilla and blksprs operations.
119
115
  These instabilities are due to Triton and thus cannot be fixed by this library alone.
120
116
  However, for all intents and purposes, these very minor differences should not matter and can safely be ignored.
@@ -1,4 +1,4 @@
1
- blksprs/__init__.py,sha256=NRxydw4i9jg7WeDuojfEePdtdbughV9AZsEcT9yywK4,1615
1
+ blksprs/__init__.py,sha256=XERzTtkiElDeBppOO8rNrF6OktUQf_yozDiA4DUXqTY,1615
2
2
  blksprs/layouting/distribution_layout.py,sha256=ur1ty_2U-Hfj78hMWsLZvu7ZuGhzW3qGLKMc72DfTZM,5861
3
3
  blksprs/layouting/sparsity_layout.py,sha256=eXHmu2h7K5Q-YUpfOxocJoeP_5ZoQFZf_eHLxRZQbYU,11207
4
4
  blksprs/ops/conversion.py,sha256=nv5gXiyZkUtk1kCIlPr0Vpaj4G8G6dJdW7StlbV3nDw,21914
@@ -17,7 +17,7 @@ blksprs/utils/blksprs_tensor.py,sha256=pfoz59aJixj_fIoFx76ySiygwRQUemmgjMKepZ2c4
17
17
  blksprs/utils/processing.py,sha256=RNkEDc0g-sNHRuMPkRzNWU13d3_lIkXMJdoqES4yQTM,3738
18
18
  blksprs/utils/tools.py,sha256=TKygEKge4wJtJnXXDg8BTL8vzBpqIJsQ_A3_5FmLpcE,859
19
19
  blksprs/utils/validation.py,sha256=G8eQlvJVMKfEX3k2AwBD0A6Ck-gFoRLpLNY6HXsB3fA,4348
20
- blksprs-2.1.3.dist-info/METADATA,sha256=6ZrxPPpkLwXgmq1d-4VQBNPNjlRm76dEMI-LJyiqlfI,9712
21
- blksprs-2.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
22
- blksprs-2.1.3.dist-info/top_level.txt,sha256=qyp0IHeY3H2GQA97i4hk_To5rRBS2YcE1HRPSLy04fk,8
23
- blksprs-2.1.3.dist-info/RECORD,,
20
+ blksprs-2.1.4.dist-info/METADATA,sha256=qGLQunHEIoHlmRvFnM0TVDjOSApwGzBglpZezmfhHLU,9590
21
+ blksprs-2.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
22
+ blksprs-2.1.4.dist-info/top_level.txt,sha256=qyp0IHeY3H2GQA97i4hk_To5rRBS2YcE1HRPSLy04fk,8
23
+ blksprs-2.1.4.dist-info/RECORD,,