Coverage for pySDC/implementations/hooks/log_GPU_timings.py: 100%

1import cupy as cp 

2 

3from pySDC.core.timings import Timings 

4 

5 

6class GPUTimings(Timings): 

7 """ 

8 Hook for recording GPU timings of important operations during a pySDC run. 

9 """ 

10 

11 prefix = 'GPU_' 

12 

13 def _compute_time_elapsed(self, event_after, event_before): 

14 event_after.synchronize() 

15 return cp.cuda.get_elapsed_time(event_before, event_after) / 1e3 

16 

17 def _get_event(self): 

18 event = cp.cuda.Event() 

19 event.record() 

20 return event