From 3ad856f1bb9f74bb5480a0badfbc7dddb2ba5a5a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= <szymon.ozog@gmail.com>
Date: Fri, 18 Aug 2023 16:41:03 +0200
Subject: [PATCH] Add pycuda to triton dependencies

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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': [