Spaces:
Running
Running
File size: 1,273 Bytes
cec5823 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
# -*- coding: utf-8 -*-
"""
Created on Sun Mar 3 20:59:03 2024
@author: jamyl
"""
from rstor.data.synthetic_dataloader import DeadLeavesDataset, DeadLeavesDatasetGPU
from time import perf_counter
import numba
def test_gpu_vs_cpu_dataloader():
if not numba.cuda.is_available():
return
n = 10
print("\n")
print("=== Dead leaves with reversing")
dataset = DeadLeavesDatasetGPU(number_of_circles=256, reverse=True)
t1 = perf_counter()
for i in range(n):
_ = dataset[i]
print(f"Mean time on {n} samples (numba) : {(perf_counter()-t1)/n}")
dataset = DeadLeavesDataset(number_of_circles=256, reverse=True)
t1 = perf_counter()
for i in range(n):
_ = dataset[i]
print(f"Mean time on {n} samples (cv2): {(perf_counter()-t1)/n}")
print("=== Dead leaves without reversing")
dataset = DeadLeavesDatasetGPU(number_of_circles=256, reverse=False)
t1 = perf_counter()
for i in range(n):
_ = dataset[i]
print(f"Mean time on {n} samples (numba) : {(perf_counter()-t1)/n}")
dataset = DeadLeavesDataset(number_of_circles=256, reverse=False)
t1 = perf_counter()
for i in range(n):
_ = dataset[i]
print(f"Mean time on {n} samples (cv2): {(perf_counter()-t1)/n}")
|