operator==
功能
判断参数是否相同。
定义
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | inline bool operator==(const GenAttentionMaskParam &left, const GenAttentionMaskParam &right) { return left.headNum == right.headNum && left.seqLen == right.seqLen; } inline bool operator==(const RopeGradParam &left, const RopeGradParam &right) { return left.qSeqLen == right.qSeqLen; } inline bool operator==(const FastSoftMaxParam &left, const FastSoftMaxParam &right) { return left.headNum == right.headNum && left.qSeqLen == right.qSeqLen; } inline bool operator==(const FastSoftMaxGradParam &left, const FastSoftMaxGradParam &right) { return left.headNum == right.headNum && left.qSeqLen == right.qSeqLen; } inline bool operator==(const StridedBatchMatmulParam &left, const StridedBatchMatmulParam &right) { return left.transposeA == right.transposeA && left.transposeB == right.transposeB && left.batch == right.batch && left.headNum == right.headNum && left.m == right.m && left.n == right.n && left.k == right.k && left.lda == right.lda && left.ldb == right.ldb && left.ldc == right.ldc && left.strideA == right.strideA && left.strideB == right.strideB && left.strideC == right.strideC; } inline bool operator==(const UnpadWithHiddenStateParam &left, const UnpadWithHiddenStateParam &right) { return left.qSeqLen == right.qSeqLen && left.maxSeqLen == right.maxSeqLen; } inline bool operator==(const PadWithHiddenStateParam &left, const PadWithHiddenStateParam &right) { return left.qSeqLen == right.qSeqLen && left.maxSeqLen == right.maxSeqLen; } |