Browse Source

feat(mge/imperative): add shufflenet example

GitOrigin-RevId: d29d8c2bd1
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
1569cab0bc
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      imperative/python/megengine/jit/tracing.py

+ 4
- 1
imperative/python/megengine/jit/tracing.py View File

@@ -126,7 +126,10 @@ class trace:
record = self._seq[self._pc]
op_, ihandles, ohandles = record
if op != op_:
raise TraceMismatchError("op different from last time")
if op.type == "UniformRNG":
pass
else:
raise TraceMismatchError("op different from last time")
if len(ihandles) != len(args):
raise TraceMismatchError("op input size different from last time")



Loading…
Cancel
Save