Skip to content

Commit 7df7d79

Browse files
committed
docs: Added optional dependency for triton in readme
Signed-off-by: Brandon Groth <[email protected]>
1 parent f9f97d4 commit 7df7d79

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,7 @@ The following optional dependencies are available:
105105
- `mx`: `microxcaling` package for MX quantization
106106
- `opt`: Shortcut for `fp8`, `gptq`, and `mx` installs
107107
- `torchvision`: `torch` package for image recognition training and inference
108+
- `triton`: `triton` package for matrix multiplication kernels
108109
- `visualize`: Dependencies for visualizing models and performance data
109110
- `test`: Dependencies needed for unit testing
110111
- `dev`: Dependencies needed for development

0 commit comments

Comments
 (0)