Browse Source

fix(mge/imperative): remove codes newly introduced in numpy 1.19

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

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

@@ -155,9 +155,7 @@ class trace:
"last time, got an internal tensor this time" "last time, got an internal tensor this time"
) )
if x._handle != info.bound_data._handle: if x._handle != info.bound_data._handle:
if not np.array_equal(
x.numpy(), info.bound_data.numpy(), equal_nan=True
):
if not np.array_equal(x.numpy(), info.bound_data.numpy()):
raise TraceMismatchError( raise TraceMismatchError(
"const capture violated: got " "const capture violated: got "
"a different tensor this time" "a different tensor this time"


Loading…
Cancel
Save