GitLab

* Fix cuaev build on CUDA 11.5 and latest torch

* Update aev.cu
Co-authored-by: Jinze (Richard) Xue <yueyericardo@gmail.com>

* Fix cub to 1.11.0 for cuda 10.2

* quick save

* quick save

* Fix --cuaev-all-sms

* expecttest

* fix ci

* Fix AEV and CUAEV GPU Device Bug

* format

* apply review suggestion

* apply review suggestion

* add a small test for 2x, which was completely missing from tests

* remove unused code

* flake8

* fix

* rm

* Wrap cub with CUB_NS_PREFIX and remove dependency on Thrust

* update readme

* format

* Update install_cuda.sh
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* assume-unchanged

* fix

* fix cuda ci

* init

* separate to h cpp cu files

* works

* fix

* clean

* update

* fix

* only keep one copy of Result

* const result&

* move declaration to header

* fix jit

* clean

* fix

* fix jit

* fix ci test

* clean

* revert to autograd in charge of it's result

* update

* fix memory issue of result.aev_t

* Update torchani/cuaev/aev.h
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* Update torchani/cuaev/aev.h
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* Update torchani/cuaev/aev.h
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* save

* Update torchani/cuaev/aev.h
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* update

* Update torchani/cuaev/cuaev.cpp
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* Update torchani/cuaev/cuaev.cpp
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* Update torchani/cuaev/aev.h
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* remove aev_t after forward

* remove release function

* fix

* fix jit

* update
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* Update install_cuda.sh

* Update install_cuda.sh

* Update install_cuda.sh

* update

* update

* update

* update

* update

* update

* update

* update

* save

* Remove constant memory usage

* Update torchani/cuaev/aev.cu
Co-authored-by: Jinze Xue <yueyericardo@gmail.com>

* Update torchani/cuaev/aev.cu
Co-authored-by: Jinze Xue <yueyericardo@gmail.com>

* Update torchani/cuaev/aev.cu
Co-authored-by: Jinze Xue <yueyericardo@gmail.com>

* format
Co-authored-by: Jinze Xue <yueyericardo@gmail.com>

* init

* init

* double backward test

* fix doublebackward test

* add another test

* rm gaev

* radial done

* angular init

* angular done

* update

* force training benchmark

* format

* update

* benchmark

* update

* update

* clean redundancy codes

* update

* adapt review request

* update

* update

* update

* update

* update

* update

* fix

* fix

* cuAngularAEVs code deduplicate

* pairwise double backward

* cuRadialAEVs dedup

* pairwiseDistance dedup

* format

* readme build notes

* save

* update

* save

* save

* update

* fix

* save

* add equations on comments
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* improve documentation and fix buggy behavior of shuffle() and cache()

* remove warnings

* Fast Build CUAEV

* fix

* --cuaev for fastbuild, --cuaev-all-sms for all

* update readme

* fix

* fix

* fix

* fix

* fix

* format

* Update torchani/cuaev/aev.cu
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* update

* rm

* update

* update

* update

* ase test atolerent

* preparation

* radial preparation 30%

* radial backward kernel done

* reuse Gmr (exp part) result for gradient

* radial kernel every block run by column major, to avoid atomicAdd waiting

* apply code review

* static_cast

* implicit cast

* format

* angular preparation

* angular backward works, but slow, AtomicAdd should be avoided

* angular opti: use share memory to avoid AtomicAdd

* format

* equation optimization

* remove unnecessary shared mem for atomi

* remove a lot (warpsize * nbr) unnecessary shared mem for atomj

* format

* update

* clean

* fix

* fix

* test file

* fix
Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

Co-authored-by: Gao, Xiang <qasdfgtyuiop@gmail.com>

* Update docs.yml

* Update docs.yml

* Update docs.yml

* Update docs.yml

* Update docs.yml

* Update docs.yml

* Apply suggestions from code review
Co-authored-by: Richard Xue <yueyericardo@gmail.com>
Co-authored-by: Richard Xue <yueyericardo@gmail.com>

* line-limit 120

* CRLF to LF

* clang-format

* line-limit 120

* cuda port for aev computation

* benchmark, setup and import

* fix bug

* fix flake8 and jit

* fix collections.abc for python>3.7

* setup gitingore

* format code with formatter

* ignore more

* Use torch API to set streams

* Use pytorch's caching allocator

* empty line

* fix

* cuaev correntness testr, instruction of install on readme

* readme

* fix readme

* fix readme

* fix readme

* fix readme

* add usage in readme

* fix readme

* add test in readme

* fix readme

* -std=c++14

* bug fix - add async data copy

* bug fix - add missing stream sync

* code refactor and cosmetic changes

* aev benchmark for big protein

* remove mdtraj

* remove print

* move pdb to dataset folder

* cosmetic changes

* Move torchani/extensions -> torchani/cuaev

* clang-format -i

* cleanup

* return aev from cuComputeAEV

* Update aev.py

* Update aev.py

* fix flake8

* fix LGTM unused local variable

* clang-format

* fix

* save

* install change
Co-authored-by: richard <yueyericardo@gmail.com>
Co-authored-by: Xiang Gao <qasdfgtyuiop@gmail.com>

* More about cuaev

* Update setup.py

* save

* clang-format

* commit farhads and jinzes changes to fix some bugs [WIP] (#22)

* commit farhads and jinzes changes to fix some bugs

* Add tests for correct inputs

* add missing

* fix
Co-authored-by: Ignacio Pickering <ign.pickering@gmail.com>

* Move parsing resource code to neurochem/ (#15)

* Move parsing resource code to neurochem/

* fix bug

* fix

* fix
Co-authored-by: Ignacio Pickering <ign.pickering@gmail.com>