¿Es posible llamar la función numba.cuda.random device dentro de funciones de dispositivo CUDA de numba creadas por el usuario?

Tengo un kernel de CUDA y varias funciones de dispositivo en numba para un proyecto. Cuando trato de llamar a xoroshiro128p_uniform_float32 del módulo numba.cuda.random, siempre obtengo:

import numba
from numba import cuda 
from numba.cuda.random import create_xoroshiro128p_states
from numba.cuda.random import xoroshiro128p_uniform_float64

@cuda.jit('void(float32[:,:])', device=True) 
def device(rng_states):
    thread_id = cuda.grid(1) 
    probability = xoroshiro128p_uniform_float64(rng_states, thread_id) 

@cuda.jit() 
def kernel(rng_states): 
    device(rng_states) 

BPG = 10 
TPB = 10 
rng_states = create_xoroshiro128p_states(BPG * TPB, seed=42069) 
kernel(TPB, BPG)

TypingError: Failed in cuda mode pipeline (step: nopython frontend) 
Untyped global name 'xoroshiro128p_uniform_float32': Cannot determine Numba type of <class 'function'="">

¿Alguien ha llamado correctamente funciones importadas en una función de dispositivo CUDA en numba antes?