Skip to content

Commit 3954565

Browse files
FindHaofacebook-github-bot
authored andcommitted
Refactor dependencies in pyproject.toml and remove Triton installation script (#103)
Summary: PyTorch nightly uses a pretty new triton and there is no need to install triton from pypi. So, - Cleared the `dependencies` section in `pyproject.toml` and moved Triton and PyTorch Triton to `optional-dependencies`. - Deleted the `.ci/install-triton-pip.sh` script as it is no longer needed. - Updated the GitHub Actions workflow to remove the Triton installation step. These changes streamline the dependency management and simplify the CI setup. Pull Request resolved: #103 Reviewed By: sfzhu93 Differential Revision: D82350340 Pulled By: FindHao fbshipit-source-id: 3c88be6c6337afdb0135f3637c36438da4d9c03f
1 parent 45c389f commit 3954565

File tree

3 files changed

+7
-77
lines changed

3 files changed

+7
-77
lines changed

.ci/install-triton-pip.sh

Lines changed: 0 additions & 67 deletions
This file was deleted.

.github/workflows/test.yml

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -192,12 +192,6 @@ jobs:
192192
run: |
193193
bash .ci/setup.sh
194194
195-
- name: Install Triton from pip
196-
env:
197-
CONDA_ENV: tritonparse-pip
198-
run: |
199-
bash .ci/install-triton-pip.sh
200-
201195
- name: Install project dependencies
202196
env:
203197
CONDA_ENV: tritonparse-pip

pyproject.toml

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,7 @@ build-backend = "setuptools.build_meta"
55
[project]
66
name = "tritonparse"
77
dynamic = ["version"]
8-
dependencies = [
9-
"triton>3.3.1",
10-
]
8+
dependencies = []
119
requires-python = ">=3.10"
1210
description = "TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels"
1311
readme = "README.md"
@@ -17,8 +15,13 @@ authors = [
1715
license = "BSD-3-Clause"
1816
license-files = ["LICENSE*"]
1917

20-
2118
[project.optional-dependencies]
19+
triton = [
20+
"triton>3.3.1",
21+
]
22+
pytorch-triton = [
23+
"pytorch-triton>=3.4.0",
24+
]
2225
test = [
2326
"coverage>=7.0.0",
2427
]

0 commit comments

Comments
 (0)