Enable ConcreteFunction XLA JIT
还没有人认领这个 Issue。
- 主要语言
- 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 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库，在一个分支上完成修改。
- 提交 Pull Request，并在描述里引用这个 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