added cuda extension for efficent implementation

This commit is contained in:
Zach Teed
2020-08-22 18:49:24 -06:00
parent 5b1f510d6b
commit c86b3dc8f3
13 changed files with 519 additions and 191 deletions

1
.gitignore vendored
View File

@@ -5,3 +5,4 @@ datasets
pytorch_env
models
build
correlation.egg-info