pytorch/torch_npu/csrc/libs · yanzi2021/pytorch - AtomGit

6f15099d创建于 2025年9月2日历史提交

文件	最后提交记录	最后更新时间
CMakeLists.txt	add npu device init API for libtorch	3 年前
README.md	!2885 [Feature] add ci for libtorch_npu. Merge pull request !2885 from huangyunlong/2.0ci	3 年前
init_npu.cpp	!24312 lazy setdevice Merge pull request !24312 from huangyunlong/2.9ls	8 个月前
init_npu.h	!8532 [fix]add finalize_npu for libtorch_npu. Merge pull request !8532 from huangyunlong/2.2lib	2 年前
torch_npu.h	[Feature]rename head file and add cmake for libtorch_npu	2 年前

caution:

This folder is used to export interfaces related to NPU, such as operators, device init and so on. And it will only be built in libtorch_npu without Python, which is used in inference with NPU.