文件名称：LLVM 的高性能自动微分。
文件大小：1.1MB
文件格式：ZIP
更新时间：2024-07-18 12:41:26
c rust machine-learning deep-learning enzyme
LLVM的Enzyme高性能自动微分器Enzyme 是一个插件，可以对静态可分析的 LLVM 执行自动微分 (AD)。可以通过对要区分的函数调用__enzyme_autodiff来使用酶，如下所示。运行酶转换过程然后用其第一个参数的梯度替换对__enzyme_autodiff的调用。doublefoo (double );doublegrad_foo (double x) {return__enzyme_autodiff (foo, x); }Enzyme 是高效的，它对优化代码执行 AD 的能力使 Enzyme 能够达到或超过最先进的 AD 工具的性能。可以在我们的网站上找到有关安装和使用 Enzyme 的详细信息：https ://enzyme.mit.edu 。下面是一个关于如何安装 Enzyme 的简短示例：cd /path/to/Enzyme/enzyme mkdir build && cd build cmake -G Ninja .. -DLLVM_DIR=/path/to/llvm/lib/cmake/llvm -DLLVM_EXTERNAL_LIT=/