Skip to content

Commit c47b626

Browse files
committed
Remove kernel timing
1 parent 6a99b05 commit c47b626

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

src/aspire/abinitio/commonline_sync3n.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff 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()

0 commit comments

Comments
 (0)