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

Refactor tensors #147

Merged
merged 3 commits into from
Apr 2, 2024
Merged

Refactor tensors #147

merged 3 commits into from
Apr 2, 2024

Conversation

dacorvo
Copy link
Collaborator

@dacorvo dacorvo commented Apr 2, 2024

This makes the distinction between quantized activations (8-bit per-tensor only) and quantized weights (per-axis) clearer.

This better separates the responsibilities between Tensor subclasses
and their quantizers.
It also creates dedicated quantize_weight and quantize_activation helper
to better illustrate the differences between the two quantized Tensor types.
@dacorvo dacorvo force-pushed the refactor_tensors branch 4 times, most recently from c77b612 to cb66d7a Compare April 2, 2024 08:53
@dacorvo dacorvo merged commit e905dc3 into main Apr 2, 2024
4 checks passed
@dacorvo dacorvo deleted the refactor_tensors branch April 2, 2024 09:53
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.

1 participant