Skip to content

Commit

Permalink
Update compiler.py a39abc5
Browse files Browse the repository at this point in the history
Co-authored-by: ivy-dev-bot <hello@transpile-ai.com>
  • Loading branch information
Sam-Armstrong and ivy-dev-bot authored Sep 17, 2024
1 parent 2ee38cf commit 265b407
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 23 deletions.
12 changes: 7 additions & 5 deletions binaries.json
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,9 @@
{
"utils": [
"C",
"CC",
"CD",
"CI",
"CL",
"CM",
"CV",
"CX",
"D",
Expand Down Expand Up @@ -43,7 +41,6 @@
"III",
"IIL",
"IIM",
"IIV",
"IIX",
"IL",
"ILC",
Expand All @@ -62,7 +59,6 @@
"IMV",
"IMX",
"IV",
"IVC",
"IVD",
"IVI",
"IVL",
Expand Down Expand Up @@ -103,7 +99,13 @@
"VCV",
"VCX",
"VD",
"VDC",
"VDD",
"VDI",
"VDL",
"VDM",
"VDV",
"VDX",
"VI",
"VIC",
"VID",
Expand Down Expand Up @@ -149,4 +151,4 @@
}
]
}
}
}
31 changes: 13 additions & 18 deletions ivy/compiler/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down Expand Up @@ -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(
Expand Down Expand Up @@ -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'.
Expand All @@ -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(
Expand Down Expand Up @@ -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
----------
Expand Down Expand Up @@ -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(
Expand Down Expand Up @@ -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'.
Expand All @@ -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(
Expand Down

0 comments on commit 265b407

Please sign in to comment.