TORCH_EXTENSION_NAME

chenyuntc · Sep 13, 2018 · 4d7d63a · 4d7d63a
1 parent d3f27c9
commit 4d7d63a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/neural_renderer/cuda/create_texture_image_cuda.cpp b/neural_renderer/cuda/create_texture_image_cuda.cpp
@@ -28,7 +28,6 @@ at::Tensor create_texture_image(
     return create_texture_image_cuda(vertices_all, textures, image, eps);
 }
 
-// PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
-PYBIND11_MODULE(create_texture_image, m) {
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
     m.def("create_texture_image", &create_texture_image, "CREATE_TEXTURE_IMAGE (CUDA)");
 }
diff --git a/neural_renderer/cuda/load_textures_cuda.cpp b/neural_renderer/cuda/load_textures_cuda.cpp
@@ -34,7 +34,6 @@ at::Tensor load_textures(
 
 }
 
-//PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
-PYBIND11_MODULE(load_textures, m) {
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
     m.def("load_textures", &load_textures, "LOAD_TEXTURES (CUDA)");
 }
diff --git a/neural_renderer/cuda/rasterize_cuda.cpp b/neural_renderer/cuda/rasterize_cuda.cpp
@@ -190,8 +190,7 @@ at::Tensor backward_depth_map(
                                    image_size);
 }
 
-// PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
-PYBIND11_MODULE(rasterize, m) {
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
     m.def("forward_face_index_map", &forward_face_index_map, "FORWARD_FACE_INDEX_MAP (CUDA)");
     m.def("forward_texture_sampling", &forward_texture_sampling, "FORWARD_TEXTURE_SAMPLING (CUDA)");
     m.def("backward_pixel_map", &backward_pixel_map, "BACKWARD_PIXEL_MAP (CUDA)");