blocksparse/matmul.py (2 lines): - line 161: # save boolean version for serialization purposes, TODO save csr version - line 171: # TODO: write a kernel to do this on the gpu to allow dynamic sparsity blocksparse/transformer.py (2 lines): - line 53: # TODO: support save restore of this object.. - line 74: #self.layout = layout > 0 # save boolean version for serialization purposes, TODO: save packbits or csr version blocksparse/optimize.py (1 line): - line 284: # TODO: maybe traverse deeper? src/blocksparse_matmul_op.cc (1 line): - line 95: // TODO: pack larger values of K in gridZ src/nccl_op.cc (1 line): - line 471: // TODO: Turn on for all ranks blocksparse/conv.py (1 line): - line 433: # TODO: allow a mixture of superblock shapes, or maybe odd shapes to get better ulilization blocksparse/nccl.py (1 line): - line 113: # TODO: support regex's src/gpu_hmma.h (1 line): - line 126: // TODO: thwow a warning here.