diff --git a/pyproject.toml b/pyproject.toml index 7f90b1edb4e..63ee42ea2fb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,6 +22,7 @@ requires = [ "quart", "numba", "opencv-python-headless<=4.11.0.86", # Required to avoid numpy version conflict with vllm + "triton-ascend==3.2.0rc4", ] build-backend = "setuptools.build_meta" diff --git a/requirements.txt b/requirements.txt index 936de5f6b31..647e57526d7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -28,3 +28,6 @@ numba #--pre #--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi torch-npu==2.7.1 + +# Install triton-ascend +triton-ascend==3.2.0rc4