From 265b40714861d2df17aea71cc12373949ff670e5 Mon Sep 17 00:00:00 2001 From: Sam Armstrong <88863522+Sam-Armstrong@users.noreply.github.com> Date: Tue, 17 Sep 2024 18:16:37 +0100 Subject: [PATCH] Update compiler.py a39abc5 Co-authored-by: ivy-dev-bot --- binaries.json | 12 +++++++----- ivy/compiler/compiler.py | 31 +++++++++++++------------------ 2 files changed, 20 insertions(+), 23 deletions(-) diff --git a/binaries.json b/binaries.json index e5e7a17a7144..d74dcf6f8531 100644 --- a/binaries.json +++ b/binaries.json @@ -5,11 +5,9 @@ { "utils": [ "C", - "CC", "CD", "CI", "CL", - "CM", "CV", "CX", "D", @@ -43,7 +41,6 @@ "III", "IIL", "IIM", - "IIV", "IIX", "IL", "ILC", @@ -62,7 +59,6 @@ "IMV", "IMX", "IV", - "IVC", "IVD", "IVI", "IVL", @@ -103,7 +99,13 @@ "VCV", "VCX", "VD", + "VDC", + "VDD", "VDI", + "VDL", + "VDM", + "VDV", + "VDX", "VI", "VIC", "VID", @@ -149,4 +151,4 @@ } ] } -} +} \ No newline at end of file diff --git a/ivy/compiler/compiler.py b/ivy/compiler/compiler.py index 4cec6f1c0d97..c59dc9a8340d 100644 --- a/ivy/compiler/compiler.py +++ b/ivy/compiler/compiler.py @@ -2,15 +2,15 @@ def clear_graph_cache(): - """Clears the graph cache which gets populated if `graph_caching` is set to - `True` in `ivy.trace_graph`, `ivy.transpile` or `ivy.unify`. Use this to + """Clears the graph cache which gets populated if `graph_caching` is set + to `True` in `ivy.trace_graph`, `ivy.transpile` or `ivy.unify`. Use this to reset or clear the graph cache if needed. Examples -------- >>> import ivy - >>> ivy.clear_graph_cache() - """ + >>> ivy.clear_graph_cache()""" + from ._compiler import clear_graph_cache as _clear_graph_cache return _clear_graph_cache() @@ -55,8 +55,8 @@ def graph_transpile( Returns ------- - Either a transpiled Graph or a non-initialized LazyGraph. - """ + Either a transpiled Graph or a non-initialized LazyGraph.""" + from ._compiler import graph_transpile as _graph_transpile return _graph_transpile( @@ -96,7 +96,6 @@ def source_to_source( e.g. (source="torch_frontend", target="ivy") or (source="torch_frontend", target="tensorflow") etc. Args: - ---- object: The object (class/function) to be translated. source (str, optional): The source framework. Defaults to 'torch'. target (str, optional): The target framework. Defaults to 'tensorflow'. @@ -107,9 +106,8 @@ def source_to_source( the old implementation. Defaults to 'True'. Returns: - ------- - The translated object. - """ + The translated object.""" + from ._compiler import source_to_source as _source_to_source return _source_to_source( @@ -140,8 +138,7 @@ def trace_graph( params_v=None, v=None ): - """Takes `fn` and traces it into a more efficient composition of backend - operations. + """Takes `fn` and traces it into a more efficient composition of backend operations. Parameters ---------- @@ -211,8 +208,8 @@ def trace_graph( >>> start = time.time() >>> graph(x) >>> print(time.time() - start) - 0.0001785755157470703 - """ + 0.0001785755157470703""" + from ._compiler import trace_graph as _trace_graph return _trace_graph( @@ -252,7 +249,6 @@ def transpile( e.g. (source="torch_frontend", target="ivy") or (source="torch_frontend", target="tensorflow") etc. Args: - ---- object: The object (class/function) to be translated. source (str, optional): The source framework. Defaults to 'torch'. target (str, optional): The target framework. Defaults to 'tensorflow'. @@ -263,9 +259,8 @@ def transpile( the old implementation. Defaults to 'True'. Returns: - ------- - The translated object. - """ + The translated object.""" + from ._compiler import transpile as _transpile return _transpile(