cosdt / torch_backend

https://cosdt.github.io/torch_backend/
Other
3 stars 0 forks source link

分工 #22

Closed FFFrog closed 3 months ago

FFFrog commented 3 months ago

0、autoload torch.npu & torch_npu.npu 1、NPUEvent.h -> 元昊 2、StreamGuard.h -> 逢春 3、NPUGuardImpl.cpp -> 元昊 4、HasCompatibleShallowCopyType.cpp -> warnging 逢春 5、VariableFallbackKernel.cpp -> 李佳伟(Done) 6、BinaryOps.cpp -> 李佳伟(Done) 7、torch_npu/optim 删除 -> 泽升 8、testing删除以及新的test framework -> 泽升 9、torch_npu/utils -> 李佳伟 10、NPUPluggableAllocator.h删除 -> 贞斌 11、torch_npu/csrc/Module.cpp 单独拆出来(torch_npu/csrc/.../xxx, torch_npu/npu/xxx) -> ALL 12、npu+op-plugin 新建个仓库 - 李佳伟 13、文档 - ALL 14、namespace 统一 15、DeviceProp 下移到 csrc/npu下(可以参考XPU实现) - 元昊 16、NPU LOG 17、Refactor AutocastMode.cpp(官方提供了默认列表,对比差异,如果一样可以直接调用宏)(社区多,搞不了)- 逢春