Make Captum log_usage universal by supporting async functions (#1614)

Summary:
Pull Request resolved: #1614

As title. Checks if function being wrapped is a coroutine and uses an async wrapper instead of sync when logging.

Reviewed By: vivekmig

Differential Revision: D76837949

Author: craymichael

captum/log/dummy_log.py

@@ -32,13 +32,8 @@ def log_usage(*log_args: Any, **log_kwargs: Any):
     # pyre-fixme[3]: Return type must be annotated.
     # pyre-fixme[2]: Parameter must be annotated.
     def _log_usage(func):
-        @wraps(func)
-        # pyre-fixme[53]: Captured variable `func` is not annotated.
-        # pyre-fixme[3]: Return type must be annotated.
-        def wrapper(*args: Any, **kwargs: Any):
-            return func(*args, **kwargs)
-
-        return wrapper
+        # Apply functools.wraps decorator so func has `__wrapped__` attribute
+        return wraps(func)(func)
 
     return _log_usage