Skip to content

Commit

Permalink
torch_npu support aclnn
Browse files Browse the repository at this point in the history
  • Loading branch information
momo609 committed Nov 24, 2023
1 parent 5040148 commit e387da8
Show file tree
Hide file tree
Showing 4 changed files with 570 additions and 2 deletions.
2 changes: 1 addition & 1 deletion mmcv/ops/csrc/common/pytorch_npu_helper.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,12 @@
#ifndef PYTORCH_NPU_HELPER_HPP_
#define PYTORCH_NPU_HELPER_HPP_

#include <torch_npu/csrc/aten/CustomFunctions.h>
#include <torch_npu/csrc/framework/utils/CalcuOpUtil.h>
#include <torch_npu/csrc/framework/utils/OpAdapter.h>

#include "pytorch_cpp_helper.hpp"
#include "pytorch_device_registry.hpp"
#include "pytorch_npu_util.hpp"

#define NPU_NAME_SPACE at_npu::native

Expand Down
Loading

0 comments on commit e387da8

Please sign in to comment.