@@ -33,7 +33,9 @@ class ToCopySupported(SupportedTOSAOperatorCheck):
33
33
34
34
@staticmethod
35
35
def _merge_supported_types (
36
- dtypes1 : SupportedTypeDict , dtypes2 : SupportedTypeDict
36
+ # pyre-ignore[11]
37
+ dtypes1 : SupportedTypeDict ,
38
+ dtypes2 : SupportedTypeDict ,
37
39
) -> SupportedTypeDict :
38
40
merged_dtypes = dtypes1
39
41
for k , v in dtypes2 .items ():
@@ -92,7 +94,7 @@ def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool
92
94
if input_dtype not in supported_dtypes :
93
95
logger .info (
94
96
f"Input dtype { input_val .dtype } is not supported in "
95
- f"{ node .target .name ()} ."
97
+ f"{ node .target .name ()} ." # pyre-ignore[16]
96
98
)
97
99
return False
98
100
@@ -102,7 +104,7 @@ def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool
102
104
if output_val .dtype not in supported_dtypes [input_dtype ]:
103
105
logger .info (
104
106
f"Output dtype { output_val .dtype } is not supported in "
105
- f"{ node .target .name ()} for input dtype { input_dtype } . "
107
+ f"{ node .target .name ()} for input dtype { input_dtype } . " # pyre-ignore[16]
106
108
f"Supported output types: "
107
109
f"{ '' .join (str (t ) for t in supported_dtypes [input_dtype ])} "
108
110
)
@@ -113,7 +115,7 @@ def is_node_supported(self, node: fx.Node, tosa_spec: TosaSpecification) -> bool
113
115
if node .kwargs ["memory_format" ] in (torch .preserve_format ,):
114
116
logger .info (
115
117
f"Argument 'memory_format' is not supported for "
116
- f"{ node .target .name ()} right now."
118
+ f"{ node .target .name ()} right now." # pyre-ignore[16]
117
119
)
118
120
return False
119
121
0 commit comments