mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add pycuda to triton dependencies

Browse source
This commit is contained in:
Szymon Ożóg 2023-08-18 16:41:03 +02:00
commit 3ad856f1bb
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
setup.py
View file

@ -25,7 +25,7 @@ setup(name='tinygrad',
'llvm': ["llvmlite"],
'cuda': ["pycuda"],
'arm': ["unicorn"],
'triton': ["triton>=2.0.0.dev20221202"],
'triton': ["triton>=2.0.0.dev20221202", "pycuda"],
'webgpu': ["wgpu"],
'metal': ["pyobjc-framework-Metal", "pyobjc-framework-Cocoa", "pyobjc-framework-libdispatch"],
'linting': [