Update app.py

app.py

@@ -71,5 +71,13 @@ att_mqa_time = print_kernel_execution(c1, c2, mqa_flop, mqa_bytes)
 
 st.header('MLP')
 st.subheader('First Linear')
+mlp1_flop = 2*bs*1*d*4*d
+mlp1_bytes = 2*bs*1*d + 2*d*4*d + 2*bs*1*4*d
+c1, c2 = st.columns([2, 3])
+mlp1_time = print_kernel_execution(c1, c2, mlp1_flop, mlp1_bytes)
 
 st.subheader('Second Linear')
+mlp2_flop = 2*bs*1*d*4*d
+mlp2_bytes = 2*bs*1*d + 2*d*4*d + 2*bs*1*4*d
+c1, c2 = st.columns([2, 3])
+mlp2_time = print_kernel_execution(c1, c2, mlp2_flop, mlp2_bytes)