Jul 26, 2023 · Jul 26, 2023 · Jul 26, 2023 · Jul 26, 2023 · Jul 26, 2023 · Jul 26, 2023
diff --git a/_doc/conf.py b/_doc/conf.py

 # The following is used by sphinx.ext.linkcode to provide links to github
 linkcode_resolve = make_linkcode_resolve(
    "mlstatpy",
    "onnx-array-api",
    (
        "https://github.com/sdpython/onnx-array-api/"
        "blob/{revision}/{package}/"
        "https://data-apis.org/array-api/",
        ("2022.12/API_specification/generated/array_api.{0}.html", 1),
    ),
    "cProfile.Profile": "https://docs.python.org/3/library/profile.html#profile.Profile",
    "DOT": "https://graphviz.org/doc/info/lang.html",
    "JIT": "https://en.wikipedia.org/wiki/Just-in-time_compilation",
    "onnx": "https://onnx.ai/onnx/",
diff --git a/_doc/examples/plot_benchmark_rf.py b/_doc/examples/plot_benchmark_rf.py
 from onnxmltools.convert.xgboost.operator_converters.XGBoost import convert_xgboost
 from onnxruntime import InferenceSession, SessionOptions
 from psutil import cpu_count
 frompyquickhelper.loghelper import run_cmd
 fromsphinx_runpython.runpython import run_cmd
 from skl2onnx import to_onnx, update_registered_converter
 from skl2onnx.common.shape_calculator import calculate_linear_regressor_output_shapes
 from sklearn import set_config
diff --git a/_doc/examples/plot_onnxruntime.py b/_doc/examples/plot_onnxruntime.py
 ###############################
 # Profiling
 # +++++++++
 frompyquickhelper.pycode.profiling import profile, profile2graph
 fromonnx_array_api.profiling import profile, profile2graph

 x = np.random.randn(10000, 2).astype(np.float32)
 y = np.random.randn(10000, 2).astype(np.float32)
diff --git a/onnx_array_api/profiling.py b/onnx_array_api/profiling.py
    ::

        import pstats
        frompyquickhelper.pycode.profiling import profile2df
        fromonnx_array_api.profiling import profile2df

        ps = pstats.Stats('bench_ortmodule_nn_gpu6.prof')
        df = profile2df(pd)
    sort: str = "cumulative",
    rootrem: Optional[str] = None,
    as_df: bool = False,
    return_results=False,
    return_results: bool =False,
    **kwargs,
 ) -> str:
    """
    :param as_df: return the results as a dataframe and not text
    :param return_results: if True, return results as well
        (in the first position)
    :param kwargs: additional parameters used to create the profiler
    :param kwargs: additional parameters used to create the profiler,
        see :epkg:`cProfile.Profile`
    :return: raw results, statistics text dump (or dataframe is *as_df* is True)

    .. plot::

        import matplotlib.pyplot as plt
        from pyquickhelper.pycode.profiling import profile
        from pyquickhelper.texthelper import compare_module_version
        from onnx_array_api.profiling import profile

        def subf(x):
            return sum(x)

        def fctm():
            return compare_module_version('0.20.4', '0.22.dev0')
            x1 = subf([1, 2, 3])
            x2 = subf([1, 2, 3, 4])
            return x1 + x2

        pr, df = profile(lambda: [fctm() for i in range(0, 1000)], as_df=True)
        ax = df[['namefct', 'cum_tall']].head(n=15).set_index(
                        res = res.replace(sub[0], sub[1])
                    else:
                        raise TypeError(
                            "rootrem must contains strings or tuple not {0}"
                            ".".format(rootrem)
                            f"rootrem must contains strings or tuple not {rootrem!r}."
                        )
        return res

diff --git a/requirements-dev.txt b/requirements-dev.txt
 packaging
 pandas
 psutil
 pyquickhelper
 pytest
 pytest-cov
 rstcheck[sphinx,toml]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -54,7 +54,7 @@

		# The following is used by sphinx.ext.linkcode to provide links to github
		linkcode_resolve = make_linkcode_resolve(
		"mlstatpy",
		"onnx-array-api",
		(
		"https://github.com/sdpython/onnx-array-api/"
		"blob/{revision}/{package}/"
Expand DownExpand Up		@@ -114,6 +114,7 @@
		"https://data-apis.org/array-api/",
		("2022.12/API_specification/generated/array_api.{0}.html", 1),
		),
		"cProfile.Profile": "https://docs.python.org/3/library/profile.html#profile.Profile",
		"DOT": "https://graphviz.org/doc/info/lang.html",
		"JIT": "https://en.wikipedia.org/wiki/Just-in-time_compilation",
		"onnx": "https://onnx.ai/onnx/",
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -25,7 +25,7 @@
		from onnxmltools.convert.xgboost.operator_converters.XGBoost import convert_xgboost
		from onnxruntime import InferenceSession, SessionOptions
		from psutil import cpu_count
		frompyquickhelper.loghelper import run_cmd
		fromsphinx_runpython.runpython import run_cmd
		from skl2onnx import to_onnx, update_registered_converter
		from skl2onnx.common.shape_calculator import calculate_linear_regressor_output_shapes
		from sklearn import set_config
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -45,7 +45,7 @@ def myloss(x, y):
		###############################
		# Profiling
		# +++++++++
		frompyquickhelper.pycode.profiling import profile, profile2graph
		fromonnx_array_api.profiling import profile, profile2graph

		x = np.random.randn(10000, 2).astype(np.float32)
		y = np.random.randn(10000, 2).astype(np.float32)
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -491,7 +491,7 @@ def profile2df(
		::

		import pstats
		frompyquickhelper.pycode.profiling import profile2df
		fromonnx_array_api.profiling import profile2df

		ps = pstats.Stats('bench_ortmodule_nn_gpu6.prof')
		df = profile2df(pd)
Expand DownExpand Up		@@ -519,7 +519,7 @@ def profile(
		sort: str = "cumulative",
		rootrem: Optional[str] = None,
		as_df: bool = False,
		return_results=False,
		return_results: bool =False,
		**kwargs,
		) -> str:
		"""
Expand All		@@ -532,17 +532,22 @@ def profile(
		:param as_df: return the results as a dataframe and not text
		:param return_results: if True, return results as well
		(in the first position)
		:param kwargs: additional parameters used to create the profiler
		:param kwargs: additional parameters used to create the profiler,
		see :epkg:`cProfile.Profile`
		:return: raw results, statistics text dump (or dataframe is as_df is True)

		.. plot::

		import matplotlib.pyplot as plt
		from pyquickhelper.pycode.profiling import profile
		from pyquickhelper.texthelper import compare_module_version
		from onnx_array_api.profiling import profile

		def subf(x):
		return sum(x)

		def fctm():
		return compare_module_version('0.20.4', '0.22.dev0')
		x1 = subf([1, 2, 3])
		x2 = subf([1, 2, 3, 4])
		return x1 + x2

		pr, df = profile(lambda: [fctm() for i in range(0, 1000)], as_df=True)
		ax = df[['namefct', 'cum_tall']].head(n=15).set_index(
Expand DownExpand Up		@@ -585,8 +590,7 @@ def clean_text(res):
		res = res.replace(sub[0], sub[1])
		else:
		raise TypeError(
		"rootrem must contains strings or tuple not {0}"
		".".format(rootrem)
		f"rootrem must contains strings or tuple not {rootrem!r}."
		)
		return res

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -15,7 +15,6 @@ openpyxl
		packaging
		pandas
		psutil
		pyquickhelper
		pytest
		pytest-cov
		rstcheck[sphinx,toml]
Expand Down