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
|
Metadata-Version: 2.4
|
|
2
2
|
Name: blksprs
|
|
3
|
-
Version: 2.1.
|
|
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
|
[](https://github.com/FelixSchoen/blksprs/releases)
|
|
23
|
-
[](https://www.python.org/downloads/release/python-3119/)
|
|
24
|
+
[](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.
|
|
90
|
-
- _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.
|
|
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
|
|
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=
|
|
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.
|
|
21
|
-
blksprs-2.1.
|
|
22
|
-
blksprs-2.1.
|
|
23
|
-
blksprs-2.1.
|
|
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,,
|
|
File without changes
|
|
File without changes
|