@@ -20,7 +20,7 @@ def main():
2020 sources = glob .glob (os .path .join ("cpp" , "*.cc" )),
2121 include_dirs = ["tensor-array-repo/Tensor-Array/include" ],
2222 library_dirs = ["tensor-array-repo/Tensor-Array/lib" ],
23- # libraries=["libtensorarray_core "],
23+ libraries = ["tensorarray_core " ],
2424 define_macros = [("VERSION_INFO" , __version__ )],
2525 ),
2626 ]
@@ -48,25 +48,20 @@ def main():
4848
4949 "Environment :: GPU :: NVIDIA CUDA :: 12" ,
5050 ],
51- packages = [
52- "tensor_array.c_data.include" ,
53- "tensor_array.c_data.lib" ,
54- "tensor_array.c_data.scripts" ,
55- "tensor_array.local.scripts" ,
56- ] + find_packages (where = "src" ),
57- package_dir = {
58- "" : "src" ,
59- "tensor_array.c_data.include" : "tensor-array-repo/Tensor-Array/include" ,
60- "tensor_array.c_data.lib" : "tensor-array-repo/Tensor-Array/lib" ,
61- "tensor_array.c_data.scripts" : "tensor-array-repo/Tensor-Array/scripts" ,
62- "tensor_array.local.scripts" : "scripts" ,
51+ packages = find_packages () + ["tensor_array" ],
52+ package_dir = {
53+ "tensor_array" : 'src/tensor_array' ,
6354 },
55+ include_package_data = True ,
6456 package_data = {
65- "tensor_array.c_data.include" : ["**/*.hh" ],
66- "tensor_array.c_data.lib" : ["**/*.so" ],
67- "tensor_array.c_data.scripts" : ["**/*.sh" ],
68- "tensor_array.local.scripts" : ["**/*.sh" ],
57+ "" : ["tensor-array-repo/Tensor-Array/lib/*.so" ],
6958 },
59+ data_files = [
60+ ("tensor-array-repo/Tensor-Array/include" , glob .glob (os .path .join ("cpp" , "include" , "*.hh" ))),
61+ ("tensor-array-repo/Tensor-Array/lib" , glob .glob (os .path .join ("cpp" , "lib" , "*.so" ))),
62+ ("tensor-array-repo/Tensor-Array/scripts" , glob .glob (os .path .join ("scripts" , "*.sh" ))),
63+ ("scripts" , glob .glob (os .path .join ("scripts" , "local" , "*.sh" ))),
64+ ],
7065 ext_modules = ext_modules ,
7166 cmdclass = {
7267 "build_ext" : build_ext ,
0 commit comments