change import style for gpu device functions

Author: NimaSarajpoor

stumpy/gpu_aamp.py

@@ -10,6 +10,7 @@
 from numba import cuda
 
 from . import core, config
+from .core import _gpu_searchsorted_right
 
 logger = logging.getLogger(__name__)
 
@@ -186,7 +187,7 @@ def _compute_and_update_PI_kernel(
                 indices_R[j] = i
 
         if p_norm < profile[j, -1]:
-            idx = core._gpu_searchsorted_right(profile[j], p_norm, bfs, nlevel)
+            idx = _gpu_searchsorted_right(profile[j], p_norm, bfs, nlevel)
             for g in range(k - 1, idx, -1):
                 profile[j, g] = profile[j, g - 1]
                 indices[j, g] = indices[j, g - 1]

stumpy/gpu_stump.py

@@ -10,6 +10,7 @@
 from numba import cuda
 
 from . import core, config
+from .core import _gpu_searchsorted_right
 from .gpu_aamp import gpu_aamp
 
 logger = logging.getLogger(__name__)
@@ -194,7 +195,7 @@ def _compute_and_update_PI_kernel(
                 indices_R[j] = i
 
         if p_norm < profile[j, -1]:
-            idx = core._gpu_searchsorted_right(profile[j], p_norm, bfs, nlevel)
+            idx = _gpu_searchsorted_right(profile[j], p_norm, bfs, nlevel)
             for g in range(k - 1, idx, -1):
                 profile[j, g] = profile[j, g - 1]
                 indices[j, g] = indices[j, g - 1]