Vengineerの妄想(準備期間)

人生は短いけど、長いです。人生を楽しみましょう!

TensorFlow Runtime 、CUDA対応

@Vengineerの戯言 : Twitter
SystemVerilogの世界へようこそすべては、SystemC v0.9公開から始まった 

いつものように、tensorflow 関連のソースコードを眺めていたら、

TensorFlow Runtimeに、「CUDA in TensorFlow Runtime」なるドキュメントを見つけた

github.com

GPU関連のコードもアップし始めました。こちら。。

github.com

ここ (stream) の部分を見ると、

  • cublas
  • cudart
  • cudnn
  • cufft
  • cusolver

なんどのWrapperもありますね。

お、hip_stub.cc や hip_wrapper.cc もありますね。

ここに、こんな感じのことが書いてあるので HIPなんでしょうね。

# Generated API headers and implementations for HIP libraries.
# Note: Only gpu_stream_wrapper should depend on hip_stubs.