diff --git a/gimmik/_version.py b/gimmik/_version.py index 5c3609d..1848e3c 100644 --- a/gimmik/_version.py +++ b/gimmik/_version.py @@ -1,3 +1,3 @@ # -*- coding: utf-8 -*- -__version__ = '3.2' +__version__ = '3.2.1' diff --git a/gimmik/hip.py b/gimmik/hip.py index 17079d4..a58c8fa 100644 --- a/gimmik/hip.py +++ b/gimmik/hip.py @@ -20,7 +20,7 @@ def _kernel_generators(self, dtype, dsize, *, gcn_arch=None, warp_size=64): meta = {'block': (blkx, ms, 1), 'shared': 2*bsz*blkx*dsize} yield ('bstream-msplit', args, meta) - ## Two-way k-split B loading, C streaming kernel + # Two-way k-split B loading, C streaming kernel ks, csz, blkx = 2, 24, 64 args = {'ksplit': ks, 'csz': csz, 'blockx': blkx} meta = {'block': (blkx, ks, 1), 'shared': (ks - 1)*csz*blkx*dsize}