blksprs 1.9.1__py3-none-any.whl → 1.9.3__py3-none-any.whl

blksprs/utils/blksprs_tensor.py

@@ -1,3 +1,4 @@
+import torch
 from torch import Tensor
 
 
@@ -5,4 +6,5 @@ class BlksprsTensor(Tensor):
     """A wrapper class representing a block-sparse tensor in compressed form.
     """
 
-    pass
+    def __repr__(self):
+        return f"BlksprsTensor({torch.Tensor(self).__repr__()})"

blksprs/utils/processing.py

@@ -66,9 +66,9 @@ def apply_function_applicable_row_wise(x: BlksprsTensor, sparsity_layout: Tensor
 
 
 def _pack_layout(sparsity_layout: Tensor) -> BlksprsTensor:
-    sparsity_layout_resized = sparsity_layout.resize(1, sparsity_layout.size(0) * sparsity_layout.size(1),
-                                                     sparsity_layout.size(2))
-    non_zero_rows = torch.any(sparsity_layout_resized, dim=-1)
-    sparsity_layout_filtered = sparsity_layout_resized[non_zero_rows].unsqueeze(0)
+    sparsity_layout_reshaped = sparsity_layout.reshape(1, sparsity_layout.size(0) * sparsity_layout.size(1),
+                                                       sparsity_layout.size(2))
+    non_zero_rows = torch.any(sparsity_layout_reshaped, dim=-1)
+    sparsity_layout_filtered = sparsity_layout_reshaped[non_zero_rows].unsqueeze(0)
 
     return sparsity_layout_filtered

{blksprs-1.9.1.dist-info → blksprs-1.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: blksprs
-Version: 1.9.1
+Version: 1.9.3
 Summary: A lightweight library for operations on blocksparse matrices in PyTorch.
 Author-email: Felix Schön <schoen@kr.tuwien.ac.at>
 Project-URL: Homepage, https://github.com/FelixSchoen/blksprs
@@ -8,14 +8,15 @@ Project-URL: Bugtracker, https://github.com/FelixSchoen/blksprs/issues
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: torch
-Provides-Extra: build
-Requires-Dist: build; extra == "build"
+Requires-Dist: numpy
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-xdist; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"
 Requires-Dist: coverage; extra == "test"
 Requires-Dist: matplotlib; extra == "test"
+Provides-Extra: build
+Requires-Dist: build; extra == "build"
 
 # blksprs
 
@@ -83,7 +84,8 @@ We recommend installing blksprs from [PyPI](https://pypi.org/project/blksprs/) u
 
 ### Dependencies
 
-- [PyTorch](https://pytorch.org/) (built with v2.5.0)
+- [PyTorch](https://pytorch.org/) (built with v2.5.1)
+- _[NumPy](https://numpy.org/) (to get rid of warnings, built with v2.2.0)_
 - _[Triton](https://github.com/triton-lang/triton) (included with PyTorch)_
 
 ## Changelog

{blksprs-1.9.1.dist-info → blksprs-1.9.3.dist-info}/RECORD

@@ -14,12 +14,12 @@ blksprs/ops/misc/broadcast_ops.py,sha256=cPtRJa3pkZfY1QG51CJ-zDn4SK-CRpX5LEXoKGG
 blksprs/ops/misc/exp.py,sha256=FnSFosBfJHuiEbD0MD-i4axLghRn4a0f8KvHXrKBB6M,3802
 blksprs/ops/misc/row_wise.py,sha256=U4Kk0-P4oOuMNjMHXxP2gP9njMIeMfz8RZrzItNIF94,17229
 blksprs/utils/benchmarking.py,sha256=4pLVlnPW_2EM-NT3n4SClaRznVYEljztLbJcccz8kZE,1360
-blksprs/utils/blksprs_tensor.py,sha256=VjplBgDhnf9sxf-1R5feA0xp5FDCDdaeZmCeoIRdCnc,151
+blksprs/utils/blksprs_tensor.py,sha256=pfoz59aJixj_fIoFx76ySiygwRQUemmgjMKepZ2c4j0,244
 blksprs/utils/layout_utils.py,sha256=49ZdPS_gMn_IrWty3FARbi2rda5a8g5DmAEL8LOrC30,670
-blksprs/utils/processing.py,sha256=6wLvPTWTzXKvXJiiwGQ0iLPaVSrV7FCff6Ru8HrJsro,3636
+blksprs/utils/processing.py,sha256=WLuMJQ8v-YovXwcDjhlDn3N31WMZXrtyeeyKSgq_zn4,3642
 blksprs/utils/tools.py,sha256=r7Y4C37vfSWUyQTGwa8NyRqgovmsq9hMufkenqYHOxo,539
 blksprs/utils/validation.py,sha256=CbxBbeQWJo8wox5eMoVzaTlP9FVBwt3-gxUOmi3EUgw,4213
-blksprs-1.9.1.dist-info/METADATA,sha256=6uy9kTiTgqyQrBgPBeGBYy6orKnJwTHyPUEaEgVu__g,8573
-blksprs-1.9.1.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-blksprs-1.9.1.dist-info/top_level.txt,sha256=qyp0IHeY3H2GQA97i4hk_To5rRBS2YcE1HRPSLy04fk,8
-blksprs-1.9.1.dist-info/RECORD,,
+blksprs-1.9.3.dist-info/METADATA,sha256=bw4yOVw09ggHlKLVjOWE2HCjT-6Iit2v4jtU3zbhTV8,8670
+blksprs-1.9.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+blksprs-1.9.3.dist-info/top_level.txt,sha256=qyp0IHeY3H2GQA97i4hk_To5rRBS2YcE1HRPSLy04fk,8
+blksprs-1.9.3.dist-info/RECORD,,

{blksprs-1.9.1.dist-info → blksprs-1.9.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.3.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any