mirror of https://github.com/open-mmlab/mmcv.git
27 lines
954 B
C++
27 lines
954 B
C++
/*************************************************************************
|
|
* Copyright (C) 2021 Cambricon.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
|
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
|
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*************************************************************************/
|
|
#ifndef PYTORCH_MLU_HELPER_HPP_
|
|
#define PYTORCH_MLU_HELPER_HPP_
|
|
|
|
#ifdef MMCV_WITH_MLU
|
|
#include "aten.h"
|
|
|
|
#define NFU_ALIGN_SIZE 128
|
|
|
|
#define PAD_UP(x, y) (((x) / (y) + (int)((x) % (y) > 0)) * (y))
|
|
|
|
#define PAD_DOWN(x, y) (((x) / (y)) * (y))
|
|
|
|
#endif
|
|
|
|
#endif // PYTORCH_MLU_HELPER_HPP_
|