How to quantize an inherited linear layer?

[nzomi]

As I understand it, we use the make_quant function to replace all specified linear layers with QuantLinear. However, if our linear layer is inherited from a class (e.g., super().forward() calling nn.Linear), how can we quantize this inherited linear layer?

class module(nn.Linear):
    def __init__(self)
        super().__init__()

    def forward(self, x):
        res = super().forward(x)      
        return res

[nzomi]

Currently, I replace the weight key to match a new module that contains a standard linear layer, but I'm looking for a more general solution.