jyapayne/llvmpy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add tests for double-precision intrinsics

Browse source
This commit is contained in:
Siu Kwan Lam 2013-01-17 17:12:11 -06:00
commit 7451f6ca4a
1 changed files with 50 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

70
llvm/test_llvmpy.py
View file

@ -916,48 +916,76 @@ tests.append(TestCPUSupport)
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
class TestIntrinsicBasic(unittest.TestCase): class TestIntrinsicBasic(unittest.TestCase):
def _build_module(self): def _build_module(self, float):
mod = Module.new('test') mod = Module.new('test')
float = Type.float()
functy = Type.function(float, [float]) functy = Type.function(float, [float])
func = mod.add_function(functy, "mysin") func = mod.add_function(functy, "mytest%s" % float)
block = func.append_basic_block("entry") block = func.append_basic_block("entry")
b = Builder.new(block) b = Builder.new(block)
return mod, func, b return mod, func, b
def _template(self, mod, func, pyfunc): def _template(self, mod, func, pyfunc):
float = func.type.pointee.return_type
ee = le.ExecutionEngine.new(mod) ee = le.ExecutionEngine.new(mod)
arg = le.GenericValue.real(float, 1.234)
arg = le.GenericValue.real(Type.float(), 1.234)
retval = ee.run_function(func, [arg]) retval = ee.run_function(func, [arg])
golden = pyfunc(1.234) golden = pyfunc(1.234)
answer = retval.as_real(Type.float()) answer = retval.as_real(float)
self.assertTrue(abs(answer - golden) / golden < 1e-7) self.assertTrue(abs(answer - golden) / golden < 1e-7)
def test_sqrt(self): def test_sqrt_f32(self):
mod, func, b = self._build_module() float = Type.float()
intr = Function.intrinsic(mod, lc.INTR_SQRT, [Type.float()]) mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_SQRT, [float])
b.ret(b.call(intr, func.args)) b.ret(b.call(intr, func.args))
self._template(mod, func, math.sqrt) self._template(mod, func, math.sqrt)
def test_cos(self): def test_sqrt_f64(self):
mod, func, b = self._build_module() float = Type.double()
intr = Function.intrinsic(mod, lc.INTR_COS, [Type.float()]) mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_SQRT, [float])
b.ret(b.call(intr, func.args))
self._template(mod, func, math.sqrt)
def test_cos_f32(self):
float = Type.float()
mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_COS, [float])
b.ret(b.call(intr, func.args)) b.ret(b.call(intr, func.args))
self._template(mod, func, math.cos) self._template(mod, func, math.cos)
def test_sin(self): def test_cos_f64(self):
mod, func, b = self._build_module() float = Type.double()
intr = Function.intrinsic(mod, lc.INTR_SIN, [Type.float()]) mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_COS, [float])
b.ret(b.call(intr, func.args))
self._template(mod, func, math.cos)
def test_sin_f32(self):
float = Type.float()
mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_SIN, [float])
b.ret(b.call(intr, func.args)) b.ret(b.call(intr, func.args))
self._template(mod, func, math.sin) self._template(mod, func, math.sin)
def test_powi(self): def test_sin_f64(self):
mod, func, b = self._build_module() float = Type.double()
intr = Function.intrinsic(mod, lc.INTR_POWI, [Type.float()]) mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_SIN, [float])
b.ret(b.call(intr, func.args))
self._template(mod, func, math.sin)
def test_powi_f32(self):
float = Type.float()
mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_POWI, [float])
b.ret(b.call(intr, [func.args[0], lc.Constant.int(Type.int(), 2)]))
self._template(mod, func, lambda x: x**2)
def test_powi_f64(self):
float = Type.double()
mod, func, b = self._build_module(float)
intr = Function.intrinsic(mod, lc.INTR_POWI, [float])
b.ret(b.call(intr, [func.args[0], lc.Constant.int(Type.int(), 2)])) b.ret(b.call(intr, [func.args[0], lc.Constant.int(Type.int(), 2)]))
self._template(mod, func, lambda x: x**2) self._template(mod, func, lambda x: x**2)