diff --git a/hpvm/projects/torch2hpvm/setup.py b/hpvm/projects/torch2hpvm/setup.py
index 7ed8b3b637848dbaddc20fbf8d7c9fa7b33b3bd8..0c2a89fc19a75aca026e4cb8a688de1cbae209e3 100644
--- a/hpvm/projects/torch2hpvm/setup.py
+++ b/hpvm/projects/torch2hpvm/setup.py
@@ -7,6 +7,7 @@ setup(
     author="Yifan Zhao, Yuanjing Shi",
     author_email="yifanz16@illinois.edu, ys26@illinois.edu",
     packages=["torch2hpvm"],
+    package_data={"torch2hpvm": ["*.json", "*.cpp.in"]},
     install_requires=[
         "jinja2>=2.11",
         "networkx>=2.5",