[BUILD] Add the missing triton/impl to setup.py (#1042)

This commit is contained in:
Da Yan
2023-01-09 11:03:45 -08:00
committed by GitHub
parent c20215dad1
commit 0f5c6e619c

View File

@ -173,7 +173,7 @@ setup(
author_email="phil@openai.com",
description="A language and compiler for custom Deep Learning operations",
long_description="",
packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/ops", "triton/runtime", "triton/ops/blocksparse"],
packages=["triton", "triton/_C", "triton/language", "triton/tools", "triton/impl", "triton/ops", "triton/runtime", "triton/ops/blocksparse"],
install_requires=[
"cmake",
"filelock",