Hacktoberfest 2026: những issue maintainer đã đánh dấu cho tháng Mười, đang mở và phù hợp người mới. Xem issue Hacktoberfest

jax jitted functions cloudpickled work but include some error messages

Đang mở
  • Có gì không ổn?
#537 4 bình luận 0 reaction 0 người được giao Xem trên GitHub

Chưa có ai nhận issue này.

Đánh giá

Độ khó
4/5
Thời gian dự kiến
3-5 ngày
Mức phù hợp với người mới
25/100
Loại issue
Lỗi
Độ rõ ràng
Cần làm rõ
Mức độ hoạt động
Đình trệ
Công nghệ
python
Lĩnh vực
tooling

Hướng nghiên cứu

Bắt đầu bằng cách chạy trình tái hiện test_jax_cloudpickle được cung cấp và kiểm tra các byte đã tuần tự hóa để tìm TRACEBACK. Đọc về hành vi của cloudpickle và JAX liên quan đến việc tuần tự hóa các hàm jax.jit, bao gồm cả việc JAX_TRACEBACK_FILTERING có liên quan hay không; issue được giải quyết khi hành vi vẫn đúng mà không có thông báo lỗi không mong muốn và trách nhiệm sở hữu đã rõ ràng.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Mô tả

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

Ngôn ngữ chính
Python
Star
1.9k
Fork
197
Merge trung bình
1 ngày 10 giờ
Pull request đã merge (30 ngày)
1

Chuẩn bị môi trường

Chúng tôi chưa kiểm tra các tệp thiết lập môi trường của dự án này. Hãy bắt đầu từ README và xem hướng dẫn đóng góp lần đầu của chúng tôi để biết các bước chung.

Bắt đầu từ đâu

  1. Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
  2. Bình luận trên issue rằng bạn sẽ nhận — tránh hai người làm cùng một việc.
  3. Fork repository và làm thay đổi trên một nhánh.
  4. Mở pull request có tham chiếu số hiệu của issue.

Issue khác của cloudpipe/cloudpickle

Tất cả issue của cloudpipe/cloudpickle

Issue tương tự

Thêm issue về Python

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.