pin cloudpickle (#85)

makslevental · web-flow · commit 22dc84fe058b · 2024-04-20T22:26:07.000Z
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
