diff --git a/cuda_core/examples/thread_block_cluster.py b/cuda_core/examples/thread_block_cluster.py index c1785022e3..e14158f8bd 100644 --- a/cuda_core/examples/thread_block_cluster.py +++ b/cuda_core/examples/thread_block_cluster.py @@ -22,6 +22,10 @@ launch, ) +if np.lib.NumpyVersion(np.__version__) < "2.2.5": + print("This example requires NumPy 2.2.5 or later", file=sys.stderr) + sys.exit(0) + # prepare include cuda_path = os.environ.get("CUDA_PATH", os.environ.get("CUDA_HOME")) if cuda_path is None: