You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

63 lines
1.4 KiB

#!/usr/bin/env python
# ULP error plot tool.
#
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
import numpy as np
import matplotlib.pyplot as plt
import sys
import re
# example usage:
# build/bin/ulp -e .0001 log 0.5 2.0 2345678 | math/tools/plot.py
def fhex(s):
return float.fromhex(s)
def parse(f):
xs = []
gs = []
ys = []
es = []
# Has to match the format used in ulp.c
r = re.compile(r'[^ (]+\(([^ )]*)\) got ([^ ]+) want ([^ ]+) [^ ]+ ulp err ([^ ]+)')
for line in f:
m = r.match(line)
if m:
x = fhex(m.group(1))
g = fhex(m.group(2))
y = fhex(m.group(3))
e = float(m.group(4))
xs.append(x)
gs.append(g)
ys.append(y)
es.append(e)
elif line.startswith('PASS') or line.startswith('FAIL'):
# Print the summary line
print(line)
return xs, gs, ys, es
def plot(xs, gs, ys, es):
if len(xs) < 2:
print('not enough samples')
return
a = min(xs)
b = max(xs)
fig, (ax0,ax1) = plt.subplots(nrows=2)
es = np.abs(es) # ignore the sign
emax = max(es)
ax0.text(a+(b-a)*0.7, emax*0.8, '%s\n%g'%(emax.hex(),emax))
ax0.plot(xs,es,'r.')
ax0.grid()
ax1.plot(xs,ys,'r.',label='want')
ax1.plot(xs,gs,'b.',label='got')
ax1.grid()
ax1.legend()
plt.show()
xs, gs, ys, es = parse(sys.stdin)
plot(xs, gs, ys, es)