Fix jettagger XN>1 issue

Author: abarajithan11

deepsocflow/py/xmodel.py

@@ -368,7 +368,7 @@ def verify_inference(model, hw, SIM, SIM_PATH='', TRACE=False):
         ''' Verify tiled output'''
         if (ib == len(BUNDLES)-1):
             if b.softmax:
-                y_tiled_exp = b.out.ftensor.numpy().reshape(1,b.r.XN,1,b.r.CO)
+                y_tiled_exp = b.out.ftensor.numpy().reshape(1,b.r.XH,1,b.r.CO)
                 y_tiled_sim = np.loadtxt(f"{hw.DATA_DIR}/{b.ib}_y_tiled_sim.txt", np.float32).reshape(y_tiled_exp.shape)/2**17
                 error = np.max(np.abs(y_tiled_sim-y_tiled_exp))
                 assert np.allclose(y_tiled_sim, y_tiled_exp, atol=0.5), f"Error={error}, \nsub:\n{y_tiled_sim-y_tiled_exp} for y_tiled_sim at {b.ib=}. \n y_tiled_sim=\n{y_tiled_sim} \n y_tiled_exp=\n{y_tiled_exp}\n \npre_softmax=\n{b.pre_softmax}"

run/jettagger.py

@@ -202,7 +202,7 @@ def test_dnn_engine(PARAMS):
     '''
     VERIFY & EXPORT
     '''
-    export_inference(loaded_model, hw, hw.ROWS)
+    export_inference(loaded_model, hw, batch_size=hw.ROWS)
     verify_inference(loaded_model, hw, SIM=SIM)
 
     d_perf = predict_model_performance(hw)