ruby-llvm-next 10.0.0

data/test/double_test.rb

@@ -0,0 +1,34 @@
+require "test_helper"
+
+class DoubleTestCase < Minitest::Test
+  def setup
+    LLVM.init_jit
+  end
+
+  def test_double
+    mod = LLVM::Module.new("Double Test")
+    mod.functions.add(:sin, [LLVM::Double], LLVM::Double)
+
+    builder = LLVM::Builder.new
+
+    mod.functions.add('test', [LLVM::Double], LLVM::Double) do |fun, p0|
+      p0.name = 'x'
+
+      bb = fun.basic_blocks.append
+      builder.position_at_end(bb)
+
+      builder.ret(builder.fadd(p0, LLVM::Double(1.0)))
+    end
+
+    engine = LLVM::MCJITCompiler.new(mod)
+
+    arg = 5.0
+    result = engine.run_function(mod.functions["test"], arg)
+    assert_equal arg + 1, result.to_f(LLVM::Double)
+
+    # TODO: fix this
+    #    assert_in_delta(Math.sin(1.0),
+    #      engine.run_function(mod.functions["sin"], 1.0).to_f(LLVM::Double),
+    #      1e-10)
+  end
+end

data/test/equality_test.rb

@@ -0,0 +1,89 @@
+require "test_helper"
+require "llvm/core"
+
+class EqualityTestCase < Minitest::Test
+  def setup
+    LLVM.init_jit
+  end
+
+  class MyModule < LLVM::Module; end
+  class MyInt < LLVM::Int32; end
+  class MyType < LLVM::Type; end
+  class MyFunction < LLVM::Function; end
+
+  def assert_equalities(options)
+    map = {
+      :equal     => method(:assert_equal),
+      :not_equal => lambda {|n, m, name| assert n != m, name },
+      :same      => method(:assert_same),
+      :not_same  => lambda {|n, m, name| assert !n.equal?(m), name },
+      :eql       => lambda {|n, m, name| assert n.eql?(m), name  },
+      :not_eql   => lambda {|n, m, name| assert !n.eql?(m), name },
+    }
+
+    map.each do |name, callable|
+      options[name].combination(2).each do |n, m|
+        callable.call(n, m, name.to_s)
+      end
+    end
+  end
+
+  def test_int_value
+    int1 = LLVM::Int32.from_i(1)
+    int2 = LLVM::Int32.from_ptr(int1.to_ptr)
+    int3 = LLVM::Int32.from_i(2)
+    int4 = MyInt.from_ptr(int1.to_ptr)
+
+    assert_equalities :equal     => [int1, int2, int4],
+                      :not_equal => [int1, int3],
+                      :same      => [int1, int1],
+                      :not_same  => [int1, int2, int3, int4],
+                      :eql       => [int1, int2],
+                      :not_eql   => [int1, int3]
+  end
+
+  def test_module
+    mod1 = LLVM::Module.new('test')
+    mod2 = LLVM::Module.from_ptr(mod1.to_ptr)
+    mod3 = LLVM::Module.new('dummy')
+    mod4 = MyModule.from_ptr(mod1.to_ptr)
+
+    assert_equalities :equal     => [mod1, mod2, mod4],
+                      :not_equal => [mod1, mod3],
+                      :same      => [mod1, mod1],
+                      :not_same  => [mod1, mod2, mod3, mod4],
+                      :eql       => [mod1, mod2],
+                      :not_eql   => [mod1, mod3]
+  end
+
+  def test_type
+    type1 = LLVM::Float.type
+    type2 = LLVM::Type.from_ptr(type1.to_ptr, nil)
+    type3 = LLVM::Double.type
+    type4 = MyType.from_ptr(type1.to_ptr, :mytype)
+
+    assert_equalities :equal     => [type1, type2, type4],
+                      :not_equal => [type1, type3],
+                      :same      => [type1, type1],
+                      :not_same  => [type1, type2, type3, type4],
+                      :eql       => [type1, type2],
+                      :not_eql   => [type1, type3]
+  end
+
+  def test_function
+    mod = LLVM::Module.new('test')
+
+    fn1 = mod.functions.add('test1', [], LLVM.Void)
+    fn2 = LLVM::Function.from_ptr(fn1.to_ptr)
+    fn3 = mod.functions.add('test2', [], LLVM.Void)
+    fn4 = MyFunction.from_ptr(fn1.to_ptr)
+
+    assert_equalities :equal     => [fn1, fn2, fn4],
+                      :not_equal => [fn1, fn3],
+                      :same      => [fn1, fn1],
+                      :not_same  => [fn1, fn2, fn3, fn4],
+                      :eql       => [fn1, fn2],
+                      :not_eql   => [fn1, fn3]
+  end
+
+end

data/test/function_test.rb

@@ -0,0 +1,100 @@
+require "test_helper"
+require "llvm/core"
+
+
+def with_function(arguments, retty, &block)
+  mod = LLVM::Module.new('test')
+  fun = mod.functions.add('fun', arguments, retty)
+  block.yield(fun)
+  mod.dispose
+end
+
+class FunctionTest < Minitest::Test
+
+  def test_type
+    with_function [], LLVM.Void do |fun|
+      type = fun.type
+
+      assert_instance_of LLVM::Type, type
+      assert_equal :pointer,  type.kind
+      assert_equal :function, type.element_type.kind
+    end
+  end
+
+  def test_function_type
+    with_function [], LLVM.Void do |fun|
+      type = fun.function_type
+
+      assert_instance_of LLVM::FunctionType, type
+      assert_equal :function, type.kind
+    end
+  end
+
+  def helper_test_attribute(name)
+    with_function [], LLVM.Void do |fun|
+      assert_equal 0, fun.attribute_count
+      assert_equal [], fun.attributes
+
+      fun.add_attribute(name)
+      assert_equal 1, fun.attribute_count
+      assert_equal [33], fun.attributes
+
+      assert_equal false, fun.verify
+
+      fun.remove_attribute(name)
+      assert_equal 0, fun.attribute_count
+      assert_equal [], fun.attributes
+
+      assert_equal false, fun.verify
+    end
+  end
+
+  def test_add_attribute_old_name
+    helper_test_attribute(:no_unwind_attribute)
+  end
+
+  def test_add_attribute_new_name
+    helper_test_attribute(:nounwind)
+  end
+
+end
+
+class FunctionTypeTest < Minitest::Test
+
+  def test_return_type
+    with_function [], LLVM.Void do |fun|
+      type = fun.function_type
+      assert_equal LLVM.Void, type.return_type
+    end
+
+    with_function [], LLVM::Int32 do |fun|
+      retty = fun.function_type.return_type
+
+      assert_kind_of LLVM::IntType, retty
+      assert_equal 32, retty.width
+    end
+  end
+
+  def test_argument_types
+    with_function [], LLVM.Void do |fun|
+      types = fun.function_type.argument_types
+      assert_equal [], types
+    end
+
+    with_function [LLVM::Int32], LLVM.Void do |fun|
+      types = fun.function_type.argument_types
+      assert_equal 1, types.size
+      a1 = types[0]
+      assert_kind_of LLVM::IntType, a1
+      assert_equal 32, a1.width
+    end
+  end
+
+  def test_vararg
+    with_function [], LLVM.Void do |fun|
+      type = fun.function_type
+      assert !type.vararg?, 'should be false'
+    end
+  end
+
+end

data/test/generic_value_test.rb

@@ -0,0 +1,22 @@
+require "test_helper"
+
+class GenericValueTestCase < Minitest::Test
+
+  def setup
+    LLVM.init_jit
+  end
+
+  def test_from_i
+    assert_equal 2, LLVM::GenericValue.from_i(2).to_i
+    assert_equal 2, LLVM::GenericValue.from_i(2.2).to_i
+  end
+
+  def test_from_float
+    assert_in_delta 2.2, LLVM::GenericValue.from_f(2.2).to_f, 1e-6
+  end
+
+  def test_from_double
+    assert_in_delta 2.2, LLVM::GenericValue.from_d(2.2).to_f(LLVM::Double), 1e-6
+  end
+
+end

data/test/instruction_test.rb

@@ -0,0 +1,30 @@
+require "test_helper"
+
+class InstructionTestCase < Minitest::Test
+  def setup
+    LLVM.init_jit
+    @module = LLVM::Module.new("InstructionTestCase")
+  end
+
+  def test_instruction
+    fn = @module.functions.add("test_instruction", [LLVM::Double], LLVM::Double) do |fn, arg|
+      fn.basic_blocks.append.build do |builder|
+        builder.ret(
+          builder.fadd(arg, LLVM.Double(3.0)))
+      end
+    end
+
+    entry = fn.basic_blocks.entry
+
+    inst1 = entry.instructions.first
+    inst2 = entry.instructions.last
+
+    assert_kind_of LLVM::Instruction, inst1
+    assert_kind_of LLVM::Instruction, inst2
+
+    assert_equal inst2, inst1.next
+    assert_equal inst1, inst2.previous
+    assert_equal entry, inst1.parent
+    assert_equal entry, inst2.parent
+  end
+end

data/test/ipo_test.rb

@@ -0,0 +1,53 @@
+require "test_helper"
+require "llvm/core"
+require 'llvm/transforms/ipo'
+require 'llvm/core/pass_manager'
+
+class IPOTestCase < Minitest::Test
+
+  def setup
+    LLVM.init_jit
+  end
+
+  def test_gdce
+    mod = LLVM::Module.new('test')
+
+    fn1 = mod.functions.add("fn1", [], LLVM.Void) do |fn|
+      fn.linkage = :internal
+      fn.basic_blocks.append.build do |builder|
+        builder.ret_void
+      end
+    end
+
+    fn2 = mod.functions.add("fn2", [], LLVM.Void) do |fn|
+      fn.linkage = :internal
+      fn.basic_blocks.append.build do |builder|
+        builder.ret_void
+      end
+    end
+
+    main = mod.functions.add("main", [], LLVM.Void) do |fn|
+      fn.basic_blocks.append.build do |builder|
+        builder.call(fn1)
+        builder.ret_void
+      end
+    end
+
+    fns = mod.functions.to_a
+    assert fns.include?(fn1)
+    assert fns.include?(fn2)
+    assert fns.include?(main)
+
+    # optimize
+    engine = LLVM::MCJITCompiler.new(mod)
+    passm  = LLVM::PassManager.new(engine)
+
+    passm.gdce!
+    passm.run(mod)
+
+    fns = mod.functions.to_a
+    assert fns.include?(fn1)
+    assert !fns.include?(fn2), 'fn2 should be eliminated'
+    assert fns.include?(main)
+  end
+end

data/test/linker_test.rb

@@ -0,0 +1,37 @@
+require 'test_helper'
+require 'llvm/core'
+require 'llvm/linker'
+
+class LinkerTestCase < Minitest::Test
+  def setup
+    LLVM.init_jit
+  end
+
+  def create_modules
+    @mod1 = define_module('mod1') do |mod|
+      mod1calc = mod.functions.add("calc", [], LLVM::Int32)
+
+      mod.functions.add("main", [], LLVM::Int32) do |fn|
+        fn.basic_blocks.append.build do |builder|
+          val = builder.call(mod1calc)
+          builder.ret val
+        end
+      end
+    end
+
+    @mod2 = define_module('mod2') do |mod|
+      mod.functions.add("calc", [], LLVM::Int32) do |fn|
+        fn.basic_blocks.append.build do |builder|
+          builder.ret LLVM::Int32.from_i(42)
+        end
+      end
+    end
+  end
+
+  def test_link_into
+    create_modules
+    @mod2.link_into(@mod1)
+
+    assert_equal 42, run_function_on_module(@mod1, 'main').to_i
+  end
+end

data/test/mcjit_test.rb

@@ -0,0 +1,94 @@
+require 'test_helper'
+
+class MCJITTestCase < Minitest::Test
+  def setup
+    LLVM.init_jit(true)
+  end
+
+  def create_square_function_module
+    LLVM::Module.new('square').tap do |mod|
+      mod.functions.add(:square, [LLVM::Int], LLVM::Int) do |fun, x|
+        fun.basic_blocks.append.build do |builder|
+          n = builder.mul(x, x)
+          builder.ret(n)
+        end
+      end
+
+      mod.verify!
+    end
+  end
+
+  def test_simple_function
+    mod = create_square_function_module
+
+    engine = LLVM::MCJITCompiler.new(mod, :opt_level => 0)
+
+    result = engine.run_function(mod.functions['square'], 5)
+    assert_equal 25, result.to_i
+  end
+
+  def test_functions_named
+    mod = LLVM::Module.new('foo').tap do |mod|
+      mod.functions.add(:foo, [], LLVM::Int)
+      mod.verify!
+    end
+
+    engine = LLVM::MCJITCompiler.new(mod, :opt_level => 0)
+
+    ['foo', :foo].each do |name|
+      engine.functions[name].tap do |fun|
+        assert fun, "function named #{name.inspect}"
+        assert_equal 'foo', fun.name
+      end
+    end
+  end
+
+  def test_add_module
+    main_mod = LLVM::Module.new('main')
+
+    main_mod.functions.add(:square, [LLVM::Int], LLVM::Int) do |square|
+      main_mod.functions.add(:call_square, [], LLVM::Int) do |call_square|
+        call_square.basic_blocks.append.build do |builder|
+          n = builder.call(square, LLVM::Int(5))
+          builder.ret(n)
+        end
+      end
+    end
+
+    main_mod.verify!
+
+    engine = LLVM::MCJITCompiler.new(main_mod, :opt_level => 0)
+    engine.modules << create_square_function_module
+
+    result = engine.run_function(main_mod.functions['call_square'])
+    assert_equal 25, result.to_i
+  end
+
+  def test_remove_module
+    mod1 = LLVM::Module.new('foo')
+    mod2 = LLVM::Module.new('bar')
+
+    foo = mod1.functions.add(:foo, [], LLVM::Int)
+    bar = mod2.functions.add(:bar, [], LLVM::Int)
+
+    engine = LLVM::MCJITCompiler.new(mod1, :opt_level => 0)
+
+    (engine.modules << mod2).tap do |ret|
+      assert_equal engine.modules, ret, '#<< returns self'
+    end
+
+    refute_nil engine.functions[:bar]
+    engine.modules.delete(mod2).tap do |ret|
+      assert_instance_of LLVM::Module, ret, '#delete returns module'
+      assert_equal mod2, ret
+    end
+    assert_nil engine.functions[:bar]
+  end
+
+  def test_accessors
+    main_mod = LLVM::Module.new('main')
+    engine = LLVM::MCJITCompiler.new(main_mod, :opt_level => 0)
+    assert_match(/^e-/, engine.data_layout.to_s)
+    assert_match(/gnu/, engine.target_machine.triple)
+  end
+end