es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: NUMBA

Crear un np.array lleno de matrices de ceros.

Estoy intentando inicializar un array “vacío” con cada elemento que contenga un array de t_list, un array de np.zeros de 8×8: t_list = np.zeros((8,8), dtype=np.float32) Ahora quisiera tener un np.array con múltiples t_list en cada índice: result = np.array((t_list, t_list, …., tlist)) Me gustaría poder controlar el número de veces . . . Read more

¿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, . . . Read more