Skip to content

Commit

Permalink
Update compiler.py 78e5a95 (#28762)
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 Jun 5, 2024
1 parent d4389a0 commit bbebaea
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 11 deletions.
2 changes: 1 addition & 1 deletion binaries.json
Original file line number Diff line number Diff line change
Expand Up @@ -72,4 +72,4 @@
}
]
}
}
}
2 changes: 1 addition & 1 deletion ivy/_version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.0.8.0"
__version__ = "0.0.9.1"
17 changes: 8 additions & 9 deletions ivy/compiler/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,13 @@ def trace_graph(
static_argnums: Optional[Union[int, Iterable[int]]] = None,
static_argnames: Optional[Union[str, Iterable[str]]] = None,
compile_mode: Optional[str] = None,
graph_caching: bool = False,
graph_caching: bool = True,
args: Optional[Sequence] = None,
kwargs: Optional[Mapping] = None,
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 @@ -92,8 +91,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 @@ -127,7 +126,7 @@ def transpile(
static_argnums: Optional[Union[int, Iterable[int]]] = None,
static_argnames: Optional[Union[str, Iterable[str]]] = None,
compile_mode: Optional[str] = None,
graph_caching: bool = False,
graph_caching: bool = True,
graph_optimizations: bool = True,
modes_to_trace: str = "all",
stateful: Optional[List] = None,
Expand Down Expand Up @@ -157,8 +156,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 Expand Up @@ -186,7 +185,7 @@ def transpile(
def unify(
*objs: Callable,
source: Optional[str] = None,
graph_caching: bool = False,
graph_caching: bool = True,
graph_optimizations: bool = True,
args: Optional[Sequence] = None,
kwargs: Optional[Mapping] = None,
Expand Down

0 comments on commit bbebaea

Please sign in to comment.