mirror of https://github.com/open-mmlab/mmcv.git
parent
44d871e7f3
commit
3c9d327cc7
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
|
@ -2,11 +2,4 @@
|
|||
#define MMCV_OPS_CSRC_COMMON__UTIL_HPP_
|
||||
const int SIZE = 8;
|
||||
|
||||
c10::SmallVector<int64_t, SIZE> array_to_vector(c10::IntArrayRef shape) {
|
||||
c10::SmallVector<int64_t, SIZE> shape_small_vec;
|
||||
for (uint64_t i = 0; i < shape.size(); i++) {
|
||||
shape_small_vec.emplace_back(shape[i]);
|
||||
}
|
||||
}
|
||||
|
||||
#endif // MMCV_OPS_CSRC_COMMON__UTIL_HPP_
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
#include "common_util.h"
|
||||
#include "pytorch_npu_helper.hpp"
|
||||
|
||||
using namespace NPU_NAME_SPACE;
|
||||
|
|
Loading…
Reference in New Issue