gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jason Ansel
2022-07-13 23:45:27 -07:00
committed by GitHub
parent 4912916c11
commit 0a3f3d5f25
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/setup.py
View File

@@ -141,7 +141,11 @@ setup(
"filelock", "filelock",
"torch", "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, include_package_data=True,
ext_modules=[CMakeExtension("triton", "triton/_C/")], ext_modules=[CMakeExtension("triton", "triton/_C/")],
cmdclass={"build_ext": CMakeBuild}, cmdclass={"build_ext": CMakeBuild},