-
Add-uint-support ：给 PyTorch 算子补 uint 支持，是一个被低估的精细活
大部分写 PyTorch 的人一辈子不会碰“给算子加类型支持”这种事。你用 tensor.float()、tensor.int()，内核里那些 AT_DISPATCH 宏替你扛下了所有类型分支，你甚至感觉不到它们存在。直到某天你要跑一个 uint16 的张量。 然后你会撞上一堵墙。PyTorch 的 uint8（对应 kByte）支持得很完整，但 uint16、uint32、uint64 这三个所…- 0
- 0