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

Move dtype settings out of metadata field into the root of Tensorstore spec #20960

Merged
merged 1 commit into from May 1, 2024

Conversation

copybara-service[bot]
Copy link

@copybara-service copybara-service bot commented Apr 26, 2024

Move dtype settings out of metadata field into the root of Tensorstore spec

Before, dtype used to be in the metadata field of tensorstore spec because of it was the legacy way to config the dtype. This setting doesn't understand the "str" name, hence, there was special logic to translate bfloat for example.

This CL moves it out of the metadata field and put the dtype directly into the Tensorstore spec to eliminate special dtype translation logic. This will also add support of other quantized types such as int4.

@copybara-service copybara-service bot force-pushed the test_628519426 branch 3 times, most recently from 5b6c7b5 to 4647df1 Compare May 1, 2024 20:14
@copybara-service copybara-service bot changed the title Update Tensorstore dtype spec building to support int4 Move dtype settings out of metadata field into the root of Tensorstore spec May 1, 2024
@copybara-service copybara-service bot changed the title Move dtype settings out of metadata field into the root of Tensorstore spec Move dtype settings out of metadata field into the root of Tensorstore spec May 1, 2024
…e spec

Before, dtype used to be in the metadata field of tensorstore spec because of it was the legacy way to config the dtype.  This setting doesn't understand the "str" name, hence, there was special logic to translate bfloat for example.

This CL moves it out of the metadata field and put the dtype directly into the Tensorstore spec to eliminate special dtype translation logic.  This will also add support of other quantized types such as int4.

PiperOrigin-RevId: 629845048
@copybara-service copybara-service bot merged commit 77988ea into main May 1, 2024
@copybara-service copybara-service bot deleted the test_628519426 branch May 1, 2024 21:49
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.

None yet

1 participant