Skip to content

Commit

Permalink
Update the number of loops in the benchmarks.
Browse files Browse the repository at this point in the history
  • Loading branch information
jeanfeydy committed Mar 23, 2021
1 parent 1bab0a1 commit fae8ffc
Show file tree
Hide file tree
Showing 7 changed files with 8 additions and 8 deletions.
8 changes: 2 additions & 6 deletions pykeops/benchmarks/benchmark_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -292,11 +292,7 @@ def benchmark(

# Actual benchmark:
elapsed = benchmark_loop(N, loops, routine, max_time, args, kwargs)

if True:
return elapsed
else:
q.put(elapsed)
return elapsed


def bench_config(
Expand Down Expand Up @@ -332,7 +328,7 @@ def bench_config(

times.append(elapsed)
if (nloops * elapsed > max_time) or (
nloops * elapsed > red_time / 10 and len(Nloops) > 0
nloops * elapsed > red_time and len(Nloops) > 0
):
nloops = Nloops.pop(0)

Expand Down
1 change: 1 addition & 0 deletions pykeops/benchmarks/plot_accuracy.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
We test various options of KeOps regarding accuracy of computations.
"""


Expand Down
2 changes: 1 addition & 1 deletion pykeops/benchmarks/plot_benchmark_convolutions.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
where :math:`f` is a Gauss or Cauchy or Laplace or inverse multiquadric kernel. See e.g. `wikipedia <https://en.wikipedia.org/wiki/Radial_basis_function>`_
"""

#####################################################################
Expand Down
1 change: 1 addition & 0 deletions pykeops/benchmarks/plot_benchmark_grad1convolutions.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
where :math:`f` is a Gauss or Cauchy or Laplace or inverse multiquadric kernel. See e.g. `wikipedia <https://en.wikipedia.org/wiki/Radial_basis_function>`_
"""

#####################################################################
Expand Down
1 change: 1 addition & 0 deletions pykeops/benchmarks/plot_benchmark_high_dimension.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
simple Gaussian RBF kernel products,
as the dimension grows.
"""


Expand Down
1 change: 1 addition & 0 deletions pykeops/benchmarks/plot_benchmark_invkernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
and `Falkon <https://falkonml.github.io/falkon/>`_ libraries,
which rely on a KeOps backend whenever relevant.
"""

#####################################################################
Expand Down
2 changes: 1 addition & 1 deletion pykeops/benchmarks/plot_benchmarks_convolutions_3D.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
First support for these approximation schemes is scheduled for
May-June 2021.
"""


Expand Down

0 comments on commit fae8ffc

Please sign in to comment.