Skip to content
New issue

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

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

Already on GitHub? # to your account

Fix (proxy): clean-up #1011

Merged
merged 7 commits into from
Sep 3, 2024
Merged

Fix (proxy): clean-up #1011

merged 7 commits into from
Sep 3, 2024

Conversation

Giuseppe5
Copy link
Collaborator

No description provided.


class BiasQuantProxyFromInjectorBase(ParameterQuantProxyFromInjector, BiasQuantProxyProtocol, ABC):

def __init__(self, quant_layer: nn.Module, quant_injector: Injector) -> None:
super().__init__(quant_layer, quant_injector)
self._cached_bias = None
self.cache_inference_quant_bias = False
self.cache_inference_quant_bias_metadata_only = False
self.requires_input_scale = self.quant_injector.requires_input_scale
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The previous solution is not compile-friendly (even though this PR is not about compile).
Happy to restore this to previous state.

Copy link
Collaborator

@nickfraser nickfraser left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fix export mode for weights. I might have a think about the act proxy forward pass logic.

Otherwise, LGTM!

src/brevitas/proxy/parameter_quant.py Show resolved Hide resolved
@nickfraser nickfraser self-requested a review September 3, 2024 10:51
Copy link
Collaborator

@nickfraser nickfraser left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

See note about the tuple checks.

src/brevitas/proxy/runtime_quant.py Outdated Show resolved Hide resolved
@nickfraser nickfraser self-requested a review September 3, 2024 14:05
@nickfraser nickfraser merged commit dfd9e9f into Xilinx:dev Sep 3, 2024
337 checks passed
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants