Skip to content

Commit 6e7f999

Browse files
author
makoeppel
committed
start on FPGA page
1 parent f7f4e03 commit 6e7f999

1 file changed

Lines changed: 180 additions & 1 deletion

File tree

  • content/resources/fpga_resources
Lines changed: 180 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,181 @@
1-
Work in progress.
1+
# CMS-ML FPGA Resource
2+
## Introduction
3+
Welcome to FPGA Resource tab! Our tab is designed to provide accurate, up-to-date, and relevant information about tooling used in CMS for ML on FPGAs.
24

5+
## FPGA Basics
6+
7+
Field-Programmable Gate Array (FPGA) are reconfigurable hardware for creating custom digital circuits. FPGAs are build out of Configurable Logic Blocks (CLBs), Programmable Interconnects, and I/O Blocks. They are used in areas where high parallel processing and high performance is needed.
8+
9+
To programm an FPGA one has two main options. The first one is to use a Hardware Design Languages (HDL) while the second one is to use High-Level Design Tools (HLS). In the HDL case one has the control about everything but the design flow can be kind of tricky. Mostly two languages are used VHDL, which is verbose and structured, and Verilog, which is more compact and looks more like a c-style language. In CMS most of the FPGAs are programmed using VHDL.
10+
11+
When it comes to vendors there are two big onces: Xilinx (part of AMD) and Altera (part of Intel). Both vendors provide there own tooling to simulate, synthesis and debug the design. Xilinx FPGAs are used for CMS and they have Vivado for design, simulation, synthesis, and debugging tasks and Vitis for software development for Xilinx FPGAs and SoCs. Intel FPGAs are programmed using Quartus Prime. For HLS tools they come with Vivado HLS (Xilinx) and HLS Compiler (Intel).
12+
13+
To simplify the pipeline from a trained model to an implementation on the FPGA CMS is supporting different tools, which will be explained in the flowing in more detail.
14+
15+
## 1. hls4ml
16+
17+
![image](https://github.com/fastmachinelearning/fastmachinelearning.github.io/raw/master/images/hls4ml_logo.svg)
18+
19+
[![Documentation Status](https://github.com/fastmachinelearning/hls4ml/actions/workflows/build-sphinx.yml/badge.svg)](https://fastmachinelearning.org/hls4ml)
20+
[![PyPI version](https://badge.fury.io/py/hls4ml.svg)](https://badge.fury.io/py/hls4ml)
21+
22+
#### Description
23+
hls4ml is a Python library designed to bring machine learning inference to FPGAs by leveraging high-level synthesis (HLS). The idea is to convert trained machine learning models from popular open-source frameworks (such as PyTorch, Tensorflow, Keras etc.) into FPGA-compatible firmware, tailored to specific needs.
24+
25+
As the project is actively evolving the hls4ml team is always looking for people trying there tools.
26+
27+
#### Current tools supported
28+
29+
| ML framework/HLS backend | (Q)Keras | PyTorch | (Q)ONNX | Vivado HLS | Intel HLS | Vitis HLS |
30+
|--------------------------|-----------|---------|----------------|------------|-----------|--------------|
31+
| MLP | supported | limited | in development | supported | supported | experimental |
32+
| CNN | supported | limited | in development | supported | supported | experimental |
33+
| RNN (LSTM) | supported | N/A | in development | supported | supported | N/A |
34+
| GNN (GarNet) | supported | N/A | N/A | N/A | N/A | N/A |
35+
36+
37+
### Compile an example model
38+
39+
```python
40+
import hls4ml
41+
42+
# Fetch a keras model from our example repository
43+
# This will download our example model to your working directory and return an example configuration file
44+
config = hls4ml.utils.fetch_example_model('KERAS_3layer.json')
45+
46+
# You can print the configuration to see some default parameters
47+
print(config)
48+
49+
# Convert it to a hls project
50+
hls_model = hls4ml.converters.keras_to_hls(config)
51+
52+
# Print full list of example models if you want to explore more
53+
hls4ml.utils.fetch_example_list()
54+
55+
# Use Vivado HLS to synthesize the model
56+
# This might take several minutes
57+
hls_model.build()
58+
59+
# Print out the report if you want
60+
hls4ml.report.read_vivado_report('my-hls-test')
61+
62+
```
63+
64+
## 2. Conifer
65+
66+
<img src="https://github.com/thesps/conifer/raw/master/conifer_v1.png" width="250" alt="conifer">
67+
68+
Conifer converts from popular BDT training frameworks, and can emit code projects in different FPGA languages.
69+
70+
Available converters:
71+
72+
- scikit-learn
73+
- xgboost
74+
- ONNX - giving access to other training libraries such as lightGBM and CatBoost with ONNXMLTools
75+
- TMVA
76+
- Tensorflow Decision Forest (tf_df)
77+
78+
Available backends:
79+
80+
- Xilinx HLS - for best results use latest Vitis HLS, but Vivado HLS is also supported (conifer uses whichever is on your `$PATH`)
81+
- VHDL - a direct-to-VHDL implementation, deeply pipelined for high clock frequencies
82+
- FPU - Forest Processing Unit reusable IP core for flexible BDT inference
83+
- C++ - intended for bit-accurate emulation on CPU with a single include header file
84+
- Python - intended for validation of model conversion and to allow inspection of a model without a configuration
85+
86+
### Usage
87+
88+
```python
89+
from sklearn.ensemble import GradientBoostingClassifier
90+
# Train a BDT
91+
clf = GradientBoostingClassifier().fit(X_train, y_train)
92+
93+
# Create a conifer config dictionary
94+
cfg = conifer.backends.xilinxhls.auto_config()
95+
# Change the bit precision (print the config to see everything modifiable)
96+
cfg['Precision'] = 'ap_fixed<12,4>'
97+
98+
# Convert the sklearn model to a conifer model
99+
model = conifer.converters.convert_from_sklearn(clf, cfg)
100+
# Write the HLS project and compile the C++-Python bridge
101+
model.compile()
102+
103+
# Run bit-accurate prediction on the CPU
104+
y_hls = model.decision_function(X)
105+
y_skl = clf.decision_function(X)
106+
107+
# Synthesize the model for the target FPGA
108+
model.build()
109+
```
110+
111+
## 3. (Q)ONNX
112+
113+
[![ReadTheDocs](https://readthedocs.org/projects/qonnx/badge/?version=latest&style=plastic)](http://qonnx.readthedocs.io/)
114+
[![PyPI version](https://badge.fury.io/py/qonnx.svg)](https://badge.fury.io/py/qonnx)
115+
[![arxiv](https://img.shields.io/badge/arXiv-2206.07527-b31b1b.svg)](https://arxiv.org/abs/2206.07527)
116+
117+
<img align="left" src="https://xilinx.github.io/finn/img/TFC_1W2A.onnx.png" alt="QONNX example" style="margin-right: 20px" width="200"/>
118+
119+
120+
QONNX (Quantized ONNX) introduces three new custom operators -- [`Quant`](docs/qonnx-custom-ops/quant_op.md), [`BipolarQuant`](docs/qonnx-custom-ops/bipolar_quant_op.md), and [`Trunc`](docs/qonnx-custom-ops/trunc_op.md) -- in order to represent arbitrary-precision uniform quantization in ONNX. This enables:
121+
122+
* Representation of binary, ternary, 3-bit, 4-bit, 6-bit or any other quantization.
123+
* Quantization is an operator itself, and can be applied to any parameter or layer input.
124+
* Flexible choices for scaling factor and zero-point granularity.
125+
* Quantized values are carried using standard `float` datatypes to remain ONNX protobuf-compatible.
126+
127+
This repository contains a set of Python utilities to work with QONNX models, including but not limited to:
128+
129+
* executing QONNX models for (slow) functional verification
130+
* shape inference, constant folding and other basic optimizations
131+
* summarizing the inference cost of a QONNX model in terms of mixed-precision MACs, parameter and activation volume
132+
* Python infrastructure for writing transformations and defining executable, shape-inferencable custom ops
133+
* (experimental) data layout conversion from standard ONNX NCHW to custom QONNX NHWC ops
134+
135+
## 4. High Granularity Quantization (HGQ)
136+
137+
[![docu](https://github.com/calad0i/HGQ/actions/workflows/sphinx-build.yml/badge.svg)](https://calad0i.github.io/HGQ/)
138+
[![pypi](https://badge.fury.io/py/hgq.svg)](https://badge.fury.io/py/hgq)
139+
[![arxiv](https://img.shields.io/badge/arXiv-2405.00645-b31b1b.svg)](https://arxiv.org/abs/2405.00645)
140+
141+
142+
[High Granularity Quantization (HGQ)](https://github.com/calad0i/HGQ/) is a library that performs gradient-based automatic bitwidth optimization and quantization-aware training algorithm for neural networks to be deployed on FPGAs. By laveraging gradients, it allows for bitwidth optimization at arbitrary granularity, up to per-weight and per-activation level.
143+
144+
![image](https://calad0i.github.io/HGQ/_images/overview.svg)
145+
146+
Conversion of models made with HGQ library is fully supported. The HGQ models are first converted to proxy model format, which can then be parsed by hls4ml bit-accurately. Below is an example of how to create a model with HGQ and convert it to hls4ml model.
147+
148+
```python
149+
import keras
150+
from HGQ.layers import HDense, HDenseBatchNorm, HQuantize
151+
from HGQ import ResetMinMax, FreeBOPs
152+
153+
model = keras.models.Sequential([
154+
HQuantize(beta=1.e-5),
155+
HDenseBatchNorm(32, beta=1.e-5, activation='relu'),
156+
HDenseBatchNorm(32, beta=1.e-5, activation='relu'),
157+
HDense(10, beta=1.e-5),
158+
])
159+
160+
opt = keras.optimizers.Adam(learning_rate=0.001)
161+
loss = keras.losses.SparseCategoricalCrossentropy(from_logits=True)
162+
model.compile(optimizer=opt, loss=loss, metrics=['accuracy'])
163+
callbacks = [ResetMinMax(), FreeBOPs()]
164+
165+
model.fit(..., callbacks=callbacks)
166+
167+
from HGQ import trace_minmax, to_proxy_model
168+
from hls4ml.converters import convert_from_keras_model
169+
170+
trace_minmax(model, x_train, cover_factor=1.0)
171+
proxy = to_proxy_model(model, aggressive=True)
172+
173+
model_hls = convert_from_keras_model(
174+
proxy,
175+
backend='vivado',
176+
output_dir=...,
177+
part=...
178+
)
179+
```
180+
181+
An interactive example of HGQ can be found in the [kaggle notebook](https://www.kaggle.com/code/calad0i/small-jet-tagger-with-hgq-1). Full documentation can be found at [calad0i.github.io/HGQ](https://calad0i.github.io/HGQ/>).

0 commit comments

Comments
 (0)