allow the input of `concat` to be tuple`

2025-02-18 20:56:57 +08:00 · 2025-02-18 20:56:57 +08:00 · cc3b402913
parent e4be9b1f78
commit cc3b402913
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/extern/acl/aclops/concat_op.py
+++ b/python/jittor/extern/acl/aclops/concat_op.py
@ -59,7 +59,7 @@ class ConcatACL(jt.Function):
        super(ConcatACL, self).__init__()

    def __call__(self, *args):
-        assert isinstance(args[0], list)
+        assert isinstance(args[0], (list, tuple))
        assert isinstance(args[1], int)
        if jt.flags.no_grad:
            return self.execute(*args)