[Util] Update brainsmith integration

Author: auphelia

src/finn/util/brainsmith_integration.py

@@ -194,6 +194,24 @@ def _register_kernels():
                 "class_name": "InferPool",
             },
         },
+        {
+            "name": "HWSoftmax",
+            "module": "finn.custom_op.fpgadataflow.hwsoftmax",
+            "class_name": "HWSoftmax",
+            "infer_transform": {
+                "module": "finn.transformation.fpgadataflow.convert_to_hw_layers",
+                "class_name": "InferHWSoftmax",
+            },
+        },
+        {
+            "name": "LayerNorm",
+            "module": "finn.custom_op.fpgadataflow.layernorm",
+            "class_name": "LayerNorm",
+            "infer_transform": {
+                "module": "finn.transformation.fpgadataflow.convert_to_hw_layers",
+                "class_name": "InferLayerNorm",
+            },
+        },
         {
             "name": "Shuffle",
             "module": "finn.custom_op.fpgadataflow.shuffle",
@@ -339,6 +357,13 @@ def _register_backends():
             "target_kernel": "finn:GlobalAccPool",
             "language": "hls",
         },
+        {
+            "name": "HWSoftmax_hls",
+            "module": "finn.custom_op.fpgadataflow.hls.hwsoftmax_hls",
+            "class_name": "HWSoftmax_hls",
+            "target_kernel": "finn:HWSoftmax",
+            "language": "hls",
+        },
         {
             "name": "LabelSelect_hls",
             "module": "finn.custom_op.fpgadataflow.hls.labelselect_hls",
@@ -410,6 +435,13 @@ def _register_backends():
             "target_kernel": "finn:ConvolutionInputGenerator",
             "language": "rtl",
         },
+        {
+            "name": "LayerNorm_rtl",
+            "module": "finn.custom_op.fpgadataflow.rtl.layernorm_rtl",
+            "class_name": "LayerNorm_rtl",
+            "target_kernel": "finn:LayerNorm",
+            "language": "rtl",
+        },
         {
             "name": "MVAU_rtl",
             "module": "finn.custom_op.fpgadataflow.rtl.matrixvectoractivation_rtl",