def test_hybrid_search(benchmark, size, dtype, device):
nns_opt = dict(knn=1, radius=0.01)
np_a = np.array(np.random.rand(size, 3), dtype=to_numpy_dtype(dtype))
np_b = np.array(np.random.rand(size, 3), dtype=to_numpy_dtype(dtype))
a = o3c.Tensor(np_a, dtype=dtype, device=device)
b = o3c.Tensor(np_b, dtype=dtype, device=device)
index = NNSOps.hybrid_setup(a, nns_opt)
benchmark(NNSOps.hybrid_search, index, b, nns_opt)
def test_binary_ew_ops(benchmark, size, dtype, op):
np_a = np.array(np.random.uniform(1, 127, size),
dtype=to_numpy_dtype(dtype))
np_b = np.array(np.random.uniform(1, 127, size),
dtype=to_numpy_dtype(dtype))
a = o3c.Tensor(np_a, dtype=dtype, device=o3c.Device("CPU:0"))
b = o3c.Tensor(np_b, dtype=dtype, device=o3c.Device("CPU:0"))
benchmark(op, a, b)
def test_knn_setup(benchmark, size, dim, dtype, device):
nns_opt = dict(knn=1, radius=0.01)
np_a = np.array(np.random.rand(size, dim), dtype=to_numpy_dtype(dtype))
a = o3c.Tensor(np_a, dtype=dtype, device=device)
benchmark(NNSOps.knn_setup, a, nns_opt)
def test_binary_ew_ops_numpy(benchmark, size, dtype, op):
np_a = np.array(np.random.uniform(1, 127, size),
dtype=to_numpy_dtype(dtype))
np_b = np.array(np.random.uniform(1, 127, size),
dtype=to_numpy_dtype(dtype))
benchmark(to_numpy_binary_op(op), np_a, np_b)
def test_float_unary_ew_ops_numpy(benchmark, size, dtype, op):
# Set upper bound to 88 to avoid overflow for exp() op.
np_a = np.array(np.random.uniform(1, 88, size),
dtype=to_numpy_dtype(dtype))
benchmark(to_numpy_unary_op(op), np_a)
def test_float_unary_ew_ops(benchmark, size, dtype, op):
# Set upper bound to 88 to avoid overflow for exp() op.
np_a = np.array(np.random.uniform(1, 88, size),
dtype=to_numpy_dtype(dtype))
a = o3c.Tensor(np_a, dtype=dtype, device=o3c.Device("CPU:0"))
benchmark(op, a)