mindspore 2.4.10__cp310-none-any.whl → 2.5.0__cp310-none-any.whl

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_legacy(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_legacy(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_

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

@@ -1,260 +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_BASE_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_BASE_H_
-
-#include "kernel_operator.h"
-
-using namespace AscendC;
-
-template <typename IN_TYPE>
-class ElewiseBaseWide {
- public:
-  __aicore__ inline ElewiseBaseWide(){};
-  __aicore__ inline void InitBinary(GM_ADDR in1, GM_ADDR in2, GM_ADDR out, GM_ADDR tiling) {
-    gm_in1 = reinterpret_cast<__gm__ IN_TYPE *>(in1);
-    gm_in2 = reinterpret_cast<__gm__ IN_TYPE *>(in2);
-    gm_out = reinterpret_cast<__gm__ IN_TYPE *>(out);
-    SetTilingInfo(tiling);
-  }
-
-  __aicore__ inline void ProcessBinary() {
-    if (broadcast_mode == 0) {
-      return BroadcastNone();
-    } else if (broadcast_mode == 3) {
-      return ScalarLeft();
-    } else if (broadcast_mode == 4) {
-      return ScalarRight();
-    }
-  }
-  __aicore__ inline void SetBinaryFunc(void (*func)(const LocalTensor<IN_TYPE> &dstLocal,
-                                                    const LocalTensor<IN_TYPE> &src0Local,
-                                                    const LocalTensor<IN_TYPE> &src1Local, const int &calCount)) {
-    elewise_binary_func_ = func;
-  }
-
- private:
-  __aicore__ inline void SetTilingInfo(GM_ADDR tiling) {
-    core_idx = get_block_idx();
-    core_num = get_block_num();
-
-    avg_block_count = (uint32_t)(*((__gm__ uint32_t *)tiling + 0));
-    avg_block_ub_num = (uint32_t)(*((__gm__ uint32_t *)tiling + 1));
-    avg_block_ub_tail = (uint32_t)(*((__gm__ uint32_t *)tiling + 2));
-    avg_block_ub_loop = (uint32_t)(*((__gm__ uint32_t *)tiling + 3));
-    tail_block_count = (uint32_t)(*((__gm__ uint32_t *)tiling + 4));
-    tail_block_ub_num = (uint32_t)(*((__gm__ uint32_t *)tiling + 5));
-    tail_block_ub_tail = (uint32_t)(*((__gm__ uint32_t *)tiling + 6));
-    tail_block_ub_loop = (uint32_t)(*((__gm__ uint32_t *)tiling + 7));
-
-    buffer_num = (uint32_t)(*((__gm__ uint32_t *)tiling + 8));
-    broadcast_mode = (uint32_t)(*((__gm__ uint32_t *)tiling + 10));
-  }
-  __aicore__ inline void SetUbParam(uint32_t &ub_count, uint32_t &ub_loop, uint32_t &ub_tail) {
-    ub_count = avg_block_ub_num;
-    ub_loop = avg_block_ub_loop;
-    ub_tail = avg_block_ub_tail;
-
-    if (core_idx == core_num - 1) {
-      ub_count = tail_block_ub_num;
-      ub_loop = tail_block_ub_loop;
-      ub_tail = tail_block_ub_tail;
-    }
-  }
-  __aicore__ inline void InitInOut(uint32_t count) {
-    pipe.InitBuffer(in1Que, buffer_num, count * sizeof(IN_TYPE));
-    pipe.InitBuffer(in2Que, buffer_num, count * sizeof(IN_TYPE));
-    pipe.InitBuffer(outQue, buffer_num, count * sizeof(IN_TYPE));
-  }
-
-  __aicore__ inline void CopyOut(uint32_t idx, uint32_t stride, uint32_t count) {
-    LocalTensor<IN_TYPE> out = outQue.DeQue<IN_TYPE>();
-    DataCopy(outGm[idx * stride], out, count);
-    outQue.FreeTensor(out);
-  }
-
-  __aicore__ inline void CopyIn1(uint32_t idx, uint32_t stride, uint32_t count) {
-    LocalTensor<IN_TYPE> in1 = in1Que.AllocTensor<IN_TYPE>();
-    DataCopy(in1, in1Gm[idx * stride], count);
-    in1Que.EnQue(in1);
-  }
-
-  __aicore__ inline void CopyIn2(uint32_t idx, uint32_t stride, uint32_t count) {
-    LocalTensor<IN_TYPE> in2 = in2Que.AllocTensor<IN_TYPE>();
-    DataCopy(in2, in2Gm[idx * stride], count);
-    in2Que.EnQue(in2);
-  }
-
-  __aicore__ inline void BroadcastNone() {
-    uint32_t ub_count, ub_loop, ub_tail;
-    SetUbParam(ub_count, ub_loop, ub_tail);
-
-    in1Gm.SetGlobalBuffer(gm_in1 + core_idx * avg_block_count);
-    in2Gm.SetGlobalBuffer(gm_in2 + core_idx * avg_block_count);
-    outGm.SetGlobalBuffer(gm_out + core_idx * avg_block_count);
-
-    InitInOut(ub_count);
-
-    uint32_t loop = 0;
-    for (; loop < ub_loop - 1; loop++) {
-      CopyIn1(loop, ub_count, ub_count);
-      CopyIn2(loop, ub_count, ub_count);
-      ComputeNone(ub_count);
-      CopyOut(loop, ub_count, ub_count);
-    }
-
-    /* for ub tail */
-    if (ub_tail <= 0) {
-      return;
-    }
-    CopyIn1(loop, ub_count, ub_count);
-    CopyIn2(loop, ub_count, ub_tail);
-    ComputeNone(ub_tail);
-    CopyOut(loop, ub_count, ub_tail);
-  }
-  __aicore__ inline void ScalarLeft() {
-    uint32_t ub_count, ub_loop, ub_tail;
-    SetUbParam(ub_count, ub_loop, ub_tail);
-
-    in1Gm.SetGlobalBuffer(gm_in1);
-    in2Gm.SetGlobalBuffer(gm_in2 + core_idx * avg_block_count);
-    outGm.SetGlobalBuffer(gm_out + core_idx * avg_block_count);
-
-    InitInOut(ub_count);
-
-    /* input tensor init once */
-    LocalTensor<IN_TYPE> scalar_t = in1Que.AllocTensor<IN_TYPE>();
-    IN_TYPE scalar_value = in1Gm.GetValue(0);
-    Duplicate(scalar_t, scalar_value, ub_count);
-    pipe_barrier(PIPE_ALL);
-
-    uint32_t loop = 0;
-    for (; loop < ub_loop - 1; loop++) {
-      CopyIn2(loop, ub_count, ub_count);
-      ComputeLeft(ub_count, scalar_t);
-      CopyOut(loop, ub_count, ub_count);
-    }
-
-    /* for ub tail */
-    if (ub_tail <= 0) {
-      return;
-    }
-    CopyIn2(loop, ub_count, ub_tail);
-    ComputeLeft(ub_tail, scalar_t);
-    CopyOut(loop, ub_count, ub_tail);
-
-    /* free*/
-    in1Que.FreeTensor(scalar_t);
-  }
-  __aicore__ inline void ScalarRight() {
-    uint32_t ub_count, ub_loop, ub_tail;
-    SetUbParam(ub_count, ub_loop, ub_tail);
-
-    in1Gm.SetGlobalBuffer(gm_in1 + core_idx * avg_block_count);
-    in2Gm.SetGlobalBuffer(gm_in2);
-    outGm.SetGlobalBuffer(gm_out + core_idx * avg_block_count);
-
-    InitInOut(ub_count);
-
-    /* input tensor init once */
-    LocalTensor<IN_TYPE> scalar_t = in2Que.AllocTensor<IN_TYPE>();
-    IN_TYPE scalar_value = in2Gm.GetValue(0);
-    Duplicate(scalar_t, scalar_value, ub_count);
-    pipe_barrier(PIPE_ALL);
-
-    uint32_t loop = 0;
-    for (; loop < ub_loop - 1; loop++) {
-      CopyIn1(loop, ub_count, ub_count);
-      ComputeRight(ub_count, scalar_t);
-      CopyOut(loop, ub_count, ub_count);
-    }
-
-    /* for ub tail */
-    if (ub_tail <= 0) {
-      return;
-    }
-    CopyIn1(loop, ub_count, ub_tail);
-    ComputeRight(ub_tail, scalar_t);
-    CopyOut(loop, ub_count, ub_tail);
-
-    /* free*/
-    in2Que.FreeTensor(scalar_t);
-  }
-
-  __aicore__ inline void ComputeLeft(uint32_t count, LocalTensor<IN_TYPE> &scalar_t) {
-    LocalTensor<IN_TYPE> in2 = in2Que.DeQue<IN_TYPE>();
-    LocalTensor<IN_TYPE> out = outQue.AllocTensor<IN_TYPE>();
-    elewise_binary_func_(out, scalar_t, in2, count);
-    in2Que.FreeTensor(in2);
-    pipe_barrier(PIPE_ALL);
-    outQue.EnQue(out);
-  }
-
-  __aicore__ inline void ComputeRight(uint32_t count, LocalTensor<IN_TYPE> &scalar_t) {
-    LocalTensor<IN_TYPE> in1 = in1Que.DeQue<IN_TYPE>();
-    LocalTensor<IN_TYPE> out = outQue.AllocTensor<IN_TYPE>();
-    elewise_binary_func_(out, in1, scalar_t, count);
-    in1Que.FreeTensor(in1);
-    pipe_barrier(PIPE_ALL);
-    outQue.EnQue(out);
-  }
-
-  __aicore__ inline void ComputeNone(uint32_t count) {
-    LocalTensor<IN_TYPE> in1 = in1Que.DeQue<IN_TYPE>();
-    LocalTensor<IN_TYPE> in2 = in2Que.DeQue<IN_TYPE>();
-    LocalTensor<IN_TYPE> out = outQue.AllocTensor<IN_TYPE>();
-    elewise_binary_func_(out, in1, in2, count);
-    in1Que.FreeTensor(in1);
-    in2Que.FreeTensor(in2);
-    pipe_barrier(PIPE_ALL);
-    outQue.EnQue(out);
-  }
-
- private:
-  void (*elewise_binary_func_)(const LocalTensor<IN_TYPE> &dstLocal, const LocalTensor<IN_TYPE> &src0Local,
-                               const LocalTensor<IN_TYPE> &src1Local, const int &calCount);
-
-  TPipe pipe;
-  TQue<AscendC::QuePosition::VECIN, 1> in1Que;
-  TQue<AscendC::QuePosition::VECIN, 1> in2Que;
-  TQue<AscendC::QuePosition::VECOUT, 1> outQue;
-
-  __gm__ IN_TYPE *__restrict__ gm_in1{nullptr};
-  __gm__ IN_TYPE *__restrict__ gm_in2{nullptr};
-  __gm__ IN_TYPE *__restrict__ gm_out{nullptr};
-
-  GlobalTensor<IN_TYPE> in1Gm;
-  GlobalTensor<IN_TYPE> in2Gm;
-  GlobalTensor<IN_TYPE> outGm;
-
-  uint32_t core_idx{0};
-  uint32_t core_num{0};
-  uint32_t buffer_num{0};
-  uint32_t broadcast_mode{0};
-
-  uint32_t avg_block_count{0};
-  uint32_t avg_block_ub_num{0};
-  uint32_t avg_block_ub_tail{0};
-  uint32_t avg_block_ub_loop{0};
-
-  uint32_t tail_block_count{0};
-  uint32_t tail_block_ub_num{0};
-  uint32_t tail_block_ub_tail{0};
-  uint32_t tail_block_ub_loop{0};
-};
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_BASE_H_

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

@@ -1,35 +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_ELEWISE_BINARY_KERNEL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_ELEWISE_BINARY_KERNEL_H_
-
-void elewise_sub(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                 uint8_t *tiling, int dtype);
-void elewise_mul(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                 uint8_t *tiling, int dtype);
-void elewise_div(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                 uint8_t *tiling, int dtype);
-void elewise_min(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                 uint8_t *tiling, int dtype);
-void elewise_max(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                 uint8_t *tiling, int dtype);
-void elewise_and(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                 uint8_t *tiling, int dtype);
-void elewise_or(uint32_t blockDim, void *l2ctrl, void *stream, uint8_t *in1, uint8_t *in2, uint8_t *out,
-                uint8_t *tiling, int dtype);
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_ELEWISE_BINARY_KERNEL_H_

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

@@ -1,66 +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_MAX_KERNEL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_MAX_KERNEL_H_
-
-#include "elewise_binary_base.h"
-
-template <typename T = int16_t>
-class MaxI16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MaxI16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Max); }
-};
-
-template <typename T = int32_t>
-class MaxInt : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MaxInt() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Max); }
-};
-
-template <typename T = half>
-class MaxFp16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MaxFp16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Max); }
-};
-
-template <typename T = float>
-class MaxFp32 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MaxFp32() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Max); }
-};
-
-extern "C" __global__ __aicore__ void max_device_legacy(GM_ADDR x1, GM_ADDR x2, GM_ADDR y, GM_ADDR tiling, int32_t dtype) {
-  if (dtype == 3) {  // int32
-    MaxInt<int32_t> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 1) {  // fp16
-    MaxFp16<half> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 0) {  // fp32
-    MaxFp32<float> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 6) {  // int16
-    MaxI16<int16_t> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  }
-}
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_MAX_KERNEL_H_

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

@@ -1,66 +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_MIN_KERNEL_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_MIN_KERNEL_H_
-
-#include "elewise_binary_base.h"
-
-template <typename T = int16_t>
-class MinI16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MinI16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Min); }
-};
-
-template <typename T = int32_t>
-class MinInt : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MinInt() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Min); }
-};
-
-template <typename T = half>
-class MinFp16 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MinFp16() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Min); }
-};
-
-template <typename T = float>
-class MinFp32 : public ElewiseBaseWide<T> {
- public:
-  __aicore__ inline MinFp32() { ElewiseBaseWide<T>::SetBinaryFunc(AscendC::Min); }
-};
-
-extern "C" __global__ __aicore__ void min_device_legacy(GM_ADDR x1, GM_ADDR x2, GM_ADDR y, GM_ADDR tiling, int32_t dtype) {
-  if (dtype == 3) {  // int32
-    MinInt<int32_t> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 1) {  // fp16
-    MinFp16<half> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 0) {  // fp32
-    MinFp32<float> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  } else if (dtype == 6) {  // int16
-    MinI16<int16_t> op;
-    op.InitBinary(x1, x2, y, tiling);
-    op.ProcessBinary();
-  }
-}
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ASCENDC_ELEWISE_BINARY_MIN_KERNEL_H_