File tree Expand file tree Collapse file tree 1 file changed +2
-9
lines changed Expand file tree Collapse file tree 1 file changed +2
-9
lines changed Original file line number Diff line number Diff line change @@ -869,10 +869,6 @@ def _estimate_all_Rijs_cu(self, clmatrix):
869869 blksz = 1024
870870 nblk = (self .n_img + blksz - 1 ) // blksz
871871
872- from time import perf_counter
873-
874- logger .info ("Launching `estimate_all_angles` kernel." )
875- toc0 = perf_counter ()
876872 for j in range (self .n_img ):
877873
878874 # ------------------------------------------
@@ -918,10 +914,8 @@ def _estimate_all_Rijs_cu(self, clmatrix):
918914 ),
919915 )
920916
921- # Force all kernels to complete before timing
917+ # Force all kernels to complete
922918 cp .cuda .runtime .deviceSynchronize ()
923- toc1 = perf_counter ()
924- logger .info (f"estimate_all_angles: { toc1 - toc0 } s" )
925919
926920 # Explicitly inform CuPy we longer need these workspace vars
927921 del hist
@@ -947,9 +941,8 @@ def _estimate_all_Rijs_cu(self, clmatrix):
947941 ),
948942 )
949943
944+ # Force all kernels to complete
950945 cp .cuda .runtime .deviceSynchronize ()
951- toc2 = perf_counter ()
952- logger .info (f"angles_to_rots: { toc2 - toc1 } s" )
953946
954947 # transfer results device to host
955948 rotations = rotations .get ()
You can’t perform that action at this time.
0 commit comments