Skip to content

Commit ca1aaac

Browse files
eigen-kfacebook-github-bot
authored andcommitted
Fix default value of cat_dim. (#11403)
Summary: This diff fixes the bug introduced by the D75467583, which broke the sharding tests, particulary test_dsp_and_tce_quantized_deepspeech. Reviewed By: hsharma35 Differential Revision: D76057466
1 parent 762333d commit ca1aaac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/cadence/aot/memory_constraints.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def is_cat_along_outermost_dim(
360360
if len(cat_node.args) > 1:
361361
cat_dim = cat_node.args[1]
362362
else:
363-
cat_dim = cat_node.kwargs.get("dim", None)
363+
cat_dim = cat_node.kwargs.get("dim", 0)
364364
if not isinstance(cat_dim, int):
365365
raise ValueError("cat_dim must be an integer.")
366366

0 commit comments

Comments
 (0)