Use TORCH_EXTENSION_NAME macro to avoid mismatched module/extension name (#5277)

* Warn users about mismatched module/extension name

* Define TORCH_EXTENSION_NAME macro
This commit is contained in:
Peter Goldsborough
2018-02-16 19:31:04 -08:00
committed by Soumith Chintala
parent 5c93ca258b
commit 22fe542b8e
8 changed files with 36 additions and 18 deletions

View File

@ -14,6 +14,6 @@ at::Tensor sigmoid_add(at::Tensor x, at::Tensor y) {
return output;
}
PYBIND11_MODULE(torch_test_cuda_extension, m) {
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("sigmoid_add", &sigmoid_add, "sigmoid(x) + sigmoid(y)");
}