We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 50f7376 commit e163575Copy full SHA for e163575
backends/cadence/aot/memory_constraints.py
@@ -360,7 +360,7 @@ def is_cat_along_outermost_dim(
360
if len(cat_node.args) > 1:
361
cat_dim = cat_node.args[1]
362
else:
363
- cat_dim = cat_node.kwargs.get("dim", None)
+ cat_dim = cat_node.kwargs.get("dim", 0)
364
if not isinstance(cat_dim, int):
365
raise ValueError("cat_dim must be an integer.")
366
0 commit comments