diff --git a/dipu/tests/python/individual_scripts/test_op_benchmark.py b/dipu/tests/python/individual_scripts/test_op_benchmark.py index be5901f23..90a255556 100644 --- a/dipu/tests/python/individual_scripts/test_op_benchmark.py +++ b/dipu/tests/python/individual_scripts/test_op_benchmark.py @@ -47,7 +47,7 @@ def batched_dot_bmm(a, b): print(r1) # TODO(fandaoyi,lljbash): find out why it gets slower # assert r0.mean < 8.8e-5 -assert r1.mean < 40.0e-5 +assert r1.mean < 4.0e-3 # Compare takes a list of measurements which we'll save in results. diff --git a/dipu/third_party/DIOPI b/dipu/third_party/DIOPI index 7233d54b5..28bbbf47a 160000 --- a/dipu/third_party/DIOPI +++ b/dipu/third_party/DIOPI @@ -1 +1 @@ -Subproject commit 7233d54b5ff3a2ba779708ea5e914aee9c5f4b7f +Subproject commit 28bbbf47a64a6f9c9620221504a1b36cb712d02c