pre-commit

Author: jmduarte

hls4ml/backends/vivado/passes/convolution_templates.py

@@ -95,7 +95,11 @@ def format(self, node):
         else:
             params['fill_fn'] = 'FillConv1DBuffer'
 
-        is_pointwise_parallel_latency = node.get_attr('filt_width') == 1 and node.get_attr('strategy').lower() == 'latency' and node.model.config.get_config_value('IOType') == 'io_parallel'
+        is_pointwise_parallel_latency = (
+            node.get_attr('filt_width') == 1
+            and node.get_attr('strategy').lower() == 'latency'
+            and node.model.config.get_config_value('IOType') == 'io_parallel'
+        )
         if is_pointwise_parallel_latency:
             params['conv_fn'] = f'pointwise_conv_{node.index}'
         else:
@@ -108,7 +112,9 @@ def format(self, node):
 
         mult_params = self._default_config_params(node)
         if is_pointwise_parallel_latency:
-            mult_params['n_in'] = int(node.get_attr('in_width') * node.get_attr('n_chan') * node.get_attr('filt_width') / mult_params['reuse'])
+            mult_params['n_in'] = int(
+                node.get_attr('in_width') * node.get_attr('n_chan') * node.get_attr('filt_width') / mult_params['reuse']
+            )
             mult_params['n_out'] = int(node.get_attr('in_width') * node.get_attr('n_filt') / mult_params['reuse'])
         else:
             mult_params['n_in'] = node.get_attr('n_chan') * node.get_attr('filt_width')

hls4ml/backends/vivado/passes/pointwise_codegen.py

@@ -66,7 +66,11 @@ class GeneratePointwiseConv1D(OptimizerPass):
     '''Generates code for pointwise 1D convolution'''
 
     def match(self, node):
-        return isinstance(node, Conv1D) and node.model.config.get_config_value('IOType') == 'io_parallel' and node.get_attr('filt_width') == 1
+        return (
+            isinstance(node, Conv1D)
+            and node.model.config.get_config_value('IOType') == 'io_parallel'
+            and node.get_attr('filt_width') == 1
+        )
 
     def transform(self, model, node):
         self._generate_pointwise_conv1d(node)

hls4ml/templates/vitis/nnet_utils/nnet_conv1d_latency.h

@@ -157,7 +157,7 @@ void pointwise_conv_1d_latency_cl(data_T data[CONFIG_T::in_width * CONFIG_T::n_c
     // Cast to "res_t" type
     for (int ii = 0; ii < CONFIG_T::out_width / CONFIG_T::reuse_factor; ii++) {
         for (int ff = 0; ff < CONFIG_T::n_filt; ff++) {
-            #pragma HLS UNROLL        
+            #pragma HLS UNROLL
             res[ii * CONFIG_T::n_filt + ff] = cast<data_T, res_T, typename CONFIG_T::mult_config>(acc[ii][ff]);
         }
     }

hls4ml/templates/vivado/nnet_utils/nnet_conv1d_latency.h

@@ -156,7 +156,7 @@ void pointwise_conv_1d_latency_cl(data_T data[CONFIG_T::in_width * CONFIG_T::n_c
     // Cast to "res_t" type
     for (int ii = 0; ii < CONFIG_T::out_width / CONFIG_T::reuse_factor; ii++) {
         for (int ff = 0; ff < CONFIG_T::n_filt; ff++) {
-            #pragma HLS UNROLL        
+            #pragma HLS UNROLL
             res[ii * CONFIG_T::n_filt + ff] = cast<data_T, res_T, typename CONFIG_T::mult_config>(acc[ii][ff]);
         }
     }

hls4ml/templates/vivado/nnet_utils/nnet_function_stubs.h

@@ -39,8 +39,7 @@ template <class data_T, class res_T, typename CONFIG_T> class DenseKernel {
 
 template <class data_T, class res_T, typename CONFIG_T> class Conv1DKernel {
   public:
-    static void conv(data_T data[CONFIG_T::in_width * CONFIG_T::n_chan],
-                     res_T res[CONFIG_T::out_width * CONFIG_T::n_filt],
+    static void conv(data_T data[CONFIG_T::in_width * CONFIG_T::n_chan], res_T res[CONFIG_T::out_width * CONFIG_T::n_filt],
                      typename CONFIG_T::weight_t weights[CONFIG_T::n_chan * CONFIG_T::n_filt],
                      typename CONFIG_T::bias_t biases[CONFIG_T::n_filt]) {
         // To be implemented in subclasses