文件最后提交记录最后更新时间
6 天前
6 天前
README.md

ComplexMatDot算子

算子概述

ComplexMatDot(复数矩阵点乘)算子实现了两个复数矩阵的逐元素乘法运算,是 BLAS 基础线性代数库中的扩展算子之一。该算子针对复数运算特性进行了优化,使用 GatherMask 操作高效完成复数矩阵的逐元素乘法。

数学表达式:

result[i, j] = matx[i, j] * maty[i, j]

包含以下接口:

接口名 功能简述
aclblasComplexMatDot 复数矩阵逐元素点乘

算子执行接口

aclblasComplexMatDot

产品支持情况

  • Ascend 950PR / Ascend 950DT:不支持
  • Atlas A3 训练系列产品 / Atlas A3 推理系列产品:支持
  • Atlas A2 训练系列产品 / Atlas A2 推理系列产品:支持

函数原型

int aclblasComplexMatDot(const float *matx, const float *maty, float *result, const int64_t m, const int64_t n, void *stream)

参数说明

参数名 输入/输出 参数类型 说明
matx 输入 const float*(FP32) 复数矩阵,维度为 m x n,存储为 2mn 个 float,Device 内存
maty 输入 const float*(FP32) 复数矩阵,维度为 m x n,存储为 2mn 个 float,Device 内存
result 输出 float*(FP32) 复数矩阵,维度为 m x n,存储为 2mn 个 float,Device 内存
m 输入 int64_t 矩阵的行数,Host 内存
n 输入 int64_t 矩阵的列数,Host 内存
stream 输入 void* 执行流,Host 内存

约束说明