3 import matplotlib.pyplot
as plt
13 os.getenv(
"COMPASS_ROOT") +
"/shesha/data/par/MICADO/micado_39m_PYR_ELTPupil.py")
14 sup.config.p_controller0.set_type(
"generic")
15 sup.config.p_centroider0.set_type(
"maskedpix")
16 sup.config.p_wfs0.roket =
False
19 xvalid = np.array(sup.rtc.rtc.d_centro[0].d_validx)
20 yvalid = np.array(sup.rtc.rtc.d_centro[0].d_validy)
21 cmat = sup.rtc.get_command_matrix(0)
22 frame = sup.wfs.get_wfs_image(0)
26 rtc.add_centroider(sup.context, sup.config.p_wfs0._nvalid,
27 sup.config.p_wfs0.npix / 2 - 0.5, sup.config.p_wfs0.pixsize,
False, 0,
29 rtc.add_controller(sup.context, sup.config.p_wfs0._nvalid,
30 sup.config.p_controller0.nslope, sup.config.p_controller0.nactu,
31 sup.config.p_controller0.delay, 0,
"generic", idx_centro=np.zeros(1), ncentro=1)
32 rtc.d_centro[0].set_npix(sup.config.p_wfs0.npix)
33 rtc.d_centro[0].load_validpos(xvalid, yvalid, xvalid.size)
34 rtc.d_control[0].set_cmat(cmat)
35 rtc.d_control[0].set_gain(sup.config.p_controller0.gain)
36 rtc.d_centro[0].load_img(frame, frame.shape[0])
39 rtcH.add_centroider(sup.context, sup.config.p_wfs0._nvalid,
40 sup.config.p_wfs0.npix / 2 - 0.5, sup.config.p_wfs0.pixsize,
False, 0,
42 rtcH.add_controller(sup.context, sup.config.p_wfs0._nvalid,
43 sup.config.p_controller0.nslope, sup.config.p_controller0.nactu,
44 sup.config.p_controller0.delay, 0,
"generic", idx_centro=np.zeros(1), ncentro=1)
45 rtcH.d_centro[0].set_npix(sup.config.p_wfs0.npix)
46 rtcH.d_centro[0].load_validpos(xvalid, yvalid, xvalid.size)
47 rtcH.d_control[0].set_cmat(cmat)
48 rtcH.d_control[0].set_gain(sup.config.p_controller0.gain)
49 rtcH.d_centro[0].load_img(frame, frame.shape[0])
51 timer = carmaWrap.timer()
53 FP32 = np.zeros(niter)
54 FP16 = np.zeros(niter)
55 FP16TC = np.zeros(niter)
63 for k
in tqdm(range(niter)):
68 FP32[k] = timer.total_time
77 for k
in tqdm(range(niter)):
82 FP16[k] = timer.total_time
85 sup._sim.c.activate_tensor_cores(
True)
92 for k
in tqdm(range(niter)):
97 FP16TC[k] = timer.total_time