Skip to content
This repository was archived by the owner on Jan 13, 2024. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion _doc/examples/plot_logistic_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,5 +40,5 @@
# it would be better to use AVX instructions and parallelisation.
# Below, the optimisation this machine can offer.

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_add.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
from skl2onnx.algebra.onnx_ops import OnnxAdd
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_einsum.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from opt_einsum import contract
from mlprodict.testing.experimental_c import (
from mlprodict.testing.experimental_c_impl.experimental_c import (
custom_einsum_float, code_optimisation)
from mlprodict.testing.einsum.einsum_fct import _einsum
print(code_optimisation())
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_onnx_topk.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@
############################################
# Available optimisation on this machine.

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###########################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_reducemax.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from skl2onnx.algebra.onnx_ops import OnnxReduceMax
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_reducemean.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from skl2onnx.algebra.onnx_ops import OnnxReduceMean
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_reducesum.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from skl2onnx.algebra.onnx_ops import OnnxReduceSumApi11
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import (
from mlprodict.testing.experimental_c_impl.experimental_c import (
code_optimisation, custom_reducesum_rk_float)
print(code_optimisation())

Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_reducesumsquare.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from skl2onnx.algebra.onnx_ops import OnnxReduceSumSquare
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_transpose.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
from skl2onnx.algebra.onnx_ops import OnnxTranspose
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_op_where.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from skl2onnx.algebra.onnx_ops import OnnxWhere, OnnxSub, OnnxMul
from cpyquickhelper.numbers import measure_time
from tqdm import tqdm
from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

###################################
Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_opml_linear_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
############################################
# Available optimisation on this machine.

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())


Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_opml_random_forest_cls_multi.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@
############################################
# Available optimisation on this machine.

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())


Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_opml_random_forest_reg.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@
############################################
# Available optimisation on this machine.

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())


Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_parallelism.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
#####################################
# Available optimisations on this machine.

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())


Expand Down
2 changes: 1 addition & 1 deletion _doc/examples/plot_time_tree_ensemble.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,7 @@ def measure_onnx_runtime(model, xt, repeat=REPEAT, number=NUMBER,
###########################################
# Available optimisation on this machine:

from mlprodict.testing.experimental_c import code_optimisation
from mlprodict.testing.experimental_c_impl.experimental_c import code_optimisation
print(code_optimisation())

plt.show()
10 changes: 5 additions & 5 deletions _doc/sphinxdoc/source/api/testing.rst
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ Experimental implementations
Helpers
+++++++

.. autosignature:: mlprodict.testing.experimental_c.code_optimisation
.. autosignature:: mlprodict.testing.experimental_c_impl.experimental_c.code_optimisation

Implementation of ONNX operators
++++++++++++++++++++++++++++++++
Expand All @@ -30,9 +30,9 @@ Einsum

.. autosignature:: mlprodict.testing.einsum.einsum_impl.decompose_einsum_equation

.. autosignature:: mlprodict.testing.experimental_c.custom_einsum_float
.. autosignature:: mlprodict.testing.experimental_c_impl.experimental_c.custom_einsum_float

.. autosignature:: mlprodict.testing.experimental_c.custom_einsum_double
.. autosignature:: mlprodict.testing.experimental_c_impl.experimental_c.custom_einsum_double

.. autosignature:: mlprodict.testing.einsum.einsum_bench.einsum_benchmark

Expand All @@ -54,6 +54,6 @@ Pad
ReduceSum
^^^^^^^^^

.. autosignature:: mlprodict.testing.experimental_c.custom_reducesum_rk_double
.. autosignature:: mlprodict.testing.experimental_c_impl.experimental_c.custom_reducesum_rk_double

.. autosignature:: mlprodict.testing.experimental_c.custom_reducesum_rk_float
.. autosignature:: mlprodict.testing.experimental_c_impl.experimental_c.custom_reducesum_rk_float
2 changes: 1 addition & 1 deletion _unittests/ut_testing/test_custom_add.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import unittest
import numpy
from pyquickhelper.pycode import ExtTestCase
from mlprodict.testing.experimental_c import ( # pylint: disable=E0611
from mlprodict.testing.experimental_c_impl.experimental_c import ( # pylint: disable=E0611
BroadcastMatrixAddLeftInplaceDouble,
BroadcastMatrixAddLeftInplaceFloat,
BroadcastMatrixAddLeftInplaceInt64)
Expand Down
2 changes: 1 addition & 1 deletion _unittests/ut_testing/test_experimental.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from onnx import helper, TensorProto
from pyquickhelper.pycode import ExtTestCase, is_travis_or_appveyor
from mlprodict.testing.experimental import custom_pad, custom_einsum
from mlprodict.testing.experimental_c import ( # pylint: disable=E0611,E0401
from mlprodict.testing.experimental_c_impl.experimental_c import ( # pylint: disable=E0611,E0401
custom_einsum_double, custom_einsum_int64, custom_einsum_float,
code_optimisation, custom_reducesum_rk_double,
custom_reducesum_rk_float)
Expand Down
5 changes: 5 additions & 0 deletions mlprodict/testing/experimental_c_impl/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
"""
@file
@brief Shortcut to *testing.experimental_c*.
"""

6 changes: 3 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -306,15 +306,15 @@ def get_extensions():
language='c++')

ext_experimental_c = Extension(
'mlprodict.testing.experimental_c',
[os.path.join(root, 'mlprodict/testing/experimental_c.cpp')],
'mlprodict.testing.experimental_c_impl.experimental_c',
[os.path.join(root, 'mlprodict/testing/experimental_c_impl/experimental_c.cpp')],
extra_compile_args=extra_compile_args,
extra_link_args=extra_link_args,
include_dirs=[
# Path to pybind11 headers
get_pybind_include(),
get_pybind_include(user=True),
os.path.join(root, 'mlprodict/testing')
os.path.join(root, 'mlprodict/testing/experimental_c_impl')
],
define_macros=define_macros,
language='c++')
Expand Down