Mitaka.rb #5 で「明日使える超高速 Ruby」を話してきた

require 'RXbyak'

rx = RXbyak.new
rx.mov :eax, [:esp, 8]        # mov eax, (esp+8)  // 第１引数のポインタ
rx.movq :xmm0, [:eax]         # movq xmm0, (eax)
rx.mov :eax, [:esp, 12]       # mov eax, (esp+12) // 第２引数のポインタ
rx.movq :xmm1, [:eax]         # movq xmm1, (eax)
rx.mulsd :xmm0, :xmm1         # mulsd xmm0, xmm1  // かけ算
rx.mov :eax, [:esp, 4]        # mov eax, (esp+4)  // 返値のポインタ
rx.movq [:eax], :xmm0         # movq (eax), xmm0
rx.ret                        # ret

puts rx.call(256.0, 256.0)    # => 65536.0
puts rx.call(123.45, 678.9)   # => 83810.205

……とまあ、(例によって)ちょっとお馬鹿なことを全力でやるニコニコ系ＬＴだったわけだが、この「どこからどうみても Ruby にしか見えない」キモかわいいコードを見ているうちに、なんかこれはこれでアリかな、という気になってくるｗ

Ruby が遅いのは承知の上で、それでも Ruby を使うのは、その人にとっては開発が短時間で済むとか、Ruby の記述の柔軟性とか、そういうポイントがあるわけだ。

だから、C/C++ で書けば速くなることはわかっていても(そしてそのスキルがあったとしても)、はいそうしましょう、というわけにはなかなかいかない。

それに本当に速くしたい部分はごくごく一部、他は Ruby のままでかまわない、ということも往々にしてある。

RXbyak はそんな「Ruby のままで速くなる方法」の一つになれる、かも？

他にも RubyInline という Ruby コードの中に C を書ける拡張があるが(知らなかったｗ。会場で id:akasata さんに教えてもらった)、JIT(動的生成) & アセンブラ(SIMD うはうは)というあたりは RXbyak の魅力かな。

あと、コンパイラ揃えるのが面倒でまだ試してないけど、RXbyak は Windows でも動く(はず)。

それから、この手の言語生成技術はただでさえややこしい＆見通しが悪くなりやすいので、できるだけ短く書ける、やわらかいコードで行う方がいい気がする。って、それなんて Ruby？

とまあ、そんな妄想を抱こうにも、いまはまだ実装率が低すぎて上のサンプルくらいしか動かないｗｗｗ

Ruby 拡張は初挑戦なのに、C++ ライブラリである Xbyak を組み込もうというのは少々無謀(Ruby 拡張は C で書くのが原則)。

しかも思いついて実装を始めたのが日曜日から、なぁんてタイトなスケジュールでは、サンプルが動いた後は資料を作るのがやっと。

そのうち、ニューラルネットワーク実装を RXbyak で動かすあたりまでは最低限作り込むつもりだけど、まずは明日の PRML 読書会の予習だ～。

(蛇足) ネタがわからない人向けの解説。

RXbyak は、アセンブラを動的に生成するコードが Ruby で書けるようになる拡張。上のサンプルコードは Ruby DSL で x86 アセンブラを書いている。
JIT とは、元となるプログラムなどから動的にネイティブコード(=マシン語)を生成すること。実行時に与えられるデータを使って最適化などできるので、C/C++ で素直に書くより速くなることが期待できる。正規表現の実装などでは普通に用いられている。
普通のプログラムは汎用的に(どのCPUでも)動くことを考慮して、CPU の新しめの機能を使い切ってはいない(現行のほとんどのパソコンが対応していても)。それらを使うにはアセンブラ(あるいは同等のコード)を書くしかない。