Skip to content

Routhleck/blocksparse-pytorch-implement

Repository files navigation

blocksparse-pytorch-implement

pytorch implements block sparse

How to use

Compile operator into python libs

at the root directory, use python setup.py build_ext --inplace to compile (DON'T USE NINJA, i don't know why it failed when I use it)

Try test_api.py

it gives a example to use.