update setup.py to support pointnet2 cuda

4edd264b · Shaoshuai Shi · cf29d887 · 4edd264b
Commit 4edd264b authored Jul 23, 2020 by Shaoshuai Shi
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

setup.py setup.py +3 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -77,7 +77,9 @@ if __name__ == '__main__':
                    'src/group_points.cpp',
                    'src/group_points_gpu.cu',
                    'src/sampling.cpp',
-                    'src/sampling_gpu.cu',
+                    'src/sampling_gpu.cu', 
+                    'src/interpolate.cpp', 
+                    'src/interpolate_gpu.cu',
                ],
            ),
        ],