@Vengineerの戯言 : Twitter SystemVerilogの世界へようこそ、すべては、SystemC v0.9公開から始まった
はじめに
TensorFlow Blog の 下記のTweet で TensorFlow Pluggable Device なるものを知りました。
🔌📱 Introducing PluggableDevice!
— TensorFlow (@TensorFlow) 2021年6月7日
PluggableDevice is a scalable device support mechanism, letting new devices connect to TensorFlow as plug-ins through C API. Now available in TF 2.5.
More info ↓ https://t.co/5IkSygrsD3
TensorFlow Blog には、「PluggableDevice: Device Plugins for TensorFlow」というエントリもアップされています。
Tensorflow の tensorflow/compiler/plugin ディレクトリに、各社のPluginを置くのだか、そのPluginのルールが明確ではなかった。 この PluggableDevice では、その点を明確にしたのだと思います。提案したのは、Intelの中の人、つまり、Intel のGPUをTensorFlowに組み込むために行ったことなのでしょうかね。
誰かが使っているの?
ということで、Tensorflow PluggableDevice で Google 君に聞いたら、
というのが見つかりました。AppleのMetal対応のPluggableDeviceのようです。
制限はあるものの、x86 : AMD と arm64 : M1 にて利用できるようですね。
Appleの github に、apple/tensorflow_macos なるものが残っていました。
こんなのもあった。
おわりに
新しいデバイスをTensorFlowに対応するための PluggableDevice なるものができて、既に Apple Metal は対応済みということでした。
Intel が対応したのがいつ出てくるのだろうか?