trunk/4109cbc8d05d541dd0a7a2feac4490bef16d43c5: torch/_native: allow overrides on namespaces other than aten (#194509)
- PyTorch 近 90 天出现 1219 次
- PyTorch 近 90 天第 1205 次发版
- 上一次:同一天稍早 · trunk/1579dafc271f285bc780fd69264e6a2d49cd342c: [CI] Enable ACL in ARM64 CUDA images (#197741)
发生了什么
torch._native could override aten ops only. Since linear_cross_entropy lives in torch/nn and its chunked ops are registered in the torch_nn namespace, the registry has to accept any namespace, which is what this commit does. The override registry could only install kernels on the aten namespace: register_op_override rejected any other lib_symbol , and the internals hardcoded aten when resolving the op, cloning its s…
摘要按规则整理自下方来源原文