jax jitted functions cloudpickled work but include some error messages
还没有人认领这个 Issue。
评估
调研方向
首先运行提供的复现程序 test_jax_cloudpickle，并检查序列化后的字节中是否包含 TRACEBACK。阅读与序列化 jax.jit 函数相关的 cloudpickle 和 JAX 行为，包括 JAX_TRACEBACK_FILTERING 是否相关；当行为在没有不需要的错误消息的情况下仍然正确，并且归属关系明确时，该 issue 即得到解决。
由索引模型根据 Issue 内容生成。
描述
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 ?
thank you for making cloudpickle
- 主要语言
- Python
- 星标
- 1.9k
- 派生
- 197
- 平均合并
- 1 天 10 小时
- 30 天内合并 PR
- 1
环境准备
我们还没有检查这个项目的环境配置文件。先看它的 README，通用步骤见我们的新手贡献指南。
从这里开始
- 先读完整个 Issue，再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库，在一个分支上完成修改。
- 提交 Pull Request，并在描述里引用这个 Issue 编号。
cloudpipe/cloudpickle 的其他 Issue
-
难度 2/5 1-3 小时 新手友好度 78/100
cloudpipe/cloudpickle#593 · 1 条评论 ·
-
难度 4/5 3-5 天 新手友好度 42/100
cloudpipe/cloudpickle#595 ·
-
难度 3/5 1-2 天 新手友好度 65/100
cloudpipe/cloudpickle#592 · 2 条评论 ·
-
难度 5/5 一周以上 新手友好度 35/100
cloudpipe/cloudpickle#589 ·
-
难度 5/5 一周以上 新手友好度 20/100
cloudpipe/cloudpickle#587 ·
查看 cloudpipe/cloudpickle 的全部 Issue
相似的 Issue
-
documentation
难度 2/5 1-3 小时 新手友好度 70/100
kristofdegrave/homeassistant-smart-charging#1413 ·
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 88/100
nasa/earthdata-varinfo#113 ·
-
curriculum documentation quality
难度 2/5 1-3 小时 新手友好度 88/100
githubnext/gh-aw-workshop#3849 ·
维护者通常 2 天内回复
-
难度 2/5 1-3 小时 新手友好度 90/100
-
难度 2/5 1-3 小时 新手友好度 84/100
维护者通常 1 天内回复