Skip to content

Commit 4831cf1

Browse files
committed
fix missing tensor in stack info for shape mismatch
1 parent 923cec5 commit 4831cf1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tripy/nvtripy/backend/api/executable.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -297,7 +297,8 @@ def extract_recursive(value, name_prefix, allowed_names):
297297
f"Unexpected tensor rank.",
298298
[
299299
f"For tensor: `{arg_name}`, expected a rank of: {len(expected_bounds.min)} but got: {len(shape)}.\n"
300-
f"Note: The provided argument was a tensor with shape: {shape}",
300+
f"Note: The provided argument was: ",
301+
tensor,
301302
],
302303
)
303304

@@ -308,7 +309,8 @@ def extract_recursive(value, name_prefix, allowed_names):
308309
[
309310
f"For tensor: `{arg_name}`, expected a shape within the bounds: min={expected_bounds.min}, max={expected_bounds.max}, but got: {shape}.\n"
310311
f"Dimension {i} has a shape of {shape[i]}, which is not within the expected bounds of [{expected_bounds.min[i]}, {expected_bounds.max[i]}].\n"
311-
f"Note: The provided argument was a tensor with shape: {shape}",
312+
f"Note: The provided argument was: ",
313+
tensor,
312314
],
313315
)
314316
raise_error(str(err))

0 commit comments

Comments
 (0)