cumo 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (266) hide show
  1. checksums.yaml +7 -0
  2. data/.gitignore +27 -0
  3. data/.travis.yml +5 -0
  4. data/3rd_party/mkmf-cu/.gitignore +36 -0
  5. data/3rd_party/mkmf-cu/Gemfile +3 -0
  6. data/3rd_party/mkmf-cu/LICENSE +21 -0
  7. data/3rd_party/mkmf-cu/README.md +36 -0
  8. data/3rd_party/mkmf-cu/Rakefile +11 -0
  9. data/3rd_party/mkmf-cu/bin/mkmf-cu-nvcc +4 -0
  10. data/3rd_party/mkmf-cu/lib/mkmf-cu.rb +32 -0
  11. data/3rd_party/mkmf-cu/lib/mkmf-cu/cli.rb +80 -0
  12. data/3rd_party/mkmf-cu/lib/mkmf-cu/nvcc.rb +157 -0
  13. data/3rd_party/mkmf-cu/mkmf-cu.gemspec +16 -0
  14. data/3rd_party/mkmf-cu/test/test_mkmf-cu.rb +67 -0
  15. data/CODE_OF_CONDUCT.md +46 -0
  16. data/Gemfile +8 -0
  17. data/LICENSE.txt +82 -0
  18. data/README.md +252 -0
  19. data/Rakefile +43 -0
  20. data/bench/broadcast_fp32.rb +138 -0
  21. data/bench/cumo_bench.rb +193 -0
  22. data/bench/numo_bench.rb +138 -0
  23. data/bench/reduction_fp32.rb +117 -0
  24. data/bin/console +14 -0
  25. data/bin/setup +8 -0
  26. data/cumo.gemspec +32 -0
  27. data/ext/cumo/cuda/cublas.c +278 -0
  28. data/ext/cumo/cuda/driver.c +421 -0
  29. data/ext/cumo/cuda/memory_pool.cpp +185 -0
  30. data/ext/cumo/cuda/memory_pool_impl.cpp +308 -0
  31. data/ext/cumo/cuda/memory_pool_impl.hpp +370 -0
  32. data/ext/cumo/cuda/memory_pool_impl_test.cpp +554 -0
  33. data/ext/cumo/cuda/nvrtc.c +207 -0
  34. data/ext/cumo/cuda/runtime.c +167 -0
  35. data/ext/cumo/cumo.c +148 -0
  36. data/ext/cumo/depend.erb +58 -0
  37. data/ext/cumo/extconf.rb +179 -0
  38. data/ext/cumo/include/cumo.h +25 -0
  39. data/ext/cumo/include/cumo/compat.h +23 -0
  40. data/ext/cumo/include/cumo/cuda/cublas.h +153 -0
  41. data/ext/cumo/include/cumo/cuda/cumo_thrust.hpp +187 -0
  42. data/ext/cumo/include/cumo/cuda/cumo_thrust_complex.hpp +79 -0
  43. data/ext/cumo/include/cumo/cuda/driver.h +22 -0
  44. data/ext/cumo/include/cumo/cuda/memory_pool.h +28 -0
  45. data/ext/cumo/include/cumo/cuda/nvrtc.h +22 -0
  46. data/ext/cumo/include/cumo/cuda/runtime.h +40 -0
  47. data/ext/cumo/include/cumo/indexer.h +238 -0
  48. data/ext/cumo/include/cumo/intern.h +142 -0
  49. data/ext/cumo/include/cumo/intern_fwd.h +38 -0
  50. data/ext/cumo/include/cumo/intern_kernel.h +6 -0
  51. data/ext/cumo/include/cumo/narray.h +429 -0
  52. data/ext/cumo/include/cumo/narray_kernel.h +149 -0
  53. data/ext/cumo/include/cumo/ndloop.h +95 -0
  54. data/ext/cumo/include/cumo/reduce_kernel.h +126 -0
  55. data/ext/cumo/include/cumo/template.h +158 -0
  56. data/ext/cumo/include/cumo/template_kernel.h +77 -0
  57. data/ext/cumo/include/cumo/types/bit.h +40 -0
  58. data/ext/cumo/include/cumo/types/bit_kernel.h +34 -0
  59. data/ext/cumo/include/cumo/types/complex.h +402 -0
  60. data/ext/cumo/include/cumo/types/complex_kernel.h +414 -0
  61. data/ext/cumo/include/cumo/types/complex_macro.h +382 -0
  62. data/ext/cumo/include/cumo/types/complex_macro_kernel.h +186 -0
  63. data/ext/cumo/include/cumo/types/dcomplex.h +46 -0
  64. data/ext/cumo/include/cumo/types/dcomplex_kernel.h +13 -0
  65. data/ext/cumo/include/cumo/types/dfloat.h +47 -0
  66. data/ext/cumo/include/cumo/types/dfloat_kernel.h +14 -0
  67. data/ext/cumo/include/cumo/types/float_def.h +34 -0
  68. data/ext/cumo/include/cumo/types/float_def_kernel.h +39 -0
  69. data/ext/cumo/include/cumo/types/float_macro.h +191 -0
  70. data/ext/cumo/include/cumo/types/float_macro_kernel.h +158 -0
  71. data/ext/cumo/include/cumo/types/int16.h +24 -0
  72. data/ext/cumo/include/cumo/types/int16_kernel.h +23 -0
  73. data/ext/cumo/include/cumo/types/int32.h +24 -0
  74. data/ext/cumo/include/cumo/types/int32_kernel.h +19 -0
  75. data/ext/cumo/include/cumo/types/int64.h +24 -0
  76. data/ext/cumo/include/cumo/types/int64_kernel.h +19 -0
  77. data/ext/cumo/include/cumo/types/int8.h +24 -0
  78. data/ext/cumo/include/cumo/types/int8_kernel.h +19 -0
  79. data/ext/cumo/include/cumo/types/int_macro.h +67 -0
  80. data/ext/cumo/include/cumo/types/int_macro_kernel.h +48 -0
  81. data/ext/cumo/include/cumo/types/real_accum.h +486 -0
  82. data/ext/cumo/include/cumo/types/real_accum_kernel.h +101 -0
  83. data/ext/cumo/include/cumo/types/robj_macro.h +80 -0
  84. data/ext/cumo/include/cumo/types/robj_macro_kernel.h +0 -0
  85. data/ext/cumo/include/cumo/types/robject.h +27 -0
  86. data/ext/cumo/include/cumo/types/robject_kernel.h +7 -0
  87. data/ext/cumo/include/cumo/types/scomplex.h +46 -0
  88. data/ext/cumo/include/cumo/types/scomplex_kernel.h +13 -0
  89. data/ext/cumo/include/cumo/types/sfloat.h +48 -0
  90. data/ext/cumo/include/cumo/types/sfloat_kernel.h +14 -0
  91. data/ext/cumo/include/cumo/types/uint16.h +25 -0
  92. data/ext/cumo/include/cumo/types/uint16_kernel.h +20 -0
  93. data/ext/cumo/include/cumo/types/uint32.h +25 -0
  94. data/ext/cumo/include/cumo/types/uint32_kernel.h +20 -0
  95. data/ext/cumo/include/cumo/types/uint64.h +25 -0
  96. data/ext/cumo/include/cumo/types/uint64_kernel.h +20 -0
  97. data/ext/cumo/include/cumo/types/uint8.h +25 -0
  98. data/ext/cumo/include/cumo/types/uint8_kernel.h +20 -0
  99. data/ext/cumo/include/cumo/types/uint_macro.h +58 -0
  100. data/ext/cumo/include/cumo/types/uint_macro_kernel.h +38 -0
  101. data/ext/cumo/include/cumo/types/xint_macro.h +169 -0
  102. data/ext/cumo/include/cumo/types/xint_macro_kernel.h +88 -0
  103. data/ext/cumo/narray/SFMT-params.h +97 -0
  104. data/ext/cumo/narray/SFMT-params19937.h +46 -0
  105. data/ext/cumo/narray/SFMT.c +620 -0
  106. data/ext/cumo/narray/SFMT.h +167 -0
  107. data/ext/cumo/narray/array.c +638 -0
  108. data/ext/cumo/narray/data.c +961 -0
  109. data/ext/cumo/narray/gen/cogen.rb +56 -0
  110. data/ext/cumo/narray/gen/cogen_kernel.rb +58 -0
  111. data/ext/cumo/narray/gen/def/bit.rb +37 -0
  112. data/ext/cumo/narray/gen/def/dcomplex.rb +39 -0
  113. data/ext/cumo/narray/gen/def/dfloat.rb +37 -0
  114. data/ext/cumo/narray/gen/def/int16.rb +36 -0
  115. data/ext/cumo/narray/gen/def/int32.rb +36 -0
  116. data/ext/cumo/narray/gen/def/int64.rb +36 -0
  117. data/ext/cumo/narray/gen/def/int8.rb +36 -0
  118. data/ext/cumo/narray/gen/def/robject.rb +37 -0
  119. data/ext/cumo/narray/gen/def/scomplex.rb +39 -0
  120. data/ext/cumo/narray/gen/def/sfloat.rb +37 -0
  121. data/ext/cumo/narray/gen/def/uint16.rb +36 -0
  122. data/ext/cumo/narray/gen/def/uint32.rb +36 -0
  123. data/ext/cumo/narray/gen/def/uint64.rb +36 -0
  124. data/ext/cumo/narray/gen/def/uint8.rb +36 -0
  125. data/ext/cumo/narray/gen/erbpp2.rb +346 -0
  126. data/ext/cumo/narray/gen/narray_def.rb +268 -0
  127. data/ext/cumo/narray/gen/spec.rb +425 -0
  128. data/ext/cumo/narray/gen/tmpl/accum.c +86 -0
  129. data/ext/cumo/narray/gen/tmpl/accum_binary.c +121 -0
  130. data/ext/cumo/narray/gen/tmpl/accum_binary_kernel.cu +61 -0
  131. data/ext/cumo/narray/gen/tmpl/accum_index.c +119 -0
  132. data/ext/cumo/narray/gen/tmpl/accum_index_kernel.cu +66 -0
  133. data/ext/cumo/narray/gen/tmpl/accum_kernel.cu +12 -0
  134. data/ext/cumo/narray/gen/tmpl/alloc_func.c +107 -0
  135. data/ext/cumo/narray/gen/tmpl/allocate.c +37 -0
  136. data/ext/cumo/narray/gen/tmpl/aref.c +66 -0
  137. data/ext/cumo/narray/gen/tmpl/aref_cpu.c +50 -0
  138. data/ext/cumo/narray/gen/tmpl/aset.c +56 -0
  139. data/ext/cumo/narray/gen/tmpl/binary.c +162 -0
  140. data/ext/cumo/narray/gen/tmpl/binary2.c +70 -0
  141. data/ext/cumo/narray/gen/tmpl/binary2_kernel.cu +15 -0
  142. data/ext/cumo/narray/gen/tmpl/binary_kernel.cu +31 -0
  143. data/ext/cumo/narray/gen/tmpl/binary_s.c +45 -0
  144. data/ext/cumo/narray/gen/tmpl/binary_s_kernel.cu +15 -0
  145. data/ext/cumo/narray/gen/tmpl/bincount.c +181 -0
  146. data/ext/cumo/narray/gen/tmpl/cast.c +44 -0
  147. data/ext/cumo/narray/gen/tmpl/cast_array.c +13 -0
  148. data/ext/cumo/narray/gen/tmpl/class.c +9 -0
  149. data/ext/cumo/narray/gen/tmpl/class_kernel.cu +6 -0
  150. data/ext/cumo/narray/gen/tmpl/clip.c +121 -0
  151. data/ext/cumo/narray/gen/tmpl/coerce_cast.c +10 -0
  152. data/ext/cumo/narray/gen/tmpl/complex_accum_kernel.cu +129 -0
  153. data/ext/cumo/narray/gen/tmpl/cond_binary.c +68 -0
  154. data/ext/cumo/narray/gen/tmpl/cond_binary_kernel.cu +18 -0
  155. data/ext/cumo/narray/gen/tmpl/cond_unary.c +46 -0
  156. data/ext/cumo/narray/gen/tmpl/cum.c +50 -0
  157. data/ext/cumo/narray/gen/tmpl/each.c +47 -0
  158. data/ext/cumo/narray/gen/tmpl/each_with_index.c +70 -0
  159. data/ext/cumo/narray/gen/tmpl/ewcomp.c +79 -0
  160. data/ext/cumo/narray/gen/tmpl/ewcomp_kernel.cu +19 -0
  161. data/ext/cumo/narray/gen/tmpl/extract.c +22 -0
  162. data/ext/cumo/narray/gen/tmpl/extract_cpu.c +26 -0
  163. data/ext/cumo/narray/gen/tmpl/extract_data.c +53 -0
  164. data/ext/cumo/narray/gen/tmpl/eye.c +105 -0
  165. data/ext/cumo/narray/gen/tmpl/eye_kernel.cu +19 -0
  166. data/ext/cumo/narray/gen/tmpl/fill.c +52 -0
  167. data/ext/cumo/narray/gen/tmpl/fill_kernel.cu +29 -0
  168. data/ext/cumo/narray/gen/tmpl/float_accum_kernel.cu +106 -0
  169. data/ext/cumo/narray/gen/tmpl/format.c +62 -0
  170. data/ext/cumo/narray/gen/tmpl/format_to_a.c +49 -0
  171. data/ext/cumo/narray/gen/tmpl/frexp.c +38 -0
  172. data/ext/cumo/narray/gen/tmpl/gemm.c +203 -0
  173. data/ext/cumo/narray/gen/tmpl/init_class.c +20 -0
  174. data/ext/cumo/narray/gen/tmpl/init_module.c +12 -0
  175. data/ext/cumo/narray/gen/tmpl/inspect.c +21 -0
  176. data/ext/cumo/narray/gen/tmpl/lib.c +50 -0
  177. data/ext/cumo/narray/gen/tmpl/lib_kernel.cu +24 -0
  178. data/ext/cumo/narray/gen/tmpl/logseq.c +102 -0
  179. data/ext/cumo/narray/gen/tmpl/logseq_kernel.cu +31 -0
  180. data/ext/cumo/narray/gen/tmpl/map_with_index.c +98 -0
  181. data/ext/cumo/narray/gen/tmpl/median.c +66 -0
  182. data/ext/cumo/narray/gen/tmpl/minmax.c +47 -0
  183. data/ext/cumo/narray/gen/tmpl/module.c +9 -0
  184. data/ext/cumo/narray/gen/tmpl/module_kernel.cu +1 -0
  185. data/ext/cumo/narray/gen/tmpl/new_dim0.c +15 -0
  186. data/ext/cumo/narray/gen/tmpl/new_dim0_kernel.cu +8 -0
  187. data/ext/cumo/narray/gen/tmpl/poly.c +50 -0
  188. data/ext/cumo/narray/gen/tmpl/pow.c +97 -0
  189. data/ext/cumo/narray/gen/tmpl/pow_kernel.cu +29 -0
  190. data/ext/cumo/narray/gen/tmpl/powint.c +17 -0
  191. data/ext/cumo/narray/gen/tmpl/qsort.c +212 -0
  192. data/ext/cumo/narray/gen/tmpl/rand.c +168 -0
  193. data/ext/cumo/narray/gen/tmpl/rand_norm.c +121 -0
  194. data/ext/cumo/narray/gen/tmpl/real_accum_kernel.cu +75 -0
  195. data/ext/cumo/narray/gen/tmpl/seq.c +112 -0
  196. data/ext/cumo/narray/gen/tmpl/seq_kernel.cu +43 -0
  197. data/ext/cumo/narray/gen/tmpl/set2.c +57 -0
  198. data/ext/cumo/narray/gen/tmpl/sort.c +48 -0
  199. data/ext/cumo/narray/gen/tmpl/sort_index.c +111 -0
  200. data/ext/cumo/narray/gen/tmpl/store.c +41 -0
  201. data/ext/cumo/narray/gen/tmpl/store_array.c +187 -0
  202. data/ext/cumo/narray/gen/tmpl/store_array_kernel.cu +58 -0
  203. data/ext/cumo/narray/gen/tmpl/store_bit.c +86 -0
  204. data/ext/cumo/narray/gen/tmpl/store_bit_kernel.cu +66 -0
  205. data/ext/cumo/narray/gen/tmpl/store_from.c +81 -0
  206. data/ext/cumo/narray/gen/tmpl/store_from_kernel.cu +58 -0
  207. data/ext/cumo/narray/gen/tmpl/store_kernel.cu +3 -0
  208. data/ext/cumo/narray/gen/tmpl/store_numeric.c +9 -0
  209. data/ext/cumo/narray/gen/tmpl/to_a.c +43 -0
  210. data/ext/cumo/narray/gen/tmpl/unary.c +132 -0
  211. data/ext/cumo/narray/gen/tmpl/unary2.c +60 -0
  212. data/ext/cumo/narray/gen/tmpl/unary_kernel.cu +72 -0
  213. data/ext/cumo/narray/gen/tmpl/unary_ret2.c +34 -0
  214. data/ext/cumo/narray/gen/tmpl/unary_s.c +86 -0
  215. data/ext/cumo/narray/gen/tmpl/unary_s_kernel.cu +58 -0
  216. data/ext/cumo/narray/gen/tmpl_bit/allocate.c +24 -0
  217. data/ext/cumo/narray/gen/tmpl_bit/aref.c +54 -0
  218. data/ext/cumo/narray/gen/tmpl_bit/aref_cpu.c +57 -0
  219. data/ext/cumo/narray/gen/tmpl_bit/aset.c +56 -0
  220. data/ext/cumo/narray/gen/tmpl_bit/binary.c +98 -0
  221. data/ext/cumo/narray/gen/tmpl_bit/bit_count.c +64 -0
  222. data/ext/cumo/narray/gen/tmpl_bit/bit_count_cpu.c +88 -0
  223. data/ext/cumo/narray/gen/tmpl_bit/bit_count_kernel.cu +76 -0
  224. data/ext/cumo/narray/gen/tmpl_bit/bit_reduce.c +133 -0
  225. data/ext/cumo/narray/gen/tmpl_bit/each.c +48 -0
  226. data/ext/cumo/narray/gen/tmpl_bit/each_with_index.c +70 -0
  227. data/ext/cumo/narray/gen/tmpl_bit/extract.c +30 -0
  228. data/ext/cumo/narray/gen/tmpl_bit/extract_cpu.c +29 -0
  229. data/ext/cumo/narray/gen/tmpl_bit/fill.c +69 -0
  230. data/ext/cumo/narray/gen/tmpl_bit/format.c +64 -0
  231. data/ext/cumo/narray/gen/tmpl_bit/format_to_a.c +51 -0
  232. data/ext/cumo/narray/gen/tmpl_bit/inspect.c +21 -0
  233. data/ext/cumo/narray/gen/tmpl_bit/mask.c +136 -0
  234. data/ext/cumo/narray/gen/tmpl_bit/none_p.c +14 -0
  235. data/ext/cumo/narray/gen/tmpl_bit/store_array.c +108 -0
  236. data/ext/cumo/narray/gen/tmpl_bit/store_bit.c +70 -0
  237. data/ext/cumo/narray/gen/tmpl_bit/store_from.c +60 -0
  238. data/ext/cumo/narray/gen/tmpl_bit/to_a.c +47 -0
  239. data/ext/cumo/narray/gen/tmpl_bit/unary.c +81 -0
  240. data/ext/cumo/narray/gen/tmpl_bit/where.c +90 -0
  241. data/ext/cumo/narray/gen/tmpl_bit/where2.c +95 -0
  242. data/ext/cumo/narray/index.c +880 -0
  243. data/ext/cumo/narray/kwargs.c +153 -0
  244. data/ext/cumo/narray/math.c +142 -0
  245. data/ext/cumo/narray/narray.c +1948 -0
  246. data/ext/cumo/narray/ndloop.c +2105 -0
  247. data/ext/cumo/narray/rand.c +45 -0
  248. data/ext/cumo/narray/step.c +474 -0
  249. data/ext/cumo/narray/struct.c +886 -0
  250. data/lib/cumo.rb +3 -0
  251. data/lib/cumo/cuda.rb +11 -0
  252. data/lib/cumo/cuda/compile_error.rb +36 -0
  253. data/lib/cumo/cuda/compiler.rb +161 -0
  254. data/lib/cumo/cuda/device.rb +47 -0
  255. data/lib/cumo/cuda/link_state.rb +31 -0
  256. data/lib/cumo/cuda/module.rb +40 -0
  257. data/lib/cumo/cuda/nvrtc_program.rb +27 -0
  258. data/lib/cumo/linalg.rb +12 -0
  259. data/lib/cumo/narray.rb +2 -0
  260. data/lib/cumo/narray/extra.rb +1278 -0
  261. data/lib/erbpp.rb +294 -0
  262. data/lib/erbpp/line_number.rb +137 -0
  263. data/lib/erbpp/narray_def.rb +381 -0
  264. data/numo-narray-version +1 -0
  265. data/run.gdb +7 -0
  266. metadata +353 -0
@@ -0,0 +1,167 @@
1
+ /**
2
+ * @file SFMT.h
3
+ *
4
+ * @brief SIMD oriented Fast Mersenne Twister(SFMT) pseudorandom
5
+ * number generator
6
+ *
7
+ * @author Mutsuo Saito (Hiroshima University)
8
+ * @author Makoto Matsumoto (Hiroshima University)
9
+ *
10
+ * Copyright (C) 2006, 2007 Mutsuo Saito, Makoto Matsumoto and Hiroshima
11
+ * University. All rights reserved.
12
+ *
13
+ * The new BSD License is applied to this software.
14
+ * see LICENSE.txt
15
+ *
16
+ * @note We assume that your system has inttypes.h. If your system
17
+ * doesn't have inttypes.h, you have to typedef uint32_t and uint64_t,
18
+ * and you have to define PRIu64 and PRIx64 in this file as follows:
19
+ * @verbatim
20
+ typedef unsigned int uint32_t
21
+ typedef unsigned long long uint64_t
22
+ #define PRIu64 "llu"
23
+ #define PRIx64 "llx"
24
+ @endverbatim
25
+ * uint32_t must be exactly 32-bit unsigned integer type (no more, no
26
+ * less), and uint64_t must be exactly 64-bit unsigned integer type.
27
+ * PRIu64 and PRIx64 are used for printf function to print 64-bit
28
+ * unsigned int and 64-bit unsigned int in hexadecimal format.
29
+ */
30
+
31
+ #ifndef SFMT_H
32
+ #define SFMT_H
33
+
34
+ #include <stdio.h>
35
+
36
+ #if defined(__STDC_VERSION__) && (__STDC_VERSION__ >= 199901L)
37
+ #include <inttypes.h>
38
+ #elif defined(_MSC_VER) || defined(__BORLANDC__)
39
+ typedef unsigned int uint32_t;
40
+ typedef unsigned __int64 uint64_t;
41
+ #define inline __inline
42
+ #else
43
+ #include <inttypes.h>
44
+ #if defined(__GNUC__)
45
+ #define inline __inline__
46
+ #endif
47
+ #endif
48
+
49
+ #ifndef PRIu64
50
+ #if defined(_MSC_VER) || defined(__BORLANDC__)
51
+ #define PRIu64 "I64u"
52
+ #define PRIx64 "I64x"
53
+ #else
54
+ #define PRIu64 "llu"
55
+ #define PRIx64 "llx"
56
+ #endif
57
+ #endif
58
+
59
+ #if defined(__GNUC__)
60
+ #define ALWAYSINLINE __attribute__((always_inline))
61
+ #else
62
+ #define ALWAYSINLINE
63
+ #endif
64
+
65
+ #if defined(_MSC_VER)
66
+ #if _MSC_VER >= 1200
67
+ #define PRE_ALWAYS __forceinline
68
+ #else
69
+ #define PRE_ALWAYS inline
70
+ #endif
71
+ #else
72
+ #define PRE_ALWAYS inline
73
+ #endif
74
+
75
+ /* Add cumo_ prefix to avoid symbol colision with numo */
76
+ #define gen_rand32 cumo_gen_rand32
77
+ uint32_t cumo_gen_rand32(void);
78
+ #define gen_rand64 cumo_gen_rand64
79
+ uint64_t cumo_gen_rand64(void);
80
+ #define fill_array32 cumo_fill_array32
81
+ void cumo_fill_array32(uint32_t* array, int size);
82
+ #define fill_array64 cumo_fill_array64
83
+ void cumo_fill_array64(uint64_t* array, int size);
84
+ #define init_gen_rand cumo_init_gen_rand
85
+ void cumo_init_gen_rand(uint32_t seed);
86
+ #define init_by_array cumo_init_by_array
87
+ void cumo_init_by_array(uint32_t* init_key, int key_length);
88
+ #define get_idstring cumo_get_idstring
89
+ const char* cumo_get_idstring(void);
90
+ #define get_min_array_size32 cumo_get_min_array_size32
91
+ int cumo_get_min_array_size32(void);
92
+ #define get_min_array_size64 cumo_get_min_array_size64
93
+ int cumo_get_min_array_size64(void);
94
+
95
+ /* These real versions are due to Isaku Wada */
96
+ /** generates a random number on [0,1]-real-interval */
97
+ inline static double to_real1(uint32_t v)
98
+ {
99
+ return v * (1.0/4294967295.0);
100
+ /* divided by 2^32-1 */
101
+ }
102
+
103
+ /** generates a random number on [0,1]-real-interval */
104
+ inline static double genrand_real1(void)
105
+ {
106
+ return to_real1(gen_rand32());
107
+ }
108
+
109
+ /** generates a random number on [0,1)-real-interval */
110
+ inline static double to_real2(uint32_t v)
111
+ {
112
+ return v * (1.0/4294967296.0);
113
+ /* divided by 2^32 */
114
+ }
115
+
116
+ /** generates a random number on [0,1)-real-interval */
117
+ inline static double genrand_real2(void)
118
+ {
119
+ return to_real2(gen_rand32());
120
+ }
121
+
122
+ /** generates a random number on (0,1)-real-interval */
123
+ inline static double to_real3(uint32_t v)
124
+ {
125
+ return (((double)v) + 0.5)*(1.0/4294967296.0);
126
+ /* divided by 2^32 */
127
+ }
128
+
129
+ /** generates a random number on (0,1)-real-interval */
130
+ inline static double genrand_real3(void)
131
+ {
132
+ return to_real3(gen_rand32());
133
+ }
134
+ /** These real versions are due to Isaku Wada */
135
+
136
+ /** generates a random number on [0,1) with 53-bit resolution*/
137
+ inline static double to_res53(uint64_t v)
138
+ {
139
+ return v * (1.0/18446744073709551616.0L);
140
+ }
141
+
142
+ /** generates a random number on [0,1) with 53-bit resolution from two
143
+ * 32 bit integers */
144
+ inline static double to_res53_mix(uint32_t x, uint32_t y)
145
+ {
146
+ return to_res53(x | ((uint64_t)y << 32));
147
+ }
148
+
149
+ /** generates a random number on [0,1) with 53-bit resolution
150
+ */
151
+ inline static double genrand_res53(void)
152
+ {
153
+ return to_res53(gen_rand64());
154
+ }
155
+
156
+ /** generates a random number on [0,1) with 53-bit resolution
157
+ using 32bit integer.
158
+ */
159
+ inline static double genrand_res53_mix(void)
160
+ {
161
+ uint32_t x, y;
162
+
163
+ x = gen_rand32();
164
+ y = gen_rand32();
165
+ return to_res53_mix(x, y);
166
+ }
167
+ #endif
@@ -0,0 +1,638 @@
1
+ #include <ruby.h>
2
+ #include "cumo/narray.h"
3
+
4
+ // mdai: Multi-Dimensional Array Investigation
5
+ typedef struct {
6
+ size_t shape;
7
+ VALUE val;
8
+ } na_mdai_item_t;
9
+
10
+ typedef struct {
11
+ int capa;
12
+ na_mdai_item_t *item;
13
+ int type; // Ruby numeric type - investigated separately
14
+ VALUE na_type; // NArray type
15
+ VALUE int_max;
16
+ } na_mdai_t;
17
+
18
+ // Order of Ruby object.
19
+ enum { NA_NONE, NA_BIT, NA_INT32, NA_INT64, NA_RATIONAL,
20
+ NA_DFLOAT, NA_DCOMPLEX, NA_ROBJ, NA_NTYPES };
21
+
22
+ static ID id_begin;
23
+ static ID id_end;
24
+ static ID id_step;
25
+ static ID id_abs;
26
+ static ID id_cast;
27
+ static ID id_le;
28
+ static ID id_Complex;
29
+
30
+
31
+ static VALUE
32
+ na_object_type(int type, VALUE v)
33
+ {
34
+ static VALUE int32_max = Qnil;
35
+ if (NIL_P(int32_max))
36
+ int32_max = ULONG2NUM(2147483647);
37
+
38
+ switch(TYPE(v)) {
39
+
40
+ case T_TRUE:
41
+ case T_FALSE:
42
+ if (type<NA_BIT)
43
+ return NA_BIT;
44
+ return type;
45
+
46
+ #if SIZEOF_LONG == 4
47
+ case T_FIXNUM:
48
+ if (type<NA_INT32)
49
+ return NA_INT32;
50
+ return type;
51
+ case T_BIGNUM:
52
+ if (type<NA_INT64) {
53
+ v = rb_funcall(v,id_abs,0);
54
+ if (RTEST(rb_funcall(v,id_le,1,int32_max))) {
55
+ if (type<NA_INT32)
56
+ return NA_INT32;
57
+ } else {
58
+ return NA_INT64;
59
+ }
60
+ }
61
+ return type;
62
+
63
+ #elif SIZEOF_LONG == 8
64
+ case T_FIXNUM:
65
+ if (type<NA_INT64) {
66
+ long x = NUM2LONG(v);
67
+ if (x<0) x=-x;
68
+ if (x<=2147483647) {
69
+ if (type<NA_INT32)
70
+ return NA_INT32;
71
+ } else {
72
+ return NA_INT64;
73
+ }
74
+ }
75
+ return type;
76
+ case T_BIGNUM:
77
+ if (type<NA_INT64)
78
+ return NA_INT64;
79
+ return type;
80
+ #else
81
+ case T_FIXNUM:
82
+ case T_BIGNUM:
83
+ if (type<NA_INT64) {
84
+ v = rb_funcall(v,id_abs,0);
85
+ if (RTEST(rb_funcall(v,id_le,1,int32_max))) {
86
+ if (type<NA_INT32)
87
+ return NA_INT32;
88
+ } else {
89
+ return NA_INT64;
90
+ }
91
+ }
92
+ return type;
93
+ #endif
94
+
95
+ case T_FLOAT:
96
+ if (type<NA_DFLOAT)
97
+ return NA_DFLOAT;
98
+ return type;
99
+
100
+ case T_NIL:
101
+ return type;
102
+
103
+ default:
104
+ if (CLASS_OF(v) == rb_const_get( rb_cObject, id_Complex )) {
105
+ return NA_DCOMPLEX;
106
+ }
107
+ }
108
+ return NA_ROBJ;
109
+ }
110
+
111
+
112
+ #define MDAI_ATTR_TYPE(tp,v,attr) \
113
+ {tp = na_object_type(tp,rb_funcall(v,id_##attr,0));}
114
+
115
+ static int na_mdai_object_type(int type, VALUE v)
116
+ {
117
+ if (rb_obj_is_kind_of(v, rb_cRange)) {
118
+ MDAI_ATTR_TYPE(type,v,begin);
119
+ MDAI_ATTR_TYPE(type,v,end);
120
+ } else if (rb_obj_is_kind_of(v, na_cStep)) {
121
+ MDAI_ATTR_TYPE(type,v,begin);
122
+ MDAI_ATTR_TYPE(type,v,end);
123
+ MDAI_ATTR_TYPE(type,v,step);
124
+ } else {
125
+ type = na_object_type(type,v);
126
+ }
127
+ return type;
128
+ }
129
+
130
+
131
+ static na_mdai_t *
132
+ na_mdai_alloc(VALUE ary)
133
+ {
134
+ int i, n=4;
135
+ na_mdai_t *mdai;
136
+
137
+ mdai = ALLOC(na_mdai_t);
138
+ mdai->capa = n;
139
+ mdai->item = ALLOC_N( na_mdai_item_t, n );
140
+ for (i=0; i<n; i++) {
141
+ mdai->item[i].shape = 0;
142
+ mdai->item[i].val = Qnil;
143
+ }
144
+ mdai->item[0].val = ary;
145
+ mdai->type = NA_NONE;
146
+ mdai->na_type = Qnil;
147
+
148
+ return mdai;
149
+ }
150
+
151
+ static void
152
+ na_mdai_realloc(na_mdai_t *mdai, int n_extra)
153
+ {
154
+ int i, n;
155
+
156
+ i = mdai->capa;
157
+ mdai->capa += n_extra;
158
+ n = mdai->capa;
159
+ REALLOC_N( mdai->item, na_mdai_item_t, n );
160
+ for (; i<n; i++) {
161
+ mdai->item[i].shape = 0;
162
+ mdai->item[i].val = Qnil;
163
+ }
164
+ }
165
+
166
+ static void
167
+ na_mdai_free(void *ptr)
168
+ {
169
+ na_mdai_t *mdai = (na_mdai_t*)ptr;
170
+ xfree(mdai->item);
171
+ xfree(mdai);
172
+ }
173
+
174
+
175
+ /* investigate ndim, shape, type of Array */
176
+ static int
177
+ na_mdai_investigate(na_mdai_t *mdai, int ndim)
178
+ {
179
+ ssize_t i;
180
+ int j;
181
+ size_t len, length;
182
+ double dbeg, dstep;
183
+ VALUE v;
184
+ VALUE val;
185
+
186
+ val = mdai->item[ndim-1].val;
187
+ len = RARRAY_LEN(val);
188
+
189
+ for (i=0; i < RARRAY_LEN(val); i++) {
190
+ v = RARRAY_AREF(val,i);
191
+
192
+ if (TYPE(v) == T_ARRAY) {
193
+ /* check recursive array */
194
+ for (j=0; j<ndim; j++) {
195
+ if (mdai->item[j].val == v)
196
+ rb_raise(rb_eStandardError,
197
+ "cannot convert from a recursive Array to NArray");
198
+ }
199
+ if ( ndim >= mdai->capa ) {
200
+ na_mdai_realloc(mdai,4);
201
+ }
202
+ mdai->item[ndim].val = v;
203
+ if ( na_mdai_investigate(mdai,ndim+1) ) {
204
+ len--; /* Array is empty */
205
+ }
206
+ }
207
+ else
208
+ if (rb_obj_is_kind_of(v, rb_cRange) || rb_obj_is_kind_of(v, na_cStep)) {
209
+ nary_step_sequence(v,&length,&dbeg,&dstep);
210
+ len += length-1;
211
+ mdai->type = na_mdai_object_type(mdai->type, v);
212
+ }
213
+ else if (IsNArray(v)) {
214
+ int r;
215
+ narray_t *na;
216
+ GetNArray(v,na);
217
+ if ( na->ndim == 0 ) {
218
+ len--; /* NArray is empty */
219
+ } else {
220
+ if ( ndim+na->ndim > mdai->capa ) {
221
+ na_mdai_realloc(mdai,((na->ndim-1)/4+1)*4);
222
+ }
223
+ for ( j=0,r=ndim; j < na->ndim ; j++,r++ ) {
224
+ if ( mdai->item[r].shape < na->shape[j] )
225
+ mdai->item[r].shape = na->shape[j];
226
+ }
227
+ }
228
+ // type
229
+ if (NIL_P(mdai->na_type)) {
230
+ mdai->na_type = CLASS_OF(v);
231
+ } else {
232
+ mdai->na_type = na_upcast(CLASS_OF(v), mdai->na_type);
233
+ }
234
+ } else {
235
+ mdai->type = na_mdai_object_type(mdai->type, v);
236
+ }
237
+ }
238
+
239
+ if (len==0) return 1; /* this array is empty */
240
+ if (mdai->item[ndim-1].shape < len) {
241
+ mdai->item[ndim-1].shape = len;
242
+ }
243
+ return 0;
244
+ }
245
+
246
+
247
+ static inline int
248
+ na_mdai_ndim(na_mdai_t *mdai)
249
+ {
250
+ int i;
251
+ // Dimension
252
+ for (i=0; i < mdai->capa && mdai->item[i].shape > 0; i++) ;
253
+ return i;
254
+ }
255
+
256
+ static inline void
257
+ na_mdai_shape(na_mdai_t *mdai, int ndim, size_t *shape)
258
+ {
259
+ int i;
260
+ for (i=0; i<ndim; i++) {
261
+ shape[i] = mdai->item[i].shape;
262
+ }
263
+ }
264
+
265
+ static VALUE
266
+ na_mdai_dtype_numeric(int type)
267
+ {
268
+ VALUE tp;
269
+ // DataType
270
+ switch(type) {
271
+ case NA_BIT:
272
+ tp = cumo_cBit;
273
+ break;
274
+ case NA_INT32:
275
+ tp = cumo_cInt32;
276
+ break;
277
+ case NA_INT64:
278
+ tp = cumo_cInt64;
279
+ break;
280
+ case NA_DFLOAT:
281
+ tp = cumo_cDFloat;
282
+ break;
283
+ case NA_DCOMPLEX:
284
+ tp = cumo_cDComplex;
285
+ break;
286
+ case NA_ROBJ:
287
+ tp = cumo_cRObject;
288
+ break;
289
+ default:
290
+ tp = Qnil;
291
+ }
292
+ return tp;
293
+ }
294
+
295
+ static VALUE
296
+ na_mdai_dtype(na_mdai_t *mdai)
297
+ {
298
+ VALUE tp;
299
+
300
+ tp = na_mdai_dtype_numeric(mdai->type);
301
+
302
+ if (!NIL_P(mdai->na_type)) {
303
+ if (NIL_P(tp)) {
304
+ tp = mdai->na_type;
305
+ } else {
306
+ tp = na_upcast(mdai->na_type,tp);
307
+ }
308
+ }
309
+ return tp;
310
+ }
311
+
312
+
313
+ static inline VALUE
314
+ update_type(VALUE *ptype, VALUE dtype)
315
+ {
316
+ if (ptype) {
317
+ if (*ptype == cNArray || !RTEST(*ptype)) {
318
+ *ptype = dtype;
319
+ } else {
320
+ dtype = *ptype;
321
+ }
322
+ }
323
+ return dtype;
324
+ }
325
+
326
+ static inline void
327
+ check_subclass_of_narray(VALUE dtype)
328
+ {
329
+ if (RTEST(rb_obj_is_kind_of(dtype, rb_cClass))) {
330
+ if (RTEST(rb_funcall(dtype, id_le, 1, cNArray))) {
331
+ return;
332
+ }
333
+ }
334
+ rb_raise(nary_eCastError, "cannot convert to NArray");
335
+ }
336
+
337
+
338
+ static size_t
339
+ na_mdai_memsize(const void *ptr)
340
+ {
341
+ const na_mdai_t *mdai = (const na_mdai_t*)ptr;
342
+
343
+ return sizeof(na_mdai_t) + mdai->capa * sizeof(na_mdai_item_t);
344
+ }
345
+
346
+ static const rb_data_type_t mdai_data_type = {
347
+ "Cumo::NArray/mdai",
348
+ {NULL, na_mdai_free, na_mdai_memsize,},
349
+ 0, 0, RUBY_TYPED_FREE_IMMEDIATELY|RUBY_TYPED_WB_PROTECTED
350
+ };
351
+
352
+
353
+ static void
354
+ na_composition3_ary(VALUE ary, VALUE *ptype, VALUE *pshape, VALUE *pnary)
355
+ {
356
+ VALUE vmdai;
357
+ na_mdai_t *mdai;
358
+ int i, ndim;
359
+ size_t *shape;
360
+ VALUE dtype, dshape;
361
+
362
+ mdai = na_mdai_alloc(ary);
363
+ vmdai = TypedData_Wrap_Struct(rb_cData, &mdai_data_type, (void*)mdai);
364
+ if ( na_mdai_investigate(mdai, 1) ) {
365
+ // empty
366
+ dtype = update_type(ptype, cumo_cInt32);
367
+ if (pshape) {
368
+ *pshape = rb_ary_new3(1, INT2FIX(0));
369
+ }
370
+ if (pnary) {
371
+ check_subclass_of_narray(dtype);
372
+ shape = ALLOCA_N(size_t, 1);
373
+ shape[0] = 0;
374
+ *pnary = nary_new(dtype, 1, shape);
375
+ }
376
+ } else {
377
+ ndim = na_mdai_ndim(mdai);
378
+ shape = ALLOCA_N(size_t, ndim);
379
+ na_mdai_shape(mdai, ndim, shape);
380
+ dtype = update_type(ptype, na_mdai_dtype(mdai));
381
+ if (pshape) {
382
+ dshape = rb_ary_new2(ndim);
383
+ for (i=0; i<ndim; i++) {
384
+ rb_ary_push(dshape, SIZET2NUM(shape[i]));
385
+ }
386
+ *pshape = dshape;
387
+ }
388
+ if (pnary) {
389
+ check_subclass_of_narray(dtype);
390
+ *pnary = nary_new(dtype, ndim, shape);
391
+ }
392
+ }
393
+ RB_GC_GUARD(vmdai);
394
+ }
395
+
396
+
397
+ static void
398
+ na_composition3(VALUE obj, VALUE *ptype, VALUE *pshape, VALUE *pnary)
399
+ {
400
+ VALUE dtype, dshape;
401
+
402
+ if (TYPE(obj) == T_ARRAY) {
403
+ na_composition3_ary(obj, ptype, pshape, pnary);
404
+ }
405
+ else if (RTEST(rb_obj_is_kind_of(obj,rb_cNumeric))) {
406
+ dtype = na_mdai_dtype_numeric(na_mdai_object_type(NA_NONE, obj));
407
+ dtype = update_type(ptype, dtype);
408
+ if (pshape) {
409
+ *pshape = rb_ary_new();
410
+ }
411
+ if (pnary) {
412
+ check_subclass_of_narray(dtype);
413
+ *pnary = nary_new(dtype, 0, 0);
414
+ }
415
+ }
416
+ else if (IsNArray(obj)) {
417
+ int i, ndim;
418
+ narray_t *na;
419
+ GetNArray(obj,na);
420
+ ndim = na->ndim;
421
+ dtype = update_type(ptype, CLASS_OF(obj));
422
+ if (pshape) {
423
+ dshape = rb_ary_new2(ndim);
424
+ for (i=0; i<ndim; i++) {
425
+ rb_ary_push(dshape, SIZET2NUM(na->shape[i]));
426
+ }
427
+ *pshape = dshape;
428
+ }
429
+ if (pnary) {
430
+ *pnary = nary_new(dtype, ndim, na->shape);
431
+ }
432
+ } else {
433
+ rb_raise(rb_eTypeError,"invalid type for NArray: %s",
434
+ rb_class2name(CLASS_OF(obj)));
435
+ }
436
+ }
437
+
438
+
439
+ static VALUE
440
+ na_s_array_shape(VALUE mod, VALUE ary)
441
+ {
442
+ VALUE shape;
443
+
444
+ if (TYPE(ary) != T_ARRAY) {
445
+ // 0-dimension
446
+ return rb_ary_new();
447
+ }
448
+ na_composition3(ary, 0, &shape, 0);
449
+ return shape;
450
+ }
451
+
452
+
453
+ /*
454
+ Generate new unallocated NArray instance with shape and type defined from obj.
455
+ Cumo::NArray.new_like(obj) returns instance whose type is defined from obj.
456
+ Cumo::DFloat.new_like(obj) returns DFloat instance.
457
+
458
+ @overload new_like(obj)
459
+ @param [Numeric,Array,Cumo::NArray] obj
460
+ @return [Cumo::NArray]
461
+ @example
462
+ Cumo::NArray.new_like([[1,2,3],[4,5,6]])
463
+ => Cumo::Int32#shape=[2,3](empty)
464
+ Cumo::DFloat.new_like([[1,2],[3,4]])
465
+ => Cumo::DFloat#shape=[2,2](empty)
466
+ Cumo::NArray.new_like([1,2i,3])
467
+ => Cumo::DComplex#shape=[3](empty)
468
+ */
469
+ VALUE
470
+ na_s_new_like(VALUE type, VALUE obj)
471
+ {
472
+ VALUE newary;
473
+
474
+ na_composition3(obj, &type, 0, &newary);
475
+ return newary;
476
+ }
477
+
478
+
479
+ VALUE
480
+ na_ary_composition_dtype(VALUE ary)
481
+ {
482
+ VALUE type = Qnil;
483
+
484
+ na_composition3(ary, &type, 0, 0);
485
+ return type;
486
+ }
487
+
488
+ static VALUE
489
+ na_s_array_type(VALUE mod, VALUE ary)
490
+ {
491
+ return na_ary_composition_dtype(ary);
492
+ }
493
+
494
+
495
+ /*
496
+ Generate NArray object. NArray datatype is automatically selected.
497
+ @overload [](elements)
498
+ @param [Numeric,Array] elements
499
+ @return [NArray]
500
+ */
501
+ static VALUE
502
+ nary_s_bracket(VALUE klass, VALUE ary)
503
+ {
504
+ VALUE dtype=Qnil;
505
+
506
+ if (TYPE(ary)!=T_ARRAY) {
507
+ rb_bug("Argument is not array");
508
+ }
509
+ dtype = na_ary_composition_dtype(ary);
510
+ check_subclass_of_narray(dtype);
511
+ return rb_funcall(dtype, id_cast, 1, ary);
512
+ }
513
+
514
+
515
+ //VALUE
516
+ //nst_check_compatibility(VALUE self, VALUE ary);
517
+
518
+
519
+ /* investigate ndim, shape, type of Array */
520
+ /*
521
+ static int
522
+ na_mdai_for_struct(na_mdai_t *mdai, int ndim)
523
+ {
524
+ size_t i;
525
+ int j, r;
526
+ size_t len;
527
+ VALUE v;
528
+ VALUE val;
529
+ narray_t *na;
530
+
531
+ //fprintf(stderr,"ndim=%d\n",ndim); rb_p(mdai->na_type);
532
+ if (ndim>4) { abort(); }
533
+ val = mdai->item[ndim].val;
534
+
535
+ //fpintf(stderr,"val = "); rb_p(val);
536
+
537
+ if (CLASS_OF(val) == mdai->na_type) {
538
+ GetNArray(val,na);
539
+ if ( ndim+na->ndim > mdai->capa ) {
540
+ abort();
541
+ na_mdai_realloc(mdai,((na->ndim-1)/4+1)*4);
542
+ }
543
+ for ( j=0,r=ndim; j < na->ndim; j++,r++ ) {
544
+ if ( mdai->item[r].shape < na->shape[j] )
545
+ mdai->item[r].shape = na->shape[j];
546
+ }
547
+ return 1;
548
+ }
549
+
550
+ if (TYPE(val) == T_ARRAY) {
551
+ // check recursive array
552
+ for (j=0; j<ndim-1; j++) {
553
+ if (mdai->item[j].val == val)
554
+ rb_raise(rb_eStandardError,
555
+ "cannot convert from a recursive Array to NArray");
556
+ }
557
+ //fprintf(stderr,"check:"); rb_p(val);
558
+ // val is a Struct recort
559
+ if (RTEST( nst_check_compatibility(mdai->na_type, val) )) {
560
+ //fputs("compati\n",stderr);
561
+ return 1;
562
+ }
563
+ // otherwise, multi-dimention
564
+ if (ndim >= mdai->capa) {
565
+ //fprintf(stderr,"exeed capa\n"); abort();
566
+ na_mdai_realloc(mdai,4);
567
+ }
568
+ // finally, multidimension-check
569
+ len = RARRAY_LEN(val);
570
+ for (i=0; i < len; i++) {
571
+ v = RARRAY_AREF(val,i);
572
+ if (TYPE(v) != T_ARRAY) {
573
+ //abort();
574
+ return 0;
575
+ }
576
+ }
577
+ for (i=0; i < len; i++) {
578
+ v = RARRAY_AREF(val,i);
579
+ //fprintf(stderr,"check:"); rb_p(v);
580
+ mdai->item[ndim+1].val = v;
581
+ if ( na_mdai_for_struct( mdai, ndim+1 ) == 0 ) {
582
+ //fprintf(stderr,"not struct:"); rb_p(v);
583
+ //abort();
584
+ return 0;
585
+ }
586
+ }
587
+ if (mdai->item[ndim].shape < len) {
588
+ mdai->item[ndim].shape = len;
589
+ }
590
+ return 1;
591
+ }
592
+
593
+ //fprintf(stderr,"invalid for struct:"); rb_p(val); abort();
594
+ return 0;
595
+ }
596
+ */
597
+
598
+
599
+ /*
600
+ VALUE
601
+ na_ary_composition_for_struct(VALUE nstruct, VALUE ary)
602
+ {
603
+ volatile VALUE vmdai, vnc;
604
+ na_mdai_t *mdai;
605
+ na_compose_t *nc;
606
+
607
+ mdai = na_mdai_alloc(ary);
608
+ mdai->na_type = nstruct;
609
+ vmdai = TypedData_Wrap_Struct(rb_cData, &mdai_data_type, (void*)mdai);
610
+ na_mdai_for_struct(mdai, 0);
611
+ nc = na_compose_alloc();
612
+ vnc = WrapCompose(nc);
613
+ na_mdai_result(mdai, nc);
614
+ //fprintf(stderr,"nc->ndim=%d\n",nc->ndim);
615
+ rb_gc_force_recycle(vmdai);
616
+ return vnc;
617
+ }
618
+ */
619
+
620
+
621
+
622
+ void
623
+ Init_cumo_nary_array()
624
+ {
625
+ rb_define_singleton_method(cNArray, "array_shape", na_s_array_shape, 1);
626
+ rb_define_singleton_method(cNArray, "array_type", na_s_array_type, 1);
627
+ rb_define_singleton_method(cNArray, "new_like", na_s_new_like, 1);
628
+
629
+ rb_define_singleton_method(cNArray, "[]", nary_s_bracket, -2);
630
+
631
+ id_begin = rb_intern("begin");
632
+ id_end = rb_intern("end");
633
+ id_step = rb_intern("step");
634
+ id_cast = rb_intern("cast");
635
+ id_abs = rb_intern("abs");
636
+ id_le = rb_intern("<=");
637
+ id_Complex = rb_intern("Complex");
638
+ }