Update app.py

app.py

@@ -18,15 +18,21 @@ st.header('Query, Key, Value projection')
 
 mha_flop = 2*bs*n*d*3*d
 mha_bytes = 2*bs*n*d + 2*3*d*d + 2*bs*n*3*d
+mha_int = mha_flop/mha_bytes
+
+mha_flop = round(mha_flop/1e9, 2)
+mha_bytes = round(mha_bytes/1e6, 2)
+
 
 st.subheader("Multi-query Attention")
 c1, c2 = st.columns([2, 3])
-c1.write("FLOP:")
+c1.write("GFLOP:")
 c2.write(str(mha_flop))
-c1.write("Bytes: ")
+c1.write("MB: ")
 c2.write(str(mha_bytes))
 c1.write("Arithm. intensity:")
-c2.write(str(mha_flop/mha_bytes))
+c2.write(str(mha_int))
+
 
 mqa_flop = 2*bs*n*d*(1+2/h)*d
 mqa_bytes = 2*bs*n*d + 2*(2/h)*d*d + 2*bs*n*(2/h)*d