10
10
)
11
11
except :
12
12
raise ModuleNotFoundError ("Please install pytorch >= 1.1 before proceeding." )
13
-
13
+
14
14
import glob
15
15
16
16
from os import path
17
+
17
18
this_directory = path .abspath (path .dirname (__file__ ))
18
- with open (path .join (this_directory , ' README.md' ), encoding = ' utf-8' ) as f :
19
+ with open (path .join (this_directory , " README.md" ), encoding = " utf-8" ) as f :
19
20
long_description = f .read ()
20
21
21
22
@@ -27,7 +28,9 @@ def get_ext_modules():
27
28
extra_compile_args += ["-DVERSION_GE_1_3" ]
28
29
29
30
ext_src_root = "cuda"
30
- ext_sources = glob .glob ("{}/src/*.cpp" .format (ext_src_root )) + glob .glob ("{}/src/*.cu" .format (ext_src_root ))
31
+ ext_sources = glob .glob ("{}/src/*.cpp" .format (ext_src_root )) + glob .glob (
32
+ "{}/src/*.cu" .format (ext_src_root )
33
+ )
31
34
32
35
ext_modules = []
33
36
if CUDA_HOME :
@@ -36,7 +39,10 @@ def get_ext_modules():
36
39
name = "torch_points_kernels.points_cuda" ,
37
40
sources = ext_sources ,
38
41
include_dirs = ["{}/include" .format (ext_src_root )],
39
- extra_compile_args = {"cxx" : extra_compile_args , "nvcc" : extra_compile_args ,},
42
+ extra_compile_args = {
43
+ "cxx" : extra_compile_args ,
44
+ "nvcc" : extra_compile_args ,
45
+ },
40
46
)
41
47
)
42
48
@@ -53,25 +59,28 @@ def get_ext_modules():
53
59
)
54
60
return ext_modules
55
61
62
+
56
63
def get_cmdclass ():
57
64
return {"build_ext" : BuildExtension }
58
65
66
+
59
67
requirements = ["torch>=1.1.0" ]
60
68
61
- url = ' https://github.com/nicolas-chaulet/torch-points-kernels'
62
- __version__ = "0.6.2 "
69
+ url = " https://github.com/nicolas-chaulet/torch-points-kernels"
70
+ __version__ = "0.6.3 "
63
71
setup (
64
72
name = "torch-points-kernels" ,
65
73
version = __version__ ,
66
74
author = "Nicolas Chaulet" ,
67
75
packages = find_packages (),
76
+ description = "PyTorch kernels for spatial operations on point clouds" ,
68
77
url = url ,
69
- download_url = ' {}/archive/{}.tar.gz' .format (url , __version__ ),
78
+ download_url = " {}/archive/{}.tar.gz" .format (url , __version__ ),
70
79
install_requires = requirements ,
71
80
ext_modules = get_ext_modules (),
72
81
cmdclass = get_cmdclass (),
73
82
long_description = long_description ,
74
- long_description_content_type = ' text/markdown' ,
83
+ long_description_content_type = " text/markdown" ,
75
84
classifiers = [
76
85
"Programming Language :: Python :: 3" ,
77
86
"License :: OSI Approved :: MIT License" ,
0 commit comments