Enable ConcreteFunction XLA JIT
まだ誰も着手していません。
- 主要言語
- Java
- スター
- 928
- フォーク
- 227
- PR マージ指標
- 30日以内にマージされた PR はありません
説明
This is mostly a reminder for myself, but also documentation if someone else ends up doing it.
After reading through the python code (see here), I found that to force function JITing you need to set the _XlaMustCompile and _noinline attributes to true, as is done here.
However, this results in an error. I didn't run it down at the time, but it was reported in https://github.com/tensorflow/tensorflow/issues/50458. Once that is fixed we should try this again.
コントリビューションガイド
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
調査の方向性
まず tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/ConcreteFunction.java の475行目付近を確認し、次に tensorflow/python/eager/def_function.py の710行目付近と動作を比較します。報告されたエラーが解決されているかどうかを判断するために、TensorFlow issue 50458 を確認します。ConcreteFunction XLA JIT をそのエラーなしで有効化でき、動作が検証されれば完了です。
索引モデルが issue の本文から書いたものです。
評価
- 技術スタック
- java
- 領域
- api, machine-learning
- issue の種類
- 機能追加
- 難易度
- 4/5
- 見積もり時間
- 3〜5日
- 活発さ
- 停滞
- 明瞭さ
- 説明が足りない
- 初心者へのやさしさ
- 25/100