90 lines
2.9 KiB
C
90 lines
2.9 KiB
C
|
/**
|
|||
|
|
* Copyright (c) 2025 Huawei Technologies Co., Ltd.
|
|||
|
|
* This program is free software, you can redistribute it and/or modify it under the terms and conditions of
|
|||
|
|
* CANN Open Software License Agreement Version 2.0 (the "License").
|
|||
|
|
* Please refer to the License for details. You may not use this file except in compliance with the License.
|
|||
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
|
|||
|
|
* INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
|
|||
|
|
* See LICENSE in the root of the software repository for the full text of the License.
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#ifndef COMMON_INC_EXTERNAL_ACLNN_KERNELS_CONTIGUOUS_H
|
|||
|
|
#define COMMON_INC_EXTERNAL_ACLNN_KERNELS_CONTIGUOUS_H
|
|||
|
|
|
|||
|
|
#include "opdev/op_def.h"
|
|||
|
|
#include "opdev/common_types.h"
|
|||
|
|
|
|||
|
|
namespace l0op {
|
|||
|
|
|
|||
|
|
typedef struct {
|
|||
|
|
// 每个op::Shape 18ns
|
|||
|
|
int64_t viewOffset;
|
|||
|
|
|
|||
|
|
// Transpose
|
|||
|
|
op::Shape transposeSrcShape;
|
|||
|
|
op::Shape transposeDstShape;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> perm;
|
|||
|
|
|
|||
|
|
// broadcast to
|
|||
|
|
op::Shape broadcastSrcShape;
|
|||
|
|
op::Shape broadcastDstShape;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> shape;
|
|||
|
|
|
|||
|
|
// slice
|
|||
|
|
op::Shape sliceSrcShape;
|
|||
|
|
op::Shape sliceDstShape;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> offset;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> size;
|
|||
|
|
|
|||
|
|
// strided slice
|
|||
|
|
op::Shape stridedsliceSrcShape;
|
|||
|
|
op::Shape stridedsliceDstShape;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> begin;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> end;
|
|||
|
|
op::FVector<int64_t, op::MAX_DIM_NUM> strides;
|
|||
|
|
|
|||
|
|
// optimizer
|
|||
|
|
bool mayBroadcast;
|
|||
|
|
bool mayTranspose;
|
|||
|
|
bool maySlice;
|
|||
|
|
bool mayStridedslice;
|
|||
|
|
} ContiguousParam;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 将非连续Tensor转换为连续Tensor
|
|||
|
|
* @param x
|
|||
|
|
* @param executor
|
|||
|
|
* @return aclTensor 转换后的tensor
|
|||
|
|
*/
|
|||
|
|
const aclTensor* Contiguous(const aclTensor* x, aclOpExecutor* executor);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 将连续tensor拷贝到非连续的tensor上
|
|||
|
|
* @param x
|
|||
|
|
* @param y
|
|||
|
|
* @param executor
|
|||
|
|
* @return aclTensor 转换后的tensor
|
|||
|
|
*/
|
|||
|
|
const aclTensor* ViewCopy(const aclTensor* x, const aclTensor* y, aclOpExecutor* executor);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 对Tensor创建一个View，要求Tensor满足PickView的条件
|
|||
|
|
* @param x 输入Tensor，可以是一整块的非连续Tensor
|
|||
|
|
* @param executor
|
|||
|
|
* @return 输出Shape是一个连续Tensor
|
|||
|
|
*/
|
|||
|
|
const aclTensor* PickViewAsContiguous(const aclTensor* x, aclOpExecutor* executor);
|
|||
|
|
|
|||
|
|
const aclTensor* ReViewToOut(const aclTensor* x, const aclTensor* y, aclOpExecutor* executor);
|
|||
|
|
|
|||
|
|
// ============内部接口=============
|
|||
|
|
bool CanOptimizeContiguous(
|
|||
|
|
const op::Shape& viewShape, const op::Strides& strides, int64_t offset, int64_t storageSize,
|
|||
|
|
ContiguousParam& param);
|
|||
|
|
|
|||
|
|
bool CanOptimizeView(const op::Shape& viewShape, const op::Strides& strides, int64_t offset, ContiguousParam& param);
|
|||
|
|
// ============内部接口=============
|
|||
|
|
} // namespace l0op
|
|||
|
|
|
|||
|
|
#endif // COMMON_INC_EXTERNAL_ACLNN_KERNELS_CONTIGUOUS_H
|