Rename PTDS environment variable to RMM_PER_THREAD_DEFAULT_STREAM

rapidsai · Nov 23, 2020 · 1ee52af · 1ee52af
1 parent d789117
commit 1ee52af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/rmm/_lib/cuda_stream_view.pyx b/python/rmm/_lib/cuda_stream_view.pyx
@@ -29,7 +29,7 @@ cdef class CudaStreamView:
             CUDA stream to wrap, default 0
         """
         if (stream == 0):
-            if int(os.environ.get("CUDA_PTDS", "0")) != 0:
+            if int(os.environ.get("RMM_PER_THREAD_DEFAULT_STREAM", "0")) != 0:
                 self.c_obj.reset(new cuda_stream_view(cudaStreamPerThread))
             else:
                 self.c_obj.reset(new cuda_stream_view())