Merged
【unfold_grad】修复issue问题 aclnnUnfoldGrad 头文件将输出参数 gradIn 修改为 @param [out] #6715
tangkaidiAdd on 16 days ago
【unfold_grad】修复issue问题 aclnnUnfoldGrad 头文件将输出参数 gradIn 修改为 @param [out] #6715
Merged
Total: 1 files changed+7-6
| @@ -31,22 +31,23 @@ extern "C" { | |||
| 31 | * @param [in] dim: npu host侧的int。 | 31 | * @param [in] dim: npu host侧的int。 |
| 32 | * @param [in] size: npu host侧的int。 | 32 | * @param [in] size: npu host侧的int。 |
| 33 | * @param [in] step: npu host侧的int。 | 33 | * @param [in] step: npu host侧的int。 |
| 34 | - * @param [in] gradIn: | 34 | + * @param [out] gradIn: |
| 35 | * npu device侧的aclTensor，数据类型支持BF16、FLOAT16、FLOAT。 | 35 | * npu device侧的aclTensor，数据类型支持BF16、FLOAT16、FLOAT。 |
| 36 | * 支持[非连续的Tensor](#)，数据格式支持ND。 | 36 | * 支持[非连续的Tensor](#)，数据格式支持ND。 |
| 37 | - * | 37 | + * |
| 38 | * @param [out] workspaceSize: 返回用户需要在npu device侧申请的workspace大小。 | 38 | * @param [out] workspaceSize: 返回用户需要在npu device侧申请的workspace大小。 |
| 39 | * @param [out] executor: 返回op执行器，包含了算子计算流程。 | 39 | * @param [out] executor: 返回op执行器，包含了算子计算流程。 |
| 40 | * @return aclnnStatus: 返回状态码。 | 40 | * @return aclnnStatus: 返回状态码。 |
| 41 | */ | 41 | */ |
| 42 | -ACLNN_API aclnnStatus aclnnUnfoldGradGetWorkspaceSize( | 42 | +ACLNN_API aclnnStatus aclnnUnfoldGradGetWorkspaceSize(const aclTensor* gradOut, const aclIntArray* inputSizes, |
| 43 | - const aclTensor* gradOut, const aclIntArray* inputSizes, int64_t dim, int64_t size, int64_t step, | 43 | + int64_t dim, int64_t size, int64_t step, const aclTensor* gradIn, |
| 44 | - const aclTensor* gradIn, uint64_t* workspaceSize, aclOpExecutor** executor); | 44 | + uint64_t* workspaceSize, aclOpExecutor** executor); |
| 45 | 45 | ||
| 46 | /** | 46 | /** |
| 47 | * @brief aclnnUnfoldGrad的第二段接口，用于执行计算。 | 47 | * @brief aclnnUnfoldGrad的第二段接口，用于执行计算。 |
| 48 | */ | 48 | */ |
| 49 | -ACLNN_API aclnnStatus aclnnUnfoldGrad(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream); | 49 | +ACLNN_API aclnnStatus aclnnUnfoldGrad(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, |
| 50 | + aclrtStream stream); | ||
| 50 | 51 | ||
| 51 | #ifdef __cplusplus | 52 | #ifdef __cplusplus |
| 52 | } | 53 | } |