Update codeflash_wrap_decorator.py

Author: KRRT7

codeflash/code_utils/codeflash_wrap_decorator.py

@@ -232,6 +232,7 @@ async def async_wrapper(*args: Any, **kwargs: Any) -> Any:  # noqa: ANN401
         )
 
         exception = None
+        counter = time.perf_counter_ns()
         gc.disable()
         try:
             ret = func(*args, **kwargs)  # coroutine creation has some overhead, though it is very small
@@ -296,6 +297,7 @@ async def async_wrapper(*args: Any, **kwargs: Any) -> Any:  # noqa: ANN401
         print(f"!$######{test_stdout_tag}######$!")
 
         exception = None
+        counter = time.perf_counter_ns()
         gc.disable()
         try:
             ret = func(*args, **kwargs)