@@ -20,7 +20,7 @@ def main():
2020sources = glob .glob (os .path .join ("cpp" ,"*.cc" )),
2121include_dirs = ["tensor-array-repo/Tensor-Array/include" ],
2222library_dirs = ["tensor-array-repo/Tensor-Array/lib" ],
23- # libraries=["libtensorarray_core "],
23+ libraries = ["tensorarray_core " ],
2424define_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 ,
6456package_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+ ],
7065ext_modules = ext_modules ,
7166cmdclass = {
7267"build_ext" :build_ext ,