14
14
from executorch .backends .arm .operators .node_visitor import NodeVisitor
15
15
from executorch .backends .arm .tosa_mapping import TosaArg
16
16
from executorch .backends .arm .tosa_specification import TosaSpecification
17
- from executorch .backends .arm .tosa_utils import (
18
- get_node_debug_info ,
19
- getNodeArgs ,
20
- tosa_shape ,
21
- )
17
+ from executorch .backends .arm .tosa_utils import getNodeArgs , tosa_shape
22
18
from torch .export .exported_program import ExportedProgram
23
19
24
20
@@ -36,7 +32,7 @@ def process_call_function(
36
32
output = TosaArg (node )
37
33
except ValueError as e :
38
34
raise ValueError (
39
- f"Failed processing call_function:\n { get_node_debug_info ( node ) } "
35
+ f"Failed processing call_function: { node . name } . "
40
36
"Is the original torch function supported?"
41
37
) from e
42
38
tosa_graph .currRegion .currBasicBlock .addTensor (
@@ -74,7 +70,7 @@ def process_inputs(
74
70
tosa_arg = TosaArg (node )
75
71
except ValueError as e :
76
72
raise ValueError (
77
- f"Failed processing input placeholder:\n { get_node_debug_info ( node ) } "
73
+ f"Failed processing input placeholder: { node . name } . "
78
74
"Is the original torch function supported?"
79
75
) from e
80
76
input_shape = tosa_arg .shape
@@ -100,7 +96,7 @@ def process_inputs_to_parameters(
100
96
tosa_arg = TosaArg (node )
101
97
except ValueError as e :
102
98
raise ValueError (
103
- f"Failed processing parameter placeholder:\n { get_node_debug_info ( node ) } "
99
+ f"Failed processing parameter placeholder: { node . name } . "
104
100
"Is the original torch function supported?"
105
101
) from e
106
102
parameter_name = edge_program .graph_signature .inputs_to_parameters [tosa_arg .name ]
@@ -129,7 +125,7 @@ def process_inputs_to_buffers(
129
125
tosa_arg = TosaArg (node )
130
126
except ValueError as e :
131
127
raise ValueError (
132
- f"Failed processing buffer placeholder:\n { get_node_debug_info ( node ) } "
128
+ f"Failed processing buffer placeholder: { node . name } . "
133
129
"Is the original torch function supported?"
134
130
) from e
135
131
buffer_name = edge_program .graph_signature .inputs_to_buffers [node .name ]
@@ -157,7 +153,7 @@ def process_inputs_to_lifted_tensor_constants(
157
153
tosa_arg = TosaArg (node )
158
154
except ValueError as e :
159
155
raise ValueError (
160
- f"Failed processing lifted tensor constant placeholder:\n { get_node_debug_info ( node ) } "
156
+ f"Failed processing lifted tensor constant placeholder: { node . name } . "
161
157
"Is the original torch function supported?"
162
158
) from e
163
159
tensor_name = edge_program .graph_signature .inputs_to_lifted_tensor_constants [
0 commit comments