diff --git a/build_tools/pytorch.py b/build_tools/pytorch.py index 98511e45cb..fdfdee9b1c 100644 --- a/build_tools/pytorch.py +++ b/build_tools/pytorch.py @@ -13,27 +13,8 @@ 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/ - - - 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]: