77
88from httomolibgpu .prep .normalize import normalize
99from httomolibgpu .recon .algorithm import (
10- FBP ,
11- LPRec ,
10+ FBP3d_tomobar ,
11+ LPRec3d_tomobar ,
1212)
1313from httomolibgpu .misc .morph import sino_360_to_180
1414from numpy .testing import assert_allclose
1818from conftest import force_clean_gpu_memory
1919
2020
21- def test_reconstruct_FBP_i12_dataset1 (i12_dataset1 ):
21+ def test_reconstruct_FBP3d_tomobar_i12_dataset1 (i12_dataset1 ):
2222 force_clean_gpu_memory ()
2323 projdata = i12_dataset1 [0 ]
2424 angles = i12_dataset1 [1 ]
@@ -30,7 +30,7 @@ def test_reconstruct_FBP_i12_dataset1(i12_dataset1):
3030 del flats , darks , projdata
3131 force_clean_gpu_memory ()
3232
33- recon_data = FBP (
33+ recon_data = FBP3d_tomobar (
3434 data_normalised ,
3535 np .deg2rad (angles ),
3636 center = 1253.75 ,
@@ -64,13 +64,13 @@ def test_reconstruct_LP_REC_i13_dataset1(i13_dataset1):
6464 # GPU archetictures older than 5.3 wont accept the data larger than
6565 # (4096, 4096, 4096), while the newer ones can accept (16384 x 16384 x 16384)
6666
67- # recon_data = FBP (
67+ # recon_data = FBP3d_tomobar (
6868 # stiched_data_180degrees,
6969 # np.deg2rad(angles[0:3000]),
7070 # center=2322,
7171 # filter_freq_cutoff=0.35,
7272 # )
73- recon_data = LPRec (
73+ recon_data = LPRec3d_tomobar (
7474 data = stiched_data_180degrees ,
7575 angles = np .deg2rad (angles [0 :3000 ]),
7676 center = 2322.08 ,
@@ -84,7 +84,7 @@ def test_reconstruct_LP_REC_i13_dataset1(i13_dataset1):
8484
8585
8686@pytest .mark .perf
87- def test_FBP_performance_i13_dataset2 (i13_dataset2 ):
87+ def test_FBP3d_tomobar_performance_i13_dataset2 (i13_dataset2 ):
8888 force_clean_gpu_memory ()
8989 dev = cp .cuda .Device ()
9090 projdata = i13_dataset2 [0 ]
@@ -98,7 +98,7 @@ def test_FBP_performance_i13_dataset2(i13_dataset2):
9898 force_clean_gpu_memory ()
9999
100100 # cold run first
101- FBP (
101+ FBP3d_tomobar (
102102 data_normalised ,
103103 np .deg2rad (angles ),
104104 center = 1253.75 ,
@@ -109,7 +109,7 @@ def test_FBP_performance_i13_dataset2(i13_dataset2):
109109 start = time .perf_counter_ns ()
110110 nvtx .RangePush ("Core" )
111111 for _ in range (10 ):
112- FBP (
112+ FBP3d_tomobar (
113113 data_normalised ,
114114 np .deg2rad (angles ),
115115 center = 1286.25 ,
@@ -122,7 +122,7 @@ def test_FBP_performance_i13_dataset2(i13_dataset2):
122122 assert "performance in ms" == duration_ms
123123
124124
125- def test_reconstruct_LPREC_i13_dataset2 (i13_dataset2 ):
125+ def test_reconstruct_LPRec3d_tomobar_i13_dataset2 (i13_dataset2 ):
126126 force_clean_gpu_memory ()
127127 projdata = i13_dataset2 [0 ]
128128 angles = i13_dataset2 [1 ]
@@ -134,7 +134,7 @@ def test_reconstruct_LPREC_i13_dataset2(i13_dataset2):
134134 del flats , darks , projdata
135135 force_clean_gpu_memory ()
136136
137- recon_data = LPRec (
137+ recon_data = LPRec3d_tomobar (
138138 data = data_normalised ,
139139 angles = np .deg2rad (angles ),
140140 center = 1286.25 ,
@@ -148,7 +148,7 @@ def test_reconstruct_LPREC_i13_dataset2(i13_dataset2):
148148
149149
150150@pytest .mark .perf
151- def test_LPREC_performance_i13_dataset2 (i13_dataset2 ):
151+ def test_LPRec3d_tomobar_performance_i13_dataset2 (i13_dataset2 ):
152152 dev = cp .cuda .Device ()
153153 projdata = i13_dataset2 [0 ]
154154 angles = i13_dataset2 [1 ]
@@ -161,7 +161,7 @@ def test_LPREC_performance_i13_dataset2(i13_dataset2):
161161 force_clean_gpu_memory ()
162162
163163 # cold run first
164- LPRec (
164+ LPRec3d_tomobar (
165165 data = data_normalised ,
166166 angles = np .deg2rad (angles ),
167167 center = 1286.25 ,
@@ -171,7 +171,7 @@ def test_LPREC_performance_i13_dataset2(i13_dataset2):
171171 start = time .perf_counter_ns ()
172172 nvtx .RangePush ("Core" )
173173 for _ in range (10 ):
174- LPRec (
174+ LPRec3d_tomobar (
175175 data = data_normalised ,
176176 angles = np .deg2rad (angles ),
177177 center = 1286.25 ,
@@ -183,7 +183,7 @@ def test_LPREC_performance_i13_dataset2(i13_dataset2):
183183 assert "performance in ms" == duration_ms
184184
185185
186- def test_reconstruct_FBP_i13_dataset3 (i13_dataset3 ):
186+ def test_reconstruct_FBP3d_tomobar_i13_dataset3 (i13_dataset3 ):
187187 force_clean_gpu_memory ()
188188 projdata = i13_dataset3 [0 ]
189189 angles = i13_dataset3 [1 ]
@@ -203,7 +203,7 @@ def test_reconstruct_FBP_i13_dataset3(i13_dataset3):
203203 # GPU archetictures older than 5.3 wont accept the data larger than
204204 # (4096, 4096, 4096), while the newer ones can accept (16384 x 16384 x 16384)
205205
206- recon_data = FBP (
206+ recon_data = FBP3d_tomobar (
207207 stiched_data_180degrees ,
208208 np .deg2rad (angles [0 :3000 ]),
209209 center = 2341 ,
0 commit comments