(batch_size, img_type, device)
| 219 | |
| 220 | |
| 221 | def check_FastDCT_body(batch_size, img_type, device): |
| 222 | data_path = os.path.join(test_data_root, good_path, img_type) |
| 223 | compare_pipelines( |
| 224 | decoder_pipe( |
| 225 | data_path=data_path, |
| 226 | batch_size=batch_size, |
| 227 | num_threads=3, |
| 228 | device_id=0, |
| 229 | device=device, |
| 230 | use_fast_idct=False, |
| 231 | ), |
| 232 | decoder_pipe( |
| 233 | data_path=data_path, |
| 234 | batch_size=batch_size, |
| 235 | num_threads=3, |
| 236 | device_id=0, |
| 237 | device="cpu", |
| 238 | use_fast_idct=True, |
| 239 | ), |
| 240 | # average difference should be no bigger than off-by-3 |
| 241 | batch_size=batch_size, |
| 242 | N_iterations=3, |
| 243 | eps=3, |
| 244 | ) |
| 245 | |
| 246 | |
| 247 | def test_FastDCT(): |
nothing calls this directly
no test coverage detected