Add numpy version check in examples/thread_block_cluster.py

rluo8 · rluo8 · commit 75f82a3bd3db · 2025-12-30T16:45:51.000+08:00
diff --git a/cuda_core/examples/thread_block_cluster.py b/cuda_core/examples/thread_block_cluster.py
@@ -21,9 +21,8 @@
     ProgramOptions,
     launch,
 )
-from packaging import version
 
-if version.parse(np.__version__) < version.parse("2.2.5"):
+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)
 

Original file line number	Diff line number	Diff line change
`@@ -21,9 +21,8 @@`
`21`	`21`	`ProgramOptions,`
`22`	`22`	`launch,`
`23`	`23`	`)`
`24`		`-from packaging import version`
`25`	`24`
`26`		`-if version.parse(np.__version__) < version.parse("2.2.5"):`
	`25`	`+if np.lib.NumpyVersion(np.__version__) < "2.2.5":`
`27`	`26`	`print("This example requires NumPy 2.2.5 or later", file=sys.stderr)`
`28`	`27`	`sys.exit(0)`
`29`	`28`