Spaces:

harmdevries
/

transformer_inference

Runtime error

harmdevries commited on Oct 23, 2022

Commit

1c3236f

1 Parent(s): 409563e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,9 +102,9 @@ def print_kernel_execution(flop, nbytes):
   nbytes = round(nbytes/1e6, 2)
   c1.write("GFLOP:")
-  c2.write(str(comp_flop))
   c1.write("MB: ")
-  c2.write(str(mem_bytes))
   c1.write("Time (ms):")
   c2.write(str(exec_time))
   c1.write("Overhead (ms):")

   nbytes = round(nbytes/1e6, 2)
   c1.write("GFLOP:")
+  c2.write(str(flop))
   c1.write("MB: ")
+  c2.write(str(nbytes))
   c1.write("Time (ms):")
   c2.write(str(exec_time))
   c1.write("Overhead (ms):")