Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

CI: 10/31/24 upstream sync #117

Closed
wants to merge 10 commits into from
Closed

CI: 10/31/24 upstream sync #117

wants to merge 10 commits into from

Conversation

charleshofer
Copy link
Collaborator

Daily sync with upstream

Jake VanderPlas and others added 10 commits October 30, 2024 15:13
…ble with NumPy.

Previously, this change did not account for that fact that `device_get` may be called on objects that have a non-NumPy-compatible `dtype` attribute, such as tensorflow tensors. This change adds new dtype handling aimed at being robust to this case.

Reverts 2bed1e8

PiperOrigin-RevId: 691568933
…kind is `TPU v5`.

PiperOrigin-RevId: 691586791
Part of a plan to move away from our "abstract value" lattice to more traditional types.

PiperOrigin-RevId: 691626481
…batching and simple transposition.

PiperOrigin-RevId: 691706218
… commits SMEM

Users are expected to call `pltpu.commit_smem` manually instead.

PiperOrigin-RevId: 691724662
Also start moving the existing C++ tests to Python.

PiperOrigin-RevId: 691729887
PiperOrigin-RevId: 691830491
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

9 participants