blksprs 2.1.6__tar.gz → 2.1.7__tar.gz

{blksprs-2.1.6 → blksprs-2.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blksprs
-Version: 2.1.6
+Version: 2.1.7
 Summary: A lightweight library for operations on block-sparse matrices in PyTorch.
 Author-email: Felix Schön <schoen@kr.tuwien.ac.at>
 Project-URL: Homepage, https://github.com/FelixSchoen/blksprs

{blksprs-2.1.6 → blksprs-2.1.7}/blksprs/__init__.py

@@ -1,6 +1,6 @@
 from blksprs.utils.blksprs_tensor import BlksprsTensor
 
-__version__ = "2.1.6"
+__version__ = "2.1.7"
 
 
 class ops:

{blksprs-2.1.6 → blksprs-2.1.7}/blksprs/utils/validation.py

@@ -6,10 +6,11 @@ VALIDATION = True
 
 
 def ensure_contiguous(*tensors: Tensor) -> tuple[Tensor, ...]:
-    if _check_skip_contiguous():
-        return tensors
+    transformed = tensors
+
+    if _check_contiguous():
+        transformed = tuple(tensor.contiguous() for tensor in tensors)
 
-    transformed = tuple(tensor.contiguous() for tensor in tensors)
     return transformed[0] if len(transformed) == 1 else transformed
 
 
@@ -133,8 +134,8 @@ def validate_sparsity_block_size(sparsity_block_size: int, *tensors):
             raise ValueError("Tensor sizes must be divisible by sparsity block size")
 
 
-def _check_skip_contiguous():
-    return not CONTIGUOUS
+def _check_contiguous():
+    return CONTIGUOUS
 
 
 def _set_skip_contiguous(skip_contiguous: bool):

{blksprs-2.1.6 → blksprs-2.1.7}/blksprs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blksprs
-Version: 2.1.6
+Version: 2.1.7
 Summary: A lightweight library for operations on block-sparse matrices in PyTorch.
 Author-email: Felix Schön <schoen@kr.tuwien.ac.at>
 Project-URL: Homepage, https://github.com/FelixSchoen/blksprs

{blksprs-2.1.6 → blksprs-2.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blksprs"
-version = "2.1.6"
+version = "2.1.7"
 authors = [{ name = "Felix Schön", email = "schoen@kr.tuwien.ac.at" }]
 description = "A lightweight library for operations on block-sparse matrices in PyTorch."
 readme = "README.md"