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},