[PACKAGING] Include triton/language/libdevice.10.bc in package data (#582)

2022-07-13 23:45:27 -07:00
parent 4912916c11
commit 0a3f3d5f25
1 changed files with 5 additions and 1 deletions
--- a/python/setup.py
+++ b/python/setup.py
@@ -141,7 +141,11 @@ setup(
        "filelock",
        "torch",
    ],
-    package_data={"triton/ops": ["*.c"], "triton/ops/blocksparse": ["*.c"]},
+    package_data={
+        "triton/ops": ["*.c"],
+        "triton/ops/blocksparse": ["*.c"],
+        "triton/language": ["*.bc"],
+    },
    include_package_data=True,
    ext_modules=[CMakeExtension("triton", "triton/_C/")],
    cmdclass={"build_ext": CMakeBuild},