-
Notifications
You must be signed in to change notification settings - Fork 18
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
PyTorch wrapper for SchNet operations #40
Conversation
148ea0e
to
9bc3607
Compare
46c9017
to
242a3cd
Compare
f72c076
to
b5f3f03
Compare
@peastman this is ready for the review. |
src/pytorch/CFConvNeighbors.h
Outdated
using torch::Device; | ||
using torch::Tensor; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The using
keyword should never be used in a header file. If you do, it pollutes the namespace of any source file that includes it.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
These are within a namespace, so the mapping is torch::Device
--> NNPOps::CFConvNeighbors::Device
, not just Device
. Also, this header is internal and would be needed for the public API.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I still wouldn't do it. This is just one of those rules of C++ development that it's best to follow consistently. There's little to be gained from breaking it and a lot to lose. Once you write a header file, you never know where someone else might include that file in the future or what assumptions they'll have made.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Removed that stuff
This looks really good! I added just a few minor comments. |
Looks good to merge whenever it's ready. |
CFConvNeighbors
CFConv