Add pycuda to triton dependencies

2023-08-18 16:41:03 +02:00
parent 07d4731e9f
commit 3ad856f1bb
1 changed files with 1 additions and 1 deletions
@@ -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': [