search

huggingface / pytorch_block_sparse

Fast Block Sparse Matrices for Pytorch
Other
545 stars 35 forks source link

`bfloat16` support #19

Open lostmsu opened 2 years ago

lostmsu commented 2 years ago

Does not seem like any dtypes other than torch.float32 are supported.