AIに欠かせないGPU。Web標準のAPI、WebGPUに対応しているJavaScriptのランタイム、DenoでオープンソースのGPUを使った行列計算ライブラリTensorFlow.jsを動かしてみました。
import { tf } from "https://code4fukui.github.io/tf-deno/tf.js"; const a = tf.tensor1d([1, 2, 3, 4]); a.print();
denoコマンドを開き、上記を貼り付けるだけで動きます。
Initialization of backend webgpu failed TypeError: Failed to execute 'call' on 'GPUAdapter': 'requiredFeatures' of 'GPUDeviceDescriptor' (Argument 2), index 1 can not be converted to a enum
と、エラーがでる対策として、tf.js にて、requestDevice に手を入れるコードが入っています。
「test1.js on tf-deno」
ひとまずテンソル（ベクトルとスクラーを一般化したもの）を1つ作って表示するだけのプログラム。
ベンチマーク「bench1.js」を動かした結果です。1320x1320のランダムの行列を2つ作って掛け算する速度をWebGPUとCPUで比較します。M4 MacBook Airで、34倍速となりました。
「code4fukui/tf-deno」
Denoで手軽に開発できる環境ができました！
Fri Night
