evoasm 0.0.2.pre7

data/ext/evoasm_ext/evoasm-search.h

@@ -0,0 +1,214 @@
+#pragma once
+
+#include <stdint.h>
+#include <stdlib.h>
+
+#include "evoasm.h"
+#include "evoasm-buf.h"
+#include "evoasm-x64.h"
+
+typedef double evoasm_loss;
+typedef uint8_t evoasm_program_size;
+
+#define EVOASM_KERNEL_SIZE_MAX UINT8_MAX
+typedef uint8_t evoasm_kernel_size;
+#define EVOASM_KERNEL_MAX_SIZE (EVOASM_KERNEL_SIZE_MAX - 1)
+
+typedef enum {
+  EVOASM_EXAMPLE_TYPE_I64,
+  EVOASM_EXAMPLE_TYPE_U64,
+  EVOASM_EXAMPLE_TYPE_F64,
+} evoasm_example_type;
+
+typedef struct {
+  union {
+    double f64;
+    int64_t i64;
+    uint64_t u64;
+  };
+} evoasm_example_val;
+
+#define EVOASM_PROGRAM_IO_MAX_ARITY 8
+
+typedef struct {
+  uint8_t arity;
+  uint16_t len;
+  evoasm_example_val *vals;
+  evoasm_example_type types[EVOASM_PROGRAM_IO_MAX_ARITY];
+} evoasm_program_io;
+
+#define EVOASM_PROGRAM_OUTPUT_MAX_ARITY EVOASM_PROGRAM_IO_MAX_ARITY
+#define EVOASM_PROGRAM_INPUT_MAX_ARITY EVOASM_PROGRAM_IO_MAX_ARITY
+typedef evoasm_program_io evoasm_program_output;
+typedef evoasm_program_io evoasm_program_input;
+
+#define EVOASM_PROGRAM_IO_N(program_io) ((uint16_t)((program_io)->len / (program_io)->arity))
+#define EVOASM_PROGRAM_INPUT_N(program_input) EVOASM_PROGRAM_IO_N((evoasm_program_io *)program_input)
+#define EVOASM_PROGRAM_OUTPUT_N(program_output) EVOASM_PROGRAM_IO_N((evoasm_program_io *)program_output)
+
+typedef struct {
+  evoasm_inst *inst;
+  evoasm_arch_params_bitmap set_params;
+  evoasm_arch_param_val param_vals[EVOASM_ARCH_MAX_PARAMS];
+} evoasm_kernel_param;
+
+typedef struct {
+  evoasm_kernel_size size;
+  /* kernel executed next (jumped to)
+   * Kernel terminates if EVOASM_KERNEL_SIZE_MAX 
+   */
+  evoasm_kernel_size branch_kernel_idx;
+  uint8_t jmp_selector;
+  evoasm_kernel_param params[];
+} evoasm_kernel_params;
+
+typedef struct {
+  evoasm_program_size size;
+} evoasm_program_params;
+
+#define EVOASM_KERNEL_MAX_OUTPUT_REGS 254
+#define EVOASM_KERNEL_MAX_INPUT_REGS 254
+#define EVOASM_PROGRAM_MAX_SIZE 64
+
+#define EVOASM_KERNEL_REG_INFO_N_TRANS_REGS 2
+
+typedef struct {
+  bool input : 1;
+  bool written : 1;
+  bool output : 1;
+  evoasm_x64_reg_id trans_regs[EVOASM_KERNEL_REG_INFO_N_TRANS_REGS];
+} evoasm_kernel_x64_reg_info;
+
+typedef union {
+  evoasm_kernel_x64_reg_info x64[EVOASM_X64_N_REGS];  
+} evoasm_kernel_reg_info;
+
+
+typedef struct {
+  evoasm_kernel_params *params;
+  evoasm_kernel_reg_info reg_info;
+  
+  union {
+    evoasm_x64_reg_id x64[EVOASM_KERNEL_MAX_OUTPUT_REGS];
+  } output_regs;
+
+  uint_fast8_t n_input_regs;
+  uint_fast8_t n_output_regs;
+  uint8_t idx;
+} evoasm_kernel;
+
+typedef struct {
+  evoasm_inst **insts;
+  evoasm_program_size min_program_size;
+  evoasm_program_size max_program_size;
+  evoasm_kernel_size min_kernel_size;
+  evoasm_kernel_size max_kernel_size;
+  uint32_t recur_limit;
+  uint16_t insts_len;
+  uint8_t params_len;
+  uint32_t pop_size;
+  uint32_t mutation_rate;
+  evoasm_program_input program_input;
+  evoasm_program_output program_output;
+  evoasm_arch_param_id *params;
+  evoasm_prng64_seed seed64;
+  evoasm_prng32_seed seed32;
+  evoasm_domain *domains[EVOASM_ARCH_MAX_PARAMS];
+
+} evoasm_search_params;
+
+
+typedef struct {
+  evoasm_arch *arch;
+  evoasm_buf *buf;
+  evoasm_buf *body_buf;
+  uint32_t index;
+  uint8_t in_arity;
+  uint8_t out_arity;
+  bool reset_rflags : 1;
+  bool need_emit    : 1;
+  void *_signal_ctx;
+  uint32_t exception_mask;
+  evoasm_example_type types[EVOASM_PROGRAM_OUTPUT_MAX_ARITY];
+  evoasm_example_val *output_vals;
+  evoasm_kernel kernels[EVOASM_PROGRAM_MAX_SIZE];
+  uint32_t recur_counters[EVOASM_PROGRAM_MAX_SIZE];
+  evoasm_program_params *params;
+  evoasm_program_input _input;
+  evoasm_program_output _output;
+  evoasm_search_params *search_params;
+  evoasm_reg_id output_regs[EVOASM_PROGRAM_IO_MAX_ARITY];
+  
+  union {
+    /* register at index i has input i % input_arity */
+    uint8_t x64[EVOASM_X64_N_REGS];
+  } reg_inputs;
+
+} evoasm_program;
+
+#define EVOASM_SEARCH_ELITE_SIZE 4
+
+typedef struct {
+  evoasm_prng64 prng64;
+  evoasm_prng32 prng32;
+  evoasm_loss best_loss;
+  evoasm_buf buf;
+  evoasm_buf body_buf;
+
+  uint32_t elite[EVOASM_SEARCH_ELITE_SIZE];
+  uint8_t elite_pos;
+  uint_fast8_t *matching;
+  evoasm_example_val *output_vals;
+  evoasm_loss *losses;
+  unsigned char *programs;
+  unsigned char *programs_main;
+  unsigned char *programs_swap;
+  unsigned char *programs_aux;
+} evoasm_population;
+
+#define EVOASM_EXAMPLES_MAX_ARITY 8
+typedef struct {
+  evoasm_example_type types[EVOASM_EXAMPLES_MAX_ARITY];
+  uint16_t len;
+  evoasm_example_val *vals;
+  uint8_t in_arity;
+  uint8_t out_arity;
+} evoasm_examples;
+
+typedef struct {
+  evoasm_arch *arch;
+  evoasm_population pop;
+  evoasm_search_params params;
+  evoasm_domain *domains;
+} evoasm_search;
+
+bool
+evoasm_search_init(evoasm_search *search,
+                   evoasm_arch *arch, evoasm_search_params *params);
+
+bool
+evoasm_search_destroy(evoasm_search *search);
+
+typedef bool (*evoasm_search_result_func)(evoasm_program *program,
+                                         evoasm_loss loss, void *user_data);
+
+void
+evoasm_search_start(evoasm_search *search, evoasm_loss max_loss,
+                   evoasm_search_result_func func, void *user_data);
+
+bool
+evoasm_program_run(evoasm_program *program,
+                  evoasm_program_input *input,
+                  evoasm_program_output *output);
+
+void
+evoasm_program_io_destroy(evoasm_program_io *program_io);
+
+evoasm_success
+evoasm_program_eliminate_introns(evoasm_program *program);
+
+#define evoasm_program_output_destroy(program_output) \
+  evoasm_program_io_destroy((evoasm_program_io *)program_output)
+
+void
+evoasm_program_output_regs(evoasm_program *program, evoasm_reg_id *output_regs, size_t *len);

data/ext/evoasm_ext/evoasm-util.h

@@ -0,0 +1,40 @@
+#pragma once
+
+#define EVOASM_INT8_MAX  INT8_MAX
+#define EVOASM_INT8_MIN  INT8_MIN
+#define EVOASM_INT16_MAX INT16_MAX
+#define EVOASM_INT16_MIN INT16_MIN
+#define EVOASM_INT32_MAX INT32_MAX
+#define EVOASM_INT32_MIN INT32_MIN
+#define EVOASM_INT64_MAX INT64_MAX
+#define EVOASM_INT64_MIN INT64_MIN
+
+
+#define EVOASM_MAX(a,b) (((a) > (b)) ? (a) : (b))
+#define EVOASM_MIN(a,b) (((a) < (b)) ? (a) : (b))
+#define EVOASM_CLAMP(x, min, max) (((x) > (max)) ? (max) : (((x) < (min)) ? (min) : (x)))
+
+#define EVOASM_ALIGN_DOWN(s, a) ((s) &~ ((a) - 1))
+#define EVOASM_ALIGN_UP(s, a) EVOASM_ALIGN_DOWN(((s) + (a) - 1), a)
+
+#define EVOASM_ARY_LEN(ary) (sizeof(ary) / sizeof(ary[0]))
+
+#ifdef __GNUC__
+# define EVOASM_UNLIKELY(e) (__builtin_expect(e, 0))
+# define EVOASM_LIKELY(e) (__builtin_expect(e, 1))
+# define evoasm_used __attribute__((used))
+# define evoasm_force_inline __attribute__((always_inline))
+#else
+# define EVOASM_UNLIKELY(e) (e)
+# define EVOASM_LIKELY(e) (e)
+# define evoasm_used
+#endif
+
+#if defined(__GNUC__)
+# define evoasm_check_return __attribute__((warn_unused_result))
+#elif defined(_MSC_VER)
+# define evoasm_check_return _Check_return_
+# define evoasm_force_inline __forceinline
+#else
+# define evoasm_check_return
+#endif