blksprs 1.10__py3-none-any.whl → 1.10.1__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.
Files changed (5) hide show
  1. blksprs/ops/conversion.py +5 -5
  2. {blksprs-1.10.dist-info → blksprs-1.10.1.dist-info}/METADATA +1 -1
  3. {blksprs-1.10.dist-info → blksprs-1.10.1.dist-info}/RECORD +5 -5
  4. {blksprs-1.10.dist-info → blksprs-1.10.1.dist-info}/WHEEL +0 -0
  5. {blksprs-1.10.dist-info → blksprs-1.10.1.dist-info}/top_level.txt +0 -0
blksprs/ops/conversion.py CHANGED
@@ -52,12 +52,12 @@ def to_dense(x: BlksprsTensor, sparsity_layout: Tensor, sparsity_block_size: int
52
52
  validate_contiguous(sparsity_reverse_lut)
53
53
 
54
54
  if sparsity_layout.size(1) == 1 and sparsity_layout.size(2) == 1 and torch.all(sparsity_layout):
55
- return BlksprsTensor(x)
55
+ return x
56
56
 
57
- return BlksprsTensor(_BlocksparseToDense.apply(x,
58
- sparsity_layout, sparsity_reverse_lut,
59
- sparsity_block_size, fill_value,
60
- triton_block_size))
57
+ return _BlocksparseToDense.apply(x,
58
+ sparsity_layout, sparsity_reverse_lut,
59
+ sparsity_block_size, fill_value,
60
+ triton_block_size)
61
61
 
62
62
 
63
63
  class _BlocksparseToDense(torch.autograd.Function):
{blksprs-1.10.dist-info → blksprs-1.10.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: blksprs
3
- Version: 1.10
3
+ Version: 1.10.1
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
{blksprs-1.10.dist-info → blksprs-1.10.1.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  blksprs/__init__.py,sha256=wnpk-20jXq7xV0xa-WpHfPQuauI2gEZz9sH-0blKxP0,1766
2
2
  blksprs/layouting/distribution_layout.py,sha256=xDGY5-J7uSD8oenlf8bEJ2amMiQG3NBf2klTTydbTJE,5140
3
3
  blksprs/layouting/sparsity_layout.py,sha256=IVtHc_nN3ZM2y4GFcys70PqDWmWc7tkHlVGlToErANk,9894
4
- blksprs/ops/conversion.py,sha256=-KeVaOUdMB0aAj68XZjyzZgf0Dfg5Tt5AnWgx4AZVCY,22320
4
+ blksprs/ops/conversion.py,sha256=NK5uXMepPJ9yYh0vnxKwx5_Ffj_bAvhqPVogf_7PY0g,22248
5
5
  blksprs/ops/distribution.py,sha256=qK5t5XgQSJxXPced8RohprqCtUMMTaEP2pFm3KU1c8o,20267
6
6
  blksprs/ops/flow.py,sha256=SWHDQ5zx0cjnPR0CcAcRNZdSusSAHSU840SwDNUr24g,6437
7
7
  blksprs/ops/matmul.py,sha256=LAQyPNwWVmBMRnAex3msLSPD_aG5SblLCMiutJWqvus,11632
@@ -18,7 +18,7 @@ blksprs/utils/layout_utils.py,sha256=49ZdPS_gMn_IrWty3FARbi2rda5a8g5DmAEL8LOrC30
18
18
  blksprs/utils/processing.py,sha256=WLuMJQ8v-YovXwcDjhlDn3N31WMZXrtyeeyKSgq_zn4,3642
19
19
  blksprs/utils/tools.py,sha256=r7Y4C37vfSWUyQTGwa8NyRqgovmsq9hMufkenqYHOxo,539
20
20
  blksprs/utils/validation.py,sha256=CbxBbeQWJo8wox5eMoVzaTlP9FVBwt3-gxUOmi3EUgw,4213
21
- blksprs-1.10.dist-info/METADATA,sha256=eiTG-EDaZAlRVbof5WsZQpzGmmL5nPoBVDok4VdatJI,9105
22
- blksprs-1.10.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
23
- blksprs-1.10.dist-info/top_level.txt,sha256=qyp0IHeY3H2GQA97i4hk_To5rRBS2YcE1HRPSLy04fk,8
24
- blksprs-1.10.dist-info/RECORD,,
21
+ blksprs-1.10.1.dist-info/METADATA,sha256=5in6lYCZo1bd8urYR0wkTxIiTTAIAANukLpKeZfGasY,9107
22
+ blksprs-1.10.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
23
+ blksprs-1.10.1.dist-info/top_level.txt,sha256=qyp0IHeY3H2GQA97i4hk_To5rRBS2YcE1HRPSLy04fk,8
24
+ blksprs-1.10.1.dist-info/RECORD,,