Remove duplicated thread-pool api

rapidsai · Oct 25, 2023 · dd4f196 · dd4f196
1 parent c11f774
commit dd4f196
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 36 deletions.
diff --git a/python/kvikio/defaults.py b/python/kvikio/defaults.py
@@ -88,7 +88,9 @@ def num_threads_reset(nthreads: int) -> None:
     Parameters
     ----------
     nthreads : int
-        The number of threads to use.
+        The number of threads to use. The default value can be specified by setting
+        the `KVIKIO_NTHREADS` environment variable. If not set, the default value
+        is 1.
     """
     libkvikio.thread_pool_nthreads_reset(nthreads)
 

diff --git a/python/kvikio/thread_pool.py b/python/kvikio/thread_pool.py