Skip to content

Commit

Permalink
Update compiler.py ed41787
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 Jul 26, 2024
1 parent 62f5f0d commit eeb6f0a
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 12 deletions.
8 changes: 6 additions & 2 deletions binaries.json
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@
"CM",
"CV",
"CX",
"D",
"DC",
"DD",
"DI",
Expand Down Expand Up @@ -95,6 +94,9 @@
"MX",
"V",
"VC",
"VCI",
"VCV",
"VCX",
"VD",
"VI",
"VIC",
Expand All @@ -106,8 +108,10 @@
"VIX",
"VL",
"VLC",
"VLD",
"VLI",
"VLL",
"VLM",
"VLV",
"VLX",
"VM",
Expand Down Expand Up @@ -139,4 +143,4 @@
}
]
}
}
}
17 changes: 7 additions & 10 deletions ivy/compiler/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,16 +16,14 @@ def source_to_source(
as well 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 'torch_frontend'.
profiling: Whether to add performance profiling.
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 @@ -56,8 +54,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 @@ -127,8 +124,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 @@ -192,8 +189,8 @@ def transpile(
Returns
-------
Either a transpiled Graph or a non-initialized LazyGraph.
"""
Either a transpiled Graph or a non-initialized LazyGraph."""

from ._compiler import transpile as _transpile

return _transpile(
Expand Down

0 comments on commit eeb6f0a

Please sign in to comment.