Skip to content
This repository has been archived by the owner on Jan 24, 2024. It is now read-only.

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
feifei-111 committed Oct 8, 2023
1 parent 6c54af4 commit 48946cf
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 10 deletions.
3 changes: 1 addition & 2 deletions sot/opcode_translator/executor/function_graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -301,7 +301,7 @@ def start_compile(self, *ret_vars: VariableBase):
found = False
for variable in self.input_variables:
if (
isinstance(variable, (TensorVariable, PaddleLayerVariable))
isinstance(variable, TensorVariable)
and variable.get_symbol().name == name
):
variable.tracker.gen_instructions(self.pycode_gen)
Expand Down Expand Up @@ -426,7 +426,6 @@ def call_layer(
"""

def infer_meta_fn(layer, *metas, **kwmetas):
metas = metas[1:]
metas = LayerInferMetaCache()(layer.value, *metas, **kwmetas)
return metas

Expand Down
8 changes: 0 additions & 8 deletions sot/opcode_translator/executor/variables/callable.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,10 +9,8 @@
import paddle

from .... import psdb
from ....symbolic.statement_ir import Symbol
from ....utils import (
EventGuard,
NameGenerator,
is_break_graph_api,
is_break_graph_tensor_methods,
is_builtin_fn,
Expand Down Expand Up @@ -503,16 +501,10 @@ class PaddleLayerVariable(LayerVariable):
tracker(Tracker): The Tracker object that tracks the information of this variable.
"""

layer_name_generator = NameGenerator("layer_")

def __init__(
self, layer: paddle.nn.Layer, graph: FunctionGraph, tracker: Tracker
):
super().__init__(layer, graph, tracker)
self.name = self.layer_name_generator.next()

def get_symbol(self) -> Symbol:
return Symbol(self.name)

def call_function(self, /, *args, **kwargs):
return self.graph.call_layer(self, *args, **kwargs)
Expand Down

0 comments on commit 48946cf

Please sign in to comment.