Tests are slow: use vjvp ?
还没有人认领这个 Issue。
评估
- 难度
- 4/5
- 预计耗时
- 3-5 天
- 新手友好度
- 25/100
- Issue 类型
- 重构
- 描述清晰度
- 需要澄清
- 活跃度
- 停滞
- 技术栈
- julia
调研方向
未指定源文件。首先使用 rrule_via_ad 重现 Zygote 的 gradtest 与 CRTU 的 test_rrule 之间的 @btime 对比,然后跟踪相关测试工具中的内存分配和计时。完成的结果应包括对这一差距的解释,并在可行的情况下,实现可测量的 ChainRules 测试运行时间缩减。
由索引模型根据 Issue 内容生成。
描述
Compare Zygote's gradtest
julia> @btime gradtest(x -> sum(abs2, x), randn(4, 3, 2))
17.977 μs (236 allocations: 5.86 KiB)
and CRTUs test_rrule
@btime test_rrule(Zygote.ZygoteRuleConfig(), x -> sum(abs2, x), randn(4, 3, 2); rrule_f=rrule_via_ad)
1.490 ms (6884 allocations: 480.10 KiB)
it's nearly 100x slower. Do we understand why? I didn't have time to look into it so just making an issue.
ChainRules tests take a pretty long time to run so this might be worth improving.
- 主要语言
- Julia
- 星标
- 50
- 派生
- 18
- PR 合并指标
- 30 天内没有已合并 PR
环境准备
这个项目没有提供开发容器、Dockerfile 或贡献指南,环境需要你自己搭建:先看它的 README,通用步骤见我们的新手贡献指南。
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 Pull Request,并在描述里引用这个 Issue 编号。
JuliaDiff/ChainRulesTestUtils.jl 的其他 Issue
-
难度 2/5 1-3 小时 新手友好度 65/100
JuliaDiff/ChainRulesTestUtils.jl#148 · 1 个 reaction ·
-
documentation good first issue
难度 2/5 1-3 小时 新手友好度 62/100
-
难度 3/5 1-2 天 新手友好度 45/100
-
难度 2/5 1-3 小时 新手友好度 45/100
-
mutation
难度 3/5 1-2 天 新手友好度 45/100
查看 JuliaDiff/ChainRulesTestUtils.jl 的全部 Issue
相似的 Issue
-
enhancement good first issue help wanted
难度 2/5 半天 新手友好度 82/100
-
authority-vechtstromen ready-for-agent
难度 1/5 1 小时以内 新手友好度 75/100
Deltares/Ribasim-NL#890 ·
维护者通常 1 天内回复
-
难度 1/5 1 小时以内 新手友好度 88/100
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 78/100
JuliaLinearAlgebra/GenericLinearAlgebra.jl#183 · 1 条评论 ·
-
难度 2/5 1-3 小时 新手友好度 72/100
JuliaGPU/KernelAbstractions.jl#897 ·
维护者通常 1 天内回复