Rust in Everything
基于 burn 0.22.1

导读

动手对照之前,先讲清楚两件事:这一卷的数字怎么来,以及值不值得转。

  1. 怎么读这一卷:锁定的版本和 burn 的 feature;每章的结构;一对对照程序怎么跑、怎么用 compare.py 逐个数比较;为什么 f32 下两边的结果不逐位相同,判断用绝对误差、容差取 1e-6 的依据;三条主线每一步落在哪一页。
  2. 为什么要转:同一个 MLP 推理,Python 加 torch 约 729 MiB、启动约 0.65 秒、峰值 RSS 约 208 MiB,burn 是一个 2.3 MiB 的二进制、启动约 3 毫秒、峰值 RSS 约 2.6 MiB;也写了不值得转的情况:unique、searchsorted、bincount、linalg.inv 在 burn 0.22.1 里找不到,svd 不能求导。
flowchart TB
  read[怎么读这一卷] --> ver[锁定版本]
  read --> run["对照怎么跑"]
  read --> tol["对得上的标准"]
  read --> lines[三条主线]
  why[为什么要转] --> size["体积 启动 内存"]
  why --> cov[算子覆盖]
  why --> notworth[不值得转的情况]

💬 关联讨论

发起讨论
加载中...