Hacktoberfest 2026: los issues que los mantenedores marcaron para octubre, abiertos y aptos para principiantes. Explorar issues de Hacktoberfest

jax jitted functions cloudpickled work but include some error messages

Abierto
#537 4 comentarios 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
4/5
Tiempo estimado
3-5 días
Aptitud para principiantes
25/100
Tipo de issue
Error
Claridad
Necesita aclaración
Estado de actividad
Estancado
Stack tecnológico
python
Área
tooling

Línea de trabajo

Comienza ejecutando el reproductor proporcionado test_jax_cloudpickle e inspeccionando los bytes serializados en busca de TRACEBACK. Lee el comportamiento de cloudpickle y JAX implicado en la serialización de funciones jax.jit, incluido si JAX_TRACEBACK_FILTERING es relevante; el issue se resuelve cuando el comportamiento sigue siendo correcto sin el mensaje de error no deseado y la responsabilidad queda clara.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

problems

cloudpickle works for jax.jit functions but a visual inspection of the cloudpickle contents shows there's a lurking error message

challenges

not sure if this belongs in cloudpickle or jax

is this my bad? I was hopeful we could just use the string jaxpr in utf8, it's more human readable, but I don't know how to regenerate a PjitFunction from a jaxpr

opportunities

a fix could reduce the size of cloudpickled jax.jit functions

def test_jax_cloudpickle():
    def jnp_func(x):
        return jax.numpy.sin(jax.numpy.cos(x))

    jitted1 = jax.jit(jnp_func)
    del jnp_func  # this to ensure jitted2 can't cheat by recompiling jnp_func within a session
    assert "jnp_func" not in locals(), "failed to remove jnp_func"
    jitted1_buf = cloudpickle.dumps(jitted1)
    rprint(jitted1_buf)
    jitted2 = cloudpickle.loads(jitted1_buf)
    assert jitted1(0.3) == jitted2(0.3), "weird"
    assert b"TRACEBACK" not in jitted1_buf, "error message in cloudpickle of jax.jit"


test_jax_cloudpickle()

Could JAX_TRACEBACK_FILTERING= be greppable ?

image

thank you for making cloudpickle

Lenguaje dominante
Python
Estrellas
1.9k
Forks
197
Merge medio
1 d 10 h
PR fusionados (30 d)
1

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de cloudpipe/cloudpickle

Todos los issues de cloudpipe/cloudpickle

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.