grnexus 1.0.2

data/lib/grnexus_core.rb

@@ -0,0 +1,130 @@
+require 'ffi'
+require 'rbconfig'
+
+module GRNEXUSCore
+  extend FFI::Library
+
+  # Detect operating system and load appropriate library
+  def self.detect_library
+    script_dir = File.dirname(File.expand_path(__FILE__))
+    os = RbConfig::CONFIG['host_os']
+
+    case os
+    when /mswin|mingw|cygwin/
+      File.join(script_dir, '..', 'exports', 'Windows', 'grnexus_core.dll')
+    when /darwin/
+      File.join(script_dir, '..', 'exports', 'Mac', 'grnexus_core.dylib')
+    when /linux/
+      File.join(script_dir, '..', 'exports', 'Linux', 'grnexus_core.so')
+    else
+      raise "Sistema operativo no soportado: #{os}"
+    end
+  end
+
+  # Load library
+  ffi_lib detect_library
+
+  # Define GRNexusData structure
+  class GRNexusData < FFI::Struct
+    layout :data,   :pointer,
+           :type,   :int,
+           :size,   :size_t,
+           :stride, :size_t,
+           :dims,   [:size_t, 3]
+  end
+
+  # Attach C functions
+  attach_function :matrix_multiply, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr], :int
+  attach_function :matrix_add, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr], :int
+  attach_function :matrix_transpose, [GRNexusData.ptr, GRNexusData.ptr], :int
+  attach_function :matrix_hadamard, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr], :int
+  attach_function :matrix_scale, [GRNexusData.ptr, :double, GRNexusData.ptr], :int
+  
+  attach_function :dense_forward, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr], :int
+  attach_function :dense_backward, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, :double], :int
+  
+  attach_function :cross_entropy_loss, [GRNexusData.ptr, GRNexusData.ptr], :double
+  attach_function :cross_entropy_gradient, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr], :int
+  attach_function :mse_loss, [GRNexusData.ptr, GRNexusData.ptr], :double
+  attach_function :mse_gradient, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr], :int
+  
+  attach_function :sgd_update, [GRNexusData.ptr, GRNexusData.ptr, :double], :int
+  attach_function :adam_update, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, :double, :double, :double, :double, :int], :int
+  attach_function :rmsprop_update, [GRNexusData.ptr, GRNexusData.ptr, GRNexusData.ptr, :double, :double, :double], :int
+  
+  attach_function :xavier_init, [GRNexusData.ptr, :size_t, :size_t], :int
+  attach_function :he_init, [GRNexusData.ptr, :size_t], :int
+  attach_function :clip_gradients, [GRNexusData.ptr, :double], :int
+
+  # Helper methods to create GRNexusData structures
+  def self.create_matrix_data(matrix)
+    rows = matrix.length
+    cols = matrix[0].length
+    size = rows * cols
+    
+    buffer = FFI::MemoryPointer.new(:double, size)
+    flat_data = matrix.flatten
+    buffer.write_array_of_double(flat_data)
+    
+    data_ptr = FFI::MemoryPointer.new(GRNexusData.size)
+    struct_instance = GRNexusData.new(data_ptr)
+    struct_instance[:data] = buffer
+    struct_instance[:type] = 3  # MATRIX
+    struct_instance[:size] = size
+    struct_instance[:stride] = cols
+    struct_instance[:dims][0] = rows
+    struct_instance[:dims][1] = cols
+    struct_instance[:dims][2] = 0
+    
+    [data_ptr, buffer]
+  end
+
+  def self.create_vector_data(vector)
+    size = vector.length
+    
+    buffer = FFI::MemoryPointer.new(:double, size)
+    buffer.write_array_of_double(vector)
+    
+    data_ptr = FFI::MemoryPointer.new(GRNexusData.size)
+    struct_instance = GRNexusData.new(data_ptr)
+    struct_instance[:data] = buffer
+    struct_instance[:type] = 2  # VECTOR
+    struct_instance[:size] = size
+    struct_instance[:stride] = 1
+    struct_instance[:dims][0] = size
+    struct_instance[:dims][1] = 0
+    struct_instance[:dims][2] = 0
+    
+    [data_ptr, buffer]
+  end
+
+  def self.create_output_matrix_data(rows, cols)
+    size = rows * cols
+    buffer = FFI::MemoryPointer.new(:double, size)
+    
+    data_ptr = FFI::MemoryPointer.new(GRNexusData.size)
+    struct_instance = GRNexusData.new(data_ptr)
+    struct_instance[:data] = buffer
+    struct_instance[:type] = 3  # MATRIX
+    struct_instance[:size] = size
+    struct_instance[:stride] = cols
+    struct_instance[:dims][0] = rows
+    struct_instance[:dims][1] = cols
+    struct_instance[:dims][2] = 0
+    
+    [data_ptr, buffer]
+  end
+
+  def self.read_matrix_data(buffer, rows, cols)
+    data = buffer.read_array_of_double(rows * cols)
+    matrix = []
+    rows.times do |i|
+      matrix << data[i * cols, cols]
+    end
+    matrix
+  end
+
+  def self.read_vector_data(buffer, size)
+    buffer.read_array_of_double(size)
+  end
+end