89
csrc/common/stub/op_api/aclnn_kernels/contiguous.h
Normal file
89
csrc/common/stub/op_api/aclnn_kernels/contiguous.h
Normal file
@@ -0,0 +1,89 @@
|
||||
/**
|
||||
* 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
|
||||
Reference in New Issue
Block a user