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.

About

pytorch implements block sparse

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published