summaryrefslogtreecommitdiff
path: root/grpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py
diff options
context:
space:
mode:
Diffstat (limited to 'grpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py')
-rwxr-xr-xgrpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py33
1 files changed, 21 insertions, 12 deletions
diff --git a/grpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py b/grpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py
index 2a770403..1e474cb4 100755
--- a/grpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py
+++ b/grpc/tools/profiling/microbenchmarks/bm_diff/bm_speedup.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python2.7
+#!/usr/bin/env python3
#
# Copyright 2017 gRPC authors.
#
@@ -14,9 +14,10 @@
# See the License for the specific language governing permissions and
# limitations under the License.
-from scipy import stats
import math
+from scipy import stats
+
_DEFAULT_THRESHOLD = 1e-10
@@ -29,25 +30,33 @@ def cmp(a, b):
def speedup(new, old, threshold=_DEFAULT_THRESHOLD):
- if (len(set(new))) == 1 and new == old: return 0
+ if (len(set(new))) == 1 and new == old:
+ return 0
s0, p0 = cmp(new, old)
- if math.isnan(p0): return 0
- if s0 == 0: return 0
- if p0 > threshold: return 0
+ if math.isnan(p0):
+ return 0
+ if s0 == 0:
+ return 0
+ if p0 > threshold:
+ return 0
if s0 < 0:
pct = 1
while pct < 100:
sp, pp = cmp(new, scale(old, 1 - pct / 100.0))
- if sp > 0: break
- if pp > threshold: break
+ if sp > 0:
+ break
+ if pp > threshold:
+ break
pct += 1
return -(pct - 1)
else:
pct = 1
while pct < 10000:
sp, pp = cmp(new, scale(old, 1 + pct / 100.0))
- if sp < 0: break
- if pp > threshold: break
+ if sp < 0:
+ break
+ if pp > threshold:
+ break
pct += 1
return pct - 1
@@ -55,5 +64,5 @@ def speedup(new, old, threshold=_DEFAULT_THRESHOLD):
if __name__ == "__main__":
new = [0.0, 0.0, 0.0, 0.0]
old = [2.96608e-06, 3.35076e-06, 3.45384e-06, 3.34407e-06]
- print speedup(new, old, 1e-5)
- print speedup(old, new, 1e-5)
+ print(speedup(new, old, 1e-5))
+ print(speedup(old, new, 1e-5))