Update Torch profiler

[jicampos]

Description

Type of change

Currently, the profiler supports only sequential models with nn.Linear layers. This PR enhances its capabilities by:

• Supporting all parameterized layers that subclass nn.Module, including custom layers.
• Adding recursive searching to detect ModuleList and Sequential types, enabling support for more complex architectures.

Note: Please delete options that are not relevant.

• New feature (non-breaking change which adds functionality)

Checklist

• I have read the guidelines for contributing.
• I have commented my code, particularly in hard-to-understand areas.
• I have made corresponding changes to the documentation.
• My changes generate no new warnings.
• I have installed and run pre-commit on the files I edited or added.
• I have added tests that prove my fix is effective or that my feature works.

[JanFSchulte]

pre-commit.ci autofix

[JanFSchulte]

@jicampos Can you have a look at the pre-commit failure? It looks like it is requesting a change in an if condition that it can't auto-fix.

[JanFSchulte]

Looks very good to me. One thing I noticed is that we are here:

hls4ml/hls4ml/model/profiling.py

Line 487 in 47939cb

elif __torch_profiling_enabled__ and isinstance(model, torch.nn.Sequential):

still restricting the profiling to nn.Sequential-type models. I think with this PR, that is no longer necessary, right?

[jicampos]

hls4ml/hls4ml/model/profiling.py

Line 487 in 47939cb

elif __torch_profiling_enabled__ and isinstance(model, torch.nn.Sequential):

still restricting the profiling to nn.Sequential-type models. I think with this PR, that is no longer necessary, right?

@JanFSchulte You are right. I'm updating that now. I think adding a test would be beneficial as well. I don't see any implemented yet.

[JanFSchulte]

Hi @jicampos I think this is basically ready, right? The only thing that would be nice to be added are pytests for it, is that something you can look into?

[jicampos]

@JanFSchulte That's correct—adding a test is all that's needed. I can look into it now that I'm back.

[JanFSchulte]

I think the remaining test failures are totally unrelated to this PR, some connection errors in some of the QKeras tests. So I'll merge now.