Unverified Commit bd007993 authored by Teddy Do's avatar Teddy Do Committed by GitHub
Browse files

Revert adding pytorch-triton as a build requirement (#2592)



* Remove pyhtorch-triton as a requirement and remove auto-fetching pytorch-triton as it is a placeeholder in pyPI
Signed-off-by: default avatartdophung <tdophung@nvidia.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci



* fix docstring
Signed-off-by: default avatartdophung <tdophung@nvidia.com>

---------
Signed-off-by: default avatartdophung <tdophung@nvidia.com>
Co-authored-by: default avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
parent 69636a08
......@@ -13,27 +13,8 @@ from typing import List
def install_requirements() -> List[str]:
"""Install dependencies for TE/PyTorch extensions.
IMPORTANT - PyTorch Index Required for pytorch-triton:
These dependencies MUST be installed using PyTorch's package index:
pip install pytorch-triton --index-url https://download.pytorch.org/whl/<version??>
- pytorch-triton is only available from PyTorch's index (not PyPI)
- The 'pytorch-triton' package on PyPI is a placeholder that will fail
- torch.compile() requires pytorch-triton, not OpenAI's 'triton' package
"""
return [
"torch>=2.1",
"einops",
"onnxscript",
"onnx",
"packaging",
"pydantic",
"nvdlfw-inspect",
"pytorch-triton",
]
"""Install dependencies for TE/PyTorch extensions."""
return ["torch>=2.1", "einops", "onnxscript", "onnx", "packaging", "pydantic", "nvdlfw-inspect"]
def test_requirements() -> List[str]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment