mindspore 2.3.0__cp39-none-any.whl → 2.3.0rc2__cp39-none-any.whl

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/asd_utils.h

@@ -1,84 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#ifndef MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_UTILS_ASD_UTILS_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_UTILS_ASD_UTILS_H_
-
-#include "asdops/types.h"
-#include "utils/log/log.h"
-#include "acme/include/base_type.h"
-
-namespace mindspore {
-namespace acme {
-inline AsdOps::TensorDType ToAsdType(DataType type) {
-  switch (type) {
-    case kTypeFloat16:
-      return AsdOps::TENSOR_DTYPE_FLOAT16;
-    case kTypeFloat32:
-      return AsdOps::TENSOR_DTYPE_FLOAT;
-    case kTypeFloat64:
-      return AsdOps::TENSOR_DTYPE_DOUBLE;
-    case kTypeUint8:
-      return AsdOps::TENSOR_DTYPE_UINT8;
-    case kTypeUint16:
-      return AsdOps::TENSOR_DTYPE_UINT16;
-    case kTypeUint32:
-      return AsdOps::TENSOR_DTYPE_UINT32;
-    case kTypeBF16:
-      return AsdOps::TENSOR_DTYPE_BF16;
-    case kTypeInt8:
-      return AsdOps::TENSOR_DTYPE_INT8;
-    case kTypeInt16:
-      return AsdOps::TENSOR_DTYPE_INT16;
-    case kTypeInt32:
-      return AsdOps::TENSOR_DTYPE_INT32;
-    case kTypeBool:
-      // I dont't known why..
-      return AsdOps::TENSOR_DTYPE_INT8;
-    default:
-      MSOP_LOG(EXCEPTION) << "Unsupported type: " << type;
-      return AsdOps::TENSOR_DTYPE_UNDEFINED;
-  }
-}
-
-inline AsdOps::TensorFormat ToAsdFormat(TensorFormat format) {
-  switch (format) {
-    case kFormatUnknown:
-      return AsdOps::TENSOR_FORMAT_UNDEFINED;
-    case kFormatNCHW:
-      return AsdOps::TENSOR_FORMAT_NCHW;
-    case kFormatND:
-      return AsdOps::TENSOR_FORMAT_ND;
-    case kFormatNHWC:
-      return AsdOps::TENSOR_FORMAT_NHWC;
-    default:
-      MSOP_LOG(EXCEPTION) << "Unsupported format: " << format;
-      return AsdOps::TENSOR_FORMAT_UNDEFINED;
-  }
-}
-
-inline AsdOps::SVector<int64_t> ToAsdDims(const ShapeInfo &shape) {
-  AsdOps::SVector<int64_t> asd_dims;
-  for (auto s : shape) {
-    asd_dims.emplace_back(s);
-  }
-  return asd_dims;
-}
-
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_UTILS_ASD_UTILS_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/comm_utils.h

@@ -1,61 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#ifndef MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_UTILS_COMM_UTILS_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_UTILS_COMM_UTILS_H_
-
-#include <unordered_map>
-#include "acme/include/base_type.h"
-#include "utils/log/log.h"
-
-#define CeilDiv(dividend, divisor) (((divisor) == 0) ? 0 : (((dividend) + (divisor)-1) / (divisor)))
-#define UpRound(in, round) ((((in) + (round)-1) / (round)) * (round))
-
-namespace mindspore {
-namespace acme {
-inline size_t GetTypeSize(DataType type) {
-  static const std::unordered_map<DataType, size_t> kTypeSize = {
-    {kTypeFloat16, sizeof(float) / 2},
-    {kTypeFloat32, sizeof(float)},
-    {kTypeFloat64, sizeof(double)},
-    {kTypeInt8, sizeof(int8_t)},
-    {kTypeInt16, sizeof(int16_t)},
-    {kTypeInt32, sizeof(int32_t)},
-    {kTypeInt64, sizeof(int64_t)},
-    {kTypeUint8, sizeof(uint8_t)},
-    {kTypeUint16, sizeof(uint16_t)},
-    {kTypeUint32, sizeof(uint32_t)},
-    {kTypeUint64, sizeof(uint64_t)},
-    {kTypeBF16, sizeof(float) / 2},
-    {kTypeBool, sizeof(bool)},
-    {kTypeComplex64, 64},
-    {kTypeComplex128, 128},
-  };
-
-  auto it = kTypeSize.find(type);
-  if (it == kTypeSize.end()) {
-    MSOP_LOG(EXCEPTION) << "Unsupported type: " << type;
-    return 0;
-  }
-
-  return it->second;
-}
-
-bool IsOpEnabled(const std::string &op_name);
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_UTILS_COMM_UTILS_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp32.h

@@ -1,224 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-#ifndef ROTARY_POS_EMB_FP32
-#define ROTARY_POS_EMB_FP32
-#include "apply_rotary_pos_emb_base.h"
-template <typename QK_DTYPE, typename COS_DTYPE, bool IF_COS_BROADCAST>
-class RopeFp32 : public RopeBase<QK_DTYPE, COS_DTYPE, IF_COS_BROADCAST> {
- public:
-  __aicore__ inline RopeFp32(RopeTilingData *tilingData) : RopeBase<QK_DTYPE, COS_DTYPE, IF_COS_BROADCAST>(tilingData) {
-    this->repeatSize_ = 64;                   // 64 = 256B / sizeof(float)
-    this->maxProcessNum_ = 3 * MAX_LEN_FP16;  // 3 for fp16 space needed
-    this->repeatTimesQ_ = (this->tilingData_->hiddenSizeQ + this->repeatSize_ - 1) / this->repeatSize_;
-    this->repeatTimesK_ = (this->tilingData_->hiddenSizeK + this->repeatSize_ - 1) / this->repeatSize_;
-    headDimAlign_ = ((this->tilingData_->headDim + ELE_NUM_FP32 - 1) / ELE_NUM_FP32) * ELE_NUM_FP32;
-    this->alignHalfHeadDim_ = (this->rotateStride_ * NUM_TWO) % ELE_NUM_FP32;
-    this->hiddenSizeAlign_ = ((this->hiddenSize_ + this->repeatSize_ - 1) / this->repeatSize_) * this->repeatSize_;
-    this->syncOffset_ =
-      (this->tilingData_->headDim % ELE_NUM_FP16 == 0) ? this->hiddenSizeAlign_ : this->headNum_ * headDimAlign_;
-    this->offsetExtraGm_ = NUM_TWO * block_idx * this->syncOffset_;
-
-    sliceSizeTmp_ = (SLICE_SIZE / this->tilingData_->headDim) * this->tilingData_->headDim;  // 向下取整
-
-    // fp16
-    this->oriPos_ = 0;
-    this->removeBefore_ = this->oriPos_ + sliceSizeTmp_;
-    this->padBefore_ = this->removeBefore_ + sliceSizeTmp_;
-    resOut_ = this->padBefore_ + sliceSizeTmp_;
-
-    // fp32
-    this->cosPad_ = 0;
-    this->sinPad_ = this->cosPad_ + sliceSizeTmp_;
-    this->negOne_ = this->sinPad_ + sliceSizeTmp_;
-    oriPosF32_ = this->negOne_ + sliceSizeTmp_;
-    PadBeforeF32_ = oriPosF32_ + sliceSizeTmp_;
-    removeBeforeF32_ = PadBeforeF32_ + sliceSizeTmp_;
-    posOneF32_ = removeBeforeF32_ + sliceSizeTmp_;
-    resOutFp32_ = posOneF32_ + sliceSizeTmp_;
-
-    this->pipe_.InitBuffer(qkfp32QueueCO2_, 1,
-                           (this->tilingData_->maxUbSize - this->batchSize_ * NUM_TWO -
-                            this->maxProcessNum_ * sizeof(QK_DTYPE)));  // 留給fp32的
-    AscendC::LocalTensor<COS_DTYPE> qkfp32_perloop_ub = qkfp32QueueCO2_.AllocTensor<COS_DTYPE>();
-    qkfp32Ubuf_ = (__ubuf__ COS_DTYPE *)qkfp32_perloop_ub.GetPhyAddr();
-    this->pipe_.InitBuffer(outQueueCO2_, 1, ((this->maxProcessNum_) * sizeof(QK_DTYPE)));
-    AscendC::LocalTensor<QK_DTYPE> cache_perloop_ub1 = outQueueCO2_.AllocTensor<QK_DTYPE>();
-    commonUbuf_ = (__ubuf__ QK_DTYPE *)cache_perloop_ub1.GetPhyAddr();
-
-    // 判断
-    if (this->tilingData_->hiddenSizeQ > sliceSizeTmp_) {
-      sliceTimeQ_ = (this->tilingData_->hiddenSizeQ + sliceSizeTmp_ - 1) / sliceSizeTmp_;    // 向上取整
-      lastSliceSizeQ_ = this->tilingData_->hiddenSizeQ - (sliceTimeQ_ - 1) * sliceSizeTmp_;  // 向上取整
-    } else {
-      sliceTimeQ_ = 1;
-      lastSliceSizeQ_ = this->tilingData_->hiddenSizeQ;
-    }
-
-    if (this->tilingData_->hiddenSizeK > sliceSizeTmp_) {
-      sliceTimeK_ = (this->tilingData_->hiddenSizeK + sliceSizeTmp_ - 1) / sliceSizeTmp_;  // 向上取整
-      lastSliceSizeK_ = this->tilingData_->hiddenSizeK - (sliceTimeK_ - 1) * sliceSizeTmp_;
-    } else {
-      sliceTimeK_ = 1;
-      lastSliceSizeK_ = this->tilingData_->hiddenSizeK;
-    }
-  }
-
-  template <typename T>
-  __aicore__ inline void CastQKDType2F32(uint32_t repeatTimes) {
-    vconv_f162f32(qkfp32Ubuf_ + oriPosF32_, commonUbuf_ + this->oriPos_, repeatTimes, 1, 1, DEFAULT_REPEAT_STRIDE,
-                  DEFAULT_REPEAT_STRIDE / NUM_TWO);
-    vconv_f162f32(qkfp32Ubuf_ + removeBeforeF32_, commonUbuf_ + this->removeBefore_, repeatTimes, 1, 1,
-                  DEFAULT_REPEAT_STRIDE, DEFAULT_REPEAT_STRIDE / NUM_TWO);
-    vconv_f162f32(qkfp32Ubuf_ + PadBeforeF32_, commonUbuf_ + this->padBefore_, repeatTimes, 1, 1, DEFAULT_REPEAT_STRIDE,
-                  DEFAULT_REPEAT_STRIDE / NUM_TWO);
-  }
-
-  template <>
-  __aicore__ inline void CastQKDType2F32<bfloat16_t>(uint32_t repeatTimes) {
-    vconv_bf162f32(qkfp32Ubuf_ + oriPosF32_, commonUbuf_ + this->oriPos_, repeatTimes, 1, 1, DEFAULT_REPEAT_STRIDE,
-                   DEFAULT_REPEAT_STRIDE / NUM_TWO);
-    vconv_bf162f32(qkfp32Ubuf_ + removeBeforeF32_, commonUbuf_ + this->removeBefore_, repeatTimes, 1, 1,
-                   DEFAULT_REPEAT_STRIDE, DEFAULT_REPEAT_STRIDE / NUM_TWO);
-    vconv_bf162f32(qkfp32Ubuf_ + PadBeforeF32_, commonUbuf_ + this->padBefore_, repeatTimes, 1, 1,
-                   DEFAULT_REPEAT_STRIDE, DEFAULT_REPEAT_STRIDE / NUM_TWO);
-  }
-
-  template <typename T>
-  __aicore__ inline void CastF322QKDType(__gm__ QK_DTYPE *dst, __ubuf__ QK_DTYPE *src1, __ubuf__ float *src,
-                                         uint32_t repeatTimes, uint32_t hiddenSize1) {
-    vconv_f322f16(src1, src, repeatTimes, 1, 1, DEFAULT_REPEAT_STRIDE / NUM_TWO, DEFAULT_REPEAT_STRIDE);
-    set_flag(PIPE_V, PIPE_MTE3, EVENT_ID1);
-
-    wait_flag(PIPE_V, PIPE_MTE3, EVENT_ID1);
-    copy_ubuf_to_gm(dst, src1, 0, 1, hiddenSize1 / ELE_NUM_FP16, 0, 0);
-  }
-
-  template <>
-  __aicore__ inline void CastF322QKDType<bfloat16_t>(__gm__ QK_DTYPE *dst, __ubuf__ QK_DTYPE *src1, __ubuf__ float *src,
-                                                     uint32_t repeatTimes, uint32_t hiddenSize1) {
-    vconv_f322bf16r(src1, src, repeatTimes, 1, 1, DEFAULT_REPEAT_STRIDE / NUM_TWO, DEFAULT_REPEAT_STRIDE);
-    set_flag(PIPE_V, PIPE_MTE3, EVENT_ID1);
-
-    wait_flag(PIPE_V, PIPE_MTE3, EVENT_ID1);
-    copy_ubuf_to_gm(dst, src1, 0, 1, hiddenSize1 / ELE_NUM_FP16, 0, 0);
-  }
-  __aicore__ inline void Process(__gm__ uint8_t *extraGm) {
-    if (this->tilingData_->cosFormat == 1) {
-      pipe_barrier((PIPE_ALL));
-      this->ExpandCosSin(qkfp32Ubuf_, this->cosGm_, (__gm__ COS_DTYPE *)extraGm);
-      this->cosGm_ = (__gm__ COS_DTYPE *)extraGm;
-      pipe_barrier((PIPE_ALL));
-      this->ExpandCosSin(qkfp32Ubuf_, this->sinGm_,
-                         (__gm__ COS_DTYPE *)extraGm + this->tilingData_->ntokens * this->tilingData_->headDim);
-      this->sinGm_ = (__gm__ COS_DTYPE *)extraGm + this->tilingData_->ntokens * this->tilingData_->headDim;
-      extraGm =
-        extraGm + this->tilingData_->ntokens * this->tilingData_->headDim * 4;  // sizeof(uint8_t) * 2 = sizeof(half)
-      pipe_barrier((PIPE_ALL));
-    }
-    uint32_t headNumTempQ = this->tilingData_->hiddenSizeQ > sliceSizeTmp_
-                              ? (sliceSizeTmp_ / this->tilingData_->headDim)
-                              : this->tilingData_->headNumQ;
-    uint32_t dynamicSliceQ =
-      this->tilingData_->hiddenSizeQ > sliceSizeTmp_ ? sliceSizeTmp_ : this->tilingData_->hiddenSizeQ;
-    uint32_t headNumTempK = this->tilingData_->hiddenSizeK > sliceSizeTmp_
-                              ? (sliceSizeTmp_ / this->tilingData_->headDim)
-                              : this->tilingData_->headNumK;
-    uint32_t dynamicSliceK =
-      this->tilingData_->hiddenSizeK > sliceSizeTmp_ ? sliceSizeTmp_ : this->tilingData_->hiddenSizeK;
-    uint32_t repeatTemp = (dynamicSliceQ + this->repeatSize_ - 1) / this->repeatSize_;
-    this->ExpandNeg(qkfp32Ubuf_, posOneF32_, headNumTempQ, repeatTemp);
-    for (uint32_t zz = 0; zz < this->dynamicRound_; ++zz) {  // 每个核 核内只需拷贝一次cos sin（每个头cos sin都一样）
-      this->CosSinBroadcast(extraGm, zz, qkfp32Ubuf_,
-                            dynamicSliceQ);  // 一次放得下：hiddensize， 一次放不下：sliceSizeTmp
-      for (uint32_t perSlice = 0; perSlice < sliceTimeQ_; ++perSlice) {  // 核内每块
-        uint32_t dynamicSliceQTemp = (perSlice == sliceTimeQ_ - 1) ? lastSliceSizeQ_ : sliceSizeTmp_;
-        headNumTempQ = dynamicSliceQTemp / this->tilingData_->headDim;
-        uint32_t repeatTimeOnce = (dynamicSliceQTemp + this->repeatSize_ - 1) / this->repeatSize_;
-        this->QkComm(this->qGm_ + block_idx * this->nlCoreRun_ * this->tilingData_->hiddenSizeQ +
-                       zz * this->tilingData_->hiddenSizeQ + perSlice * sliceSizeTmp_,
-                     extraGm, dynamicSliceQTemp, commonUbuf_, headNumTempQ);
-        if (this->alignRotary_ == 0) {
-          pipe_barrier((PIPE_V));
-          CastQKDType2F32<QK_DTYPE>(repeatTimeOnce);
-
-          pipe_barrier((PIPE_V));
-          this->CalcRopeAlign(qkfp32Ubuf_, repeatTimeOnce, oriPosF32_, removeBeforeF32_, resOutFp32_);
-        } else {
-          set_flag(PIPE_MTE2, PIPE_V, EVENT_ID1);
-          wait_flag(PIPE_MTE2, PIPE_V, EVENT_ID1);
-
-          CastQKDType2F32<QK_DTYPE>(repeatTimeOnce);
-          pipe_barrier((PIPE_V));
-          this->CalcRope(qkfp32Ubuf_, repeatTimeOnce, oriPosF32_, removeBeforeF32_, PadBeforeF32_, posOneF32_,
-                         resOutFp32_);
-
-        }  // 帶PIPE_V
-        CastF322QKDType<QK_DTYPE>(this->outQGm_ + block_idx * this->nlCoreRun_ * this->tilingData_->hiddenSizeQ +
-                                    zz * this->tilingData_->hiddenSizeQ + perSlice * sliceSizeTmp_,
-                                  commonUbuf_ + resOut_, qkfp32Ubuf_ + resOutFp32_, repeatTimeOnce, dynamicSliceQTemp);
-        pipe_barrier(PIPE_ALL);
-      }
-      for (uint32_t perSlice = 0; perSlice < sliceTimeK_; ++perSlice) {  // 核内每块
-        uint32_t dynamicSliceKTemp = (perSlice == sliceTimeK_ - 1) ? lastSliceSizeK_ : sliceSizeTmp_;
-        headNumTempK = dynamicSliceKTemp / this->tilingData_->headDim;
-        uint32_t repeatTimeOnce = (dynamicSliceKTemp + this->repeatSize_ - 1) / this->repeatSize_;
-        this->QkComm(this->kGm_ + block_idx * this->nlCoreRun_ * this->tilingData_->hiddenSizeK +
-                       zz * this->tilingData_->hiddenSizeK + perSlice * sliceSizeTmp_,
-                     extraGm, dynamicSliceKTemp, commonUbuf_, headNumTempK);
-        if (this->alignRotary_ == 0) {
-          pipe_barrier((PIPE_V));
-          CastQKDType2F32<QK_DTYPE>(repeatTimeOnce);
-
-          pipe_barrier((PIPE_V));
-          this->CalcRopeAlign(qkfp32Ubuf_, repeatTimeOnce, oriPosF32_, removeBeforeF32_, resOutFp32_);
-        } else {
-          set_flag(PIPE_MTE2, PIPE_V, EVENT_ID1);
-          wait_flag(PIPE_MTE2, PIPE_V, EVENT_ID1);
-          CastQKDType2F32<QK_DTYPE>(repeatTimeOnce);
-
-          pipe_barrier((PIPE_V));
-          this->CalcRope(qkfp32Ubuf_, repeatTimeOnce, oriPosF32_, removeBeforeF32_, PadBeforeF32_, posOneF32_,
-                         resOutFp32_);
-        }  // 帶PIPE_V
-
-        CastF322QKDType<QK_DTYPE>(this->outKGm_ + block_idx * this->nlCoreRun_ * this->tilingData_->hiddenSizeK +
-                                    zz * this->tilingData_->hiddenSizeK + perSlice * sliceSizeTmp_,
-                                  commonUbuf_ + resOut_, qkfp32Ubuf_ + resOutFp32_, repeatTimeOnce, dynamicSliceKTemp);
-        pipe_barrier(PIPE_ALL);
-      }
-    }
-  }
-
- private:
-  AscendC::TQue<AscendC::QuePosition::VECIN, 1> qkfp32QueueCO2_;
-  AscendC::TQue<AscendC::QuePosition::VECIN, 1> outQueueCO2_;
-  __ubuf__ QK_DTYPE *commonUbuf_{nullptr};
-  __ubuf__ COS_DTYPE *qkfp32Ubuf_{nullptr};
-  uint32_t oriPosF32_{0};        // fp32的buf中qk的位置
-  uint32_t PadBeforeF32_{0};     // fp32的buf中保存qk[-x : hiddensize - x]
-  uint32_t removeBeforeF32_{0};  // fp32的buf中保存qk[x : hiddensize + x]
-  uint32_t posOneF32_{0};        // fp32的buf中0 0 0 1 1 1的位置
-  uint32_t headDimAlign_;        // 对齐的headDim
-  uint32_t sliceTimeQ_;          // 切分块的次数
-  uint32_t lastSliceSizeQ_;      // 最后一块的大小
-  uint32_t sliceTimeK_;
-  uint32_t lastSliceSizeK_;
-  uint32_t sliceSizeTmp_;
-  uint32_t resOut_;
-  uint32_t resOutFp32_;
-};
-
-#endif

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/and_impl.h

@@ -1,29 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-#ifndef MS_KERNELS_INTERNAL_KERNEL_AND_IMPL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_AND_IMPL_H_
-#include "elewise_binary_impl.h"
-namespace mindspore {
-namespace internal {
-class AndImpl : public ElewiseBinaryImpl {
- public:
-  AndImpl(const OpParamPtr &param) : ElewiseBinaryImpl(param) {}
-  virtual ~AndImpl() {}
-  int Launch() override;
-};
-}  // namespace internal
-}  // namespace mindspore
-#endif  // MS_KERNELS_INTERNAL_KERNEL_AND_IMPL_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/div_impl.h

@@ -1,29 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-#ifndef MS_KERNELS_INTERNAL_KERNEL_DIV_IMPL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_DIV_IMPL_H_
-#include "elewise_binary_impl.h"
-namespace mindspore {
-namespace internal {
-class DivImpl : public ElewiseBinaryImpl {
- public:
-  DivImpl(const OpParamPtr &param) : ElewiseBinaryImpl(param) {}
-  virtual ~DivImpl() {}
-  int Launch() override;
-};
-}  // namespace internal
-}  // namespace mindspore
-#endif  // MS_KERNELS_INTERNAL_KERNEL_DIV_IMPL_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/elewise_binary_impl.h

@@ -1,48 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-#ifndef MS_KERNELS_INTERNAL_KERNEL_ELEWISE_BINARY_IMPL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ELEWISE_BINARY_IMPL_H_
-
-#include <vector>
-#include "include/internal_kernel.h"
-#include "include/param/elewise_param.h"
-
-namespace mindspore {
-namespace internal {
-class ElewiseBinaryImpl : public InternelKernelImpl {
- public:
-  ElewiseBinaryImpl(const OpParamPtr &param) : InternelKernelImpl(param) {}
-  virtual ~ElewiseBinaryImpl() {}
-  bool Init(const ValidateInfo &info) override;
-  int Launch() { return -1; };
-  int Tiling(HostRawBuf &tilingBuf) override;
-  void SetStream(const void *stream_ptr) override;
-  void SetDeviceTilingBuf(const DeviceRawBuf &tilingBuf) override;
-  uint64_t GetTilingBufSize() override;
-  std::vector<uint64_t> GetWorkSpaceSize() override;
-  int InferShape(const std::vector<DIMS> &input_shapes, std::vector<DIMS> &output_shapes) override;
-  virtual int32_t GetMaxUbCount(uint32_t op_dtype);
-  bool IsSupported() override;
-
- protected:
-  void *stream_ptr_ = nullptr;
-  uint8_t *device_tiling_ = nullptr;
-  uint32_t aligned_factor_ = 128;
-  uint32_t ub_dtype = 0;
-};
-}  // namespace internal
-}  // namespace mindspore
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ELEWISE_BINARY_IMPL_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/elewise_binary_tiling.h

@@ -1,25 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-#ifndef MS_KERNELS_INTERNAL_ASCENDC_ELEWISE_BINARY_TILING_H_
-#define MS_KERNELS_INTERNAL_ASCENDC_ELEWISE_BINARY_TILING_H_
-#include "utils/elewise_tiling.h"
-namespace mindspore::internal {
-struct ElewiseBinaryTilingData : public ElewiseTailTilingData {
-  uint32_t broadcast_mode_{0};
-  uint32_t op_dtype_{0};
-};
-}  // namespace mindspore::internal
-#endif  // MS_KERNELS_INTERNAL_ASCENDC_ELEWISE_BINARY_TILING_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/and_kernel.h

@@ -1,46 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#ifndef MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_AND_KERNEL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_AND_KERNEL_H_
-
-#include "elewise_binary_base.h"
-
-template <typename T = int16_t>
-class AndI16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline AndI16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::And); }
-};
-
-template <typename T = uint16_t>
-class AndU16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline AndU16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::And); }
-};
-
-extern "C" __global__ __aicore__ void and_device(GM_ADDR x1, GM_ADDR x2, GM_ADDR y, GM_ADDR tiling, int32_t dtype) {
-  if (dtype == 6) {  // int16
-    AndI16<int16_t> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 7) {  // uint16
-    AndU16<uint16_t> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  }
-}
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_AND_KERNEL_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/div_kernel.h

@@ -1,46 +0,0 @@
-/**
- * Copyright 2024 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#ifndef MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_DIV_KERNEL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_DIV_KERNEL_H_
-
-#include "elewise_binary_base.h"
-
-template <typename T = half>
-class DivFp16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline DivFp16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Div); }
-};
-
-template <typename T = float>
-class DivFp32 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline DivFp32() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Div); }
-};
-
-extern "C" __global__ __aicore__ void div_device(GM_ADDR x1, GM_ADDR x2, GM_ADDR y, GM_ADDR tiling, int32_t dtype) {
-  if (dtype == 1) {  // fp16
-    DivFp16<half> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 0) {  // fp32
-    DivFp32<float> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  }
-}
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_DIV_KERNEL_H_