Update build scripts

notebooks/end2end_example/cybersecurity/build_accl.py

@@ -7,34 +7,30 @@ import finn.builder.build_dataflow_config as build_cfg
 
 unsw_nb15_data = np.load("./unsw_nb15_binarized.npz")
 
-inp = unsw_nb15_data["train"][:, :-1]
-inp = np.concatenate([inp, np.zeros((inp.shape[0], 7))], -1).astype(np.float32)
-out = unsw_nb15_data["train"][:, -1].astype(np.float32)
-
-indices = np.where(out == 0)[0][:1]
-
-inp = 2 * inp[indices] - 1
-out = 2 * out[indices] - 1
+inp = np.zeros((1, 3, 32, 32)).astype(np.float32)
+out = np.zeros((1)).astype(np.float32)
 
 np.save(open("input.npy", "wb"), inp)
 np.save(open("expected_output.npy", "wb"), out)
 
 
-model_dir = os.environ['FINN_ROOT'] + "/notebooks/end2end_example/cybersecurity"
-model_file = model_dir + "/cybsec-mlp-ready.onnx"
+# model_dir = os.environ['FINN_ROOT'] + "/pretrained_models"
+# model_file = model_dir + "/cnv-w1a2.onnx"
+
+model_file = '/tmp/finn_dev_streichg/end2end_cnv_w1a1_folded.onnx'
 
 estimates_output_dir = "output_estimates_only"
 
 os.environ["RTLSIM_TRACE_DEPTH"] = "3"
 
 steps = [
-    "step_qonnx_to_finn",
-    "step_tidy_up",
-    "step_streamline",
-    "step_convert_to_hls",
-    "step_create_dataflow_partition",
-    "step_target_fps_parallelization",
-    "step_apply_folding_config",
+    # "step_qonnx_to_finn",
+    # "step_tidy_up",
+    # "step_streamline",
+    # "step_convert_to_hls",
+    # "step_create_dataflow_partition",
+    # "step_target_fps_parallelization",
+    # "step_apply_folding_config",
     "step_minimize_bit_width",
     "step_assign_partition_ids",
     "step_insert_accl",
@@ -50,8 +46,8 @@ cfg_splits = build.DataflowBuildConfig(
     verbose             = True,
     output_dir          = estimates_output_dir,
     steps               = steps,
-    mvau_wwidth_max     = 80,
-    target_fps          = 1000000,
+    # mvau_wwidth_max     = 64,
+    # target_fps          = 1000000,
     synth_clk_period_ns = 10.0,
     generate_outputs    = [
         build_cfg.DataflowOutputType.ESTIMATE_REPORTS,
@@ -62,8 +58,8 @@ cfg_splits = build.DataflowBuildConfig(
     shell_flow_type     = build_cfg.ShellFlowType.VITIS_ALVEO,
     board               = "U55C",
     num_boards          = 2,
-    # start_step="step_create_stitched_ip",
     save_intermediate_models = True,
+    # start_step="step_insert_accl",
 )
 
 build.build_distributed_dataflow_cfg(model_file, cfg_splits)