diff --git a/setup.py b/setup.py index 33f1e65741..076d14478a 100644 --- a/setup.py +++ b/setup.py @@ -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': [