Skip to content

Profiling layers in an RNN #829

Open
@olaiya

Description

@olaiya

Hello

I'm using hls4ml from the main branch. Looking at hls4ml.version I get 0.8.0.dev11+gabaea98a. I'm looking to profile the layers in my RNN. However when I use

hls4ml.model.profiling.numerical with my model I get the following error:

Traceback (most recent call last):
File "rnn_hls4ml.py", line 124, in
plots = hls4ml.model.profiling.numerical(model=model, hls_model=hls_model_reg, X=X_test)
File "/usr/local/lib/python3.8/dist-packages/hls4ml/model/profiling.py", line 460, in numerical
data = weights_hlsmodel(hls_model_unoptimized, fmt='summary', plot=plot)
File "/usr/local/lib/python3.8/dist-packages/hls4ml/model/profiling.py", line 225, in weights_hlsmodel
label = f'{name}/{suffix[iw]}'

I have generated an example in github to demonstrate the problem.

The example can be found here:

https://github.com/olaiya/rnn_hls4ml/blob/master/rnn_hls4ml.py

You can run the example with

git clone https://github.com/olaiya/rnn_hls4ml.git

cd rnn_hls4ml/

python3 rnn_hls4ml.py

How do I correctly generate the plots for profiling RNN layers?

Thanks

Manny

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions