From 0a3f3d5f250394a2b1399c247a96c7839c6d02a1 Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Wed, 13 Jul 2022 23:45:27 -0700 Subject: [PATCH] [PACKAGING] Include triton/language/libdevice.10.bc in package data (#582) --- python/setup.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/setup.py b/python/setup.py index 6c136b6c7..b0e48f251 100644 --- 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},