From 9e9b9d977a4a98bf5d7968e626d84bc43078d952 Mon Sep 17 00:00:00 2001
From: max <maksim.levental@gmail.com>
Date: Sat, 20 Apr 2024 17:09:49 -0500
Subject: [PATCH] pin cloudpickle

---
 examples/cuda_matmul_opt.py | 2 +-
 requirements.txt            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/cuda_matmul_opt.py b/examples/cuda_matmul_opt.py
index 1297a8ad..62865e85 100644
--- 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
index 30a79294..be88e418 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,4 +5,4 @@ bytecode
 inflection
 numpy
 astunparse
-cloudpickle
\ No newline at end of file
+cloudpickle>=3.0.0
\ No newline at end of file