pin cloudpickle

makslevental · makslevental · commit 9e9b9d977a4a · 2024-04-20T17:15:13.000-05:00
diff --git a/examples/cuda_matmul_opt.py b/examples/cuda_matmul_opt.py
@@ -293,7 +293,7 @@ def matmul_mod():
     if "shared" in kernel_name:
         shared_mem = 2 * BLOCK_SIZE * BLOCK_SIZE * npy_dtype().nbytes
     else:
-        shared_mem = None
+        shared_mem = 0
 
     cuda_func(
         grid_dims,
diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,4 @@ bytecode
 inflection
 numpy
 astunparse
-cloudpickle
+cloudpickle>=3.0.0
