search

aiqm / torchani

Accurate Neural Network Potential on PyTorch
https://aiqm.github.io/torchani/
MIT License
459 stars 126 forks source link

Wrap cub in it's own namespace and remove dependency on Thrust #587

Closed yueyericardo closed 3 years ago

yueyericardo commented 3 years ago

This will solve issue #549 related to cuda 11.

Related issues: https://github.com/pytorch/pytorch/pull/55292 https://github.com/dmlc/dgl/pull/2758