Skip to content

Commit 2ddc16a

Browse files
committed
Improve mixed dtype logging with op name
1 parent 2c10b33 commit 2ddc16a

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

backends/xnnpack/partition/config/xnnpack_config.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -175,14 +175,22 @@ def _check_inputs_are_valid_dtypes(self, node, valid_dtypes):
175175
if arg_val.dtype not in valid_dtypes:
176176
return False
177177

178-
# Check for mixed dtypes
178+
# Use the first dtype as reference
179179
reference_dtype = reference_dtype or arg_val.dtype
180+
181+
# Check for mixed dtypes
180182
if arg_val.dtype != reference_dtype:
183+
# Get op name if the attribute exists, otherwise use the full node target for logging
184+
op_name = (
185+
node.target.__name__
186+
if hasattr(node.target, "__name__")
187+
else str(node.target)
188+
)
181189
why(
182190
node,
183191
reason=(
184-
f"{node.target} does not support mixed input dtypes. "
185-
f"Got: [{reference_dtype}, {arg_val.dtype}]"
192+
f"{op_name} does not support mixed input dtypes, "
193+
f"got: [{reference_dtype}, {arg_val.dtype}]"
186194
),
187195
)
188196
return False

0 commit comments

Comments
 (0)