mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
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:
committed by
Soumith Chintala
parent
5c93ca258b
commit
22fe542b8e
@ -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)");
|
||||
}
|
||||
|
Reference in New Issue
Block a user