Method forward

utils/quant.py:335–337

(self, input: torch.Tensor)

        return fp8_autocast_fn(enabled=True, fp8_recipe=self._recipe)

    def forward(self, input: torch.Tensor) -> torch.Tensor:
        with self._autocast_context():
            return self.linear(input)

    def extra_repr(self) -> str:
        return (