lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

allow the input of `concat` to be tuple`

This commit is contained in:
CHEN Xinsheng 2025-02-18 20:56:57 +08:00
parent a5c37bd58e
commit 7b142f51a6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/extern/acl/aclops/concat_op.py vendored
View File

@ -59,7 +59,7 @@ class ConcatACL(jt.Function):
super(ConcatACL, self).__init__() super(ConcatACL, self).__init__()
def __call__(self, *args): def __call__(self, *args):
assert isinstance(args[0], list) assert isinstance(args[0], (list, tuple))
assert isinstance(args[1], int) assert isinstance(args[1], int)
if jt.flags.no_grad: if jt.flags.no_grad:
return self.execute(*args) return self.execute(*args)