torch_npu support aclnn

open-mmlab · Nov 29, 2023 · a634292 · a634292
1 parent ed4303e
commit a634292
Show file tree

Hide file tree

Showing 4 changed files with 594 additions and 6 deletions.
diff --git a/mmcv/ops/csrc/common/pytorch_npu_helper.hpp b/mmcv/ops/csrc/common/pytorch_npu_helper.hpp
@@ -23,6 +23,7 @@
 
 #include "pytorch_cpp_helper.hpp"
 #include "pytorch_device_registry.hpp"
+#include "pytorch_npu_util.hpp"
 
 #define NPU_NAME_SPACE at_npu::native