Skip to content

Commit e69a392

Browse files
committed
improve description of brevtias -> QONNX -> hlsm4l workflow
1 parent 16c4055 commit e69a392

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

docs/frontend/pytorch.rst

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@ Provided the underlying operation is supported in ``hls4ml``, we generally aim t
88
tested. Please reach out if you experience any issues with either case.
99

1010
The PyTorch/Brevitas parser is under heavy development and doesn't yet have the same feature set of the Keras parsers. Feel free to reach out to developers if you find a missing feature that is present in Keras parser and would like it implemented.
11-
The direct ingestion of models quantized from brevitas is not yet supported. Exporting brevitas models in the ONNX format (see `here <https://xilinx.github.io/brevitas/tutorials/onnx_export.html>`_) and reading those with the ``hls4ml`` QONNX frontend
12-
might be possible, but is untested.
11+
12+
.. note::
13+
The direct ingestion of models quantized with brevitas is not supported currently. Instead, brevitas models shoud be exported in the ONNX format (see `here <https://xilinx.github.io/brevitas/tutorials/onnx_export.html>`_) and read with the ``hls4ml``
14+
QONNX frontend. Issues may arise, for example when non power-of-2 or non-scalar quantization scales are used. Please reach out if you encounter any problems with this workflow.
1315

1416
For multi-dimensional tensors, ``hls4ml`` follows the channels-last convention adopted by Keras, whereas PyTorch uses channels-first. By default, ``hls4ml`` will automaticlly transpose any tensors associated with weights and biases of the internal layers
1517
of the model. If the ``io_parallel`` I/O type (see :ref:`Concepts`) is used, a transpose node will be added to the model that also adjusts the input tensors. This is not available in the ``io_stream`` case and inputs must be transposed by the user.

0 commit comments

Comments
 (0)