Adding profile fix from previous PR

Signed-off-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>

Author: ericspod

monai/utils/profiling.py

@@ -57,7 +57,7 @@ def torch_profiler_full(func):
 
     @wraps(func)
     def wrapper(*args, **kwargs):
-        with torch.autograd.profiler.profile() as prof:
+        with torch.autograd.profiler.profile(use_device="cuda") as prof:
             result = func(*args, **kwargs)
 
         print(prof, flush=True)
@@ -76,7 +76,7 @@ def torch_profiler_time_cpu_gpu(func):
 
     @wraps(func)
     def wrapper(*args, **kwargs):
-        with torch.autograd.profiler.profile() as prof:
+        with torch.autograd.profiler.profile(use_device="cuda") as prof:
             result = func(*args, **kwargs)
 
         cpu_time = prof.self_cpu_time_total