gpu - 分析 JAX 代码:什么是 redzone_checker,为什么要花这么多时间?
问题描述
我找到了这篇文章,但仍不清楚redzone_checker
内核在做什么以及为什么。具体来说,它是否应该占用我应用程序运行时间的 90% 以上?TensorBoard 报告说它占用了我的 JAX 代码的绝大部分运行时间,我想知道
- 实际上是这个内核花费了太多时间,还是这是使用 TensorBoard 分析 JAX 的副作用(即,输出在某种程度上具有误导性)?
- 有没有办法减少
redzone_checker
内核花费的时间?这甚至是个好主意吗?
提前感谢您的任何见解。
解决方案
在分析之前确保预热。
它可能是 jit 编译时间。
推荐阅读
- javascript - 在 React 中设置单个 setTimeout
- excel - 创建新选项卡
- json-ld - 使用 JSON-LD 上下文将来自不同数据源的多个值映射到常见 IRI
- ios - Swift 删除选项和不必要的字符 iOS
- javascript - JQuery:向所有孩子添加点击事件,不起作用
- android - 如何在 NDK r17+ 中支持具有 MIPS 架构的设备?
- dax - 标记 DAX 中排名最高的帐户
- python - Python:如何拆分字符串但保留非字母数字字符
- python-3.x - Python selenium - “没有模式对话框当前在 selenium 中打开异常”
- javascript - 如何让 DOM 元素的更新更高效?