Skip to content

Commit f4a2b82

Browse files
committed
torch_npu support aclnn
1 parent 5040148 commit f4a2b82

File tree

4 files changed

+601
-2
lines changed

4 files changed

+601
-2
lines changed

mmcv/ops/csrc/common/pytorch_npu_helper.hpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,12 @@
1818
#ifndef PYTORCH_NPU_HELPER_HPP_
1919
#define PYTORCH_NPU_HELPER_HPP_
2020

21-
#include <torch_npu/csrc/aten/CustomFunctions.h>
2221
#include <torch_npu/csrc/framework/utils/CalcuOpUtil.h>
2322
#include <torch_npu/csrc/framework/utils/OpAdapter.h>
2423

2524
#include "pytorch_cpp_helper.hpp"
2625
#include "pytorch_device_registry.hpp"
26+
#include "pytorch_npu_util.hpp"
2727

2828
#define NPU_NAME_SPACE at_npu::native
2929

0 commit comments

Comments
 (0)