mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

change seperator of function marker (#499)

pull/495/head
RunningLeon 2022-05-20 15:20:44 +08:00 committed by GitHub
parent e057b87fd1
commit a4de9f3704
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmdeploy/core/optimizers/function_marker.py
View File

@ -154,7 +154,7 @@ def mark_tensors(xs: Any, func: str, func_id: int, io_type: str, ctx: Any,
if ys not in visit: if ys not in visit:
visit.add(ys) visit.add(ys)
root = ctx.names[ctx.index] root = ctx.names[ctx.index]
name = '/'.join(str(x) for x in (root, *prefix)) name = '.'.join(str(x) for x in (root, *prefix))
ys_shape = tuple(int(s) for s in ys.shape) ys_shape = tuple(int(s) for s in ys.shape)
ret = Mark.apply(ys, ys.dtype, ys_shape, func, func_id, ret = Mark.apply(ys, ys.dtype, ys_shape, func, func_id,
io_type, name, index, attrs) io_type, name, index, attrs)