add version.txt as package data

flashinfer-ai · Jan 28, 2024 · 1f842ba · 1f842ba
1 parent 3c620ae
commit 1f842ba
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -120,4 +120,6 @@ def remove_unwanted_pytorch_nvcc_flags():
         python_requires=">=3.9",
         ext_modules=ext_modules,
         cmdclass={"build_ext": torch_cpp_ext.BuildExtension},
+        package_data={'': ["../version.txt"]},
+        include_package_data=True
     )