mindspore 2.4.1__cp310-cp310-manylinux1_x86_64.whl → 2.5.0__cp310-cp310-manylinux1_x86_64.whl

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/op_creator.h

@@ -1,113 +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_OP_CREATOR_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_OP_CREATOR_H_
-
-#include "acme/include/acme_op.h"
-#include "acme/include/op_param.h"
-
-namespace mindspore {
-namespace acme {
-AcmeOpPtr CreateMatmulOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                         const MatmulParam &param, const std::string &op_name);
-AcmeOpPtr CreateAddOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                      const std::string &op_name);
-AcmeOpPtr CreateAddLayerNormOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                               const NormParam &param, const std::string &op_name);
-AcmeOpPtr CreateAddRmsNormQuantOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                                  const NormParam &param, const std::string &op_name);
-AcmeOpPtr CreateCastOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                       const std::string &op_name);
-AcmeOpPtr CreateTransposeOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                            const TransposeParam &param, const std::string &op_name);
-AcmeOpPtr CreateQuantPerChannelOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                                  const std::string &op_name);
-AcmeOpPtr CreateSwishOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                        const std::string &op_name);
-AcmeOpPtr CreateSwiGLUOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                         const SwiGLUParam &param, const std::string &op_name);
-AcmeOpPtr CreateLogicalNotOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                             const std::string &op_name);
-AcmeOpPtr CreateSoftmaxOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                          const SoftmaxParam &param, const std::string &op_name);
-AcmeOpPtr CreateReduceSumOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                            const ReduceSumParam &param, const std::string &op_name);
-AcmeOpPtr CreateGatherOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                         const GatherParam &param, const std::string &op_name);
-AcmeOpPtr CreateApplyRotaryPosEmbOp(const InputsImmutableInfoList &inputs_ii,
-                                    const OutputsImmutableInfoList &outputs_ii, const ApplyRotaryPosEmbParam &param,
-                                    const std::string &op_name);
-AcmeOpPtr CreateApplyRotaryPosEmbNzOp(const InputsImmutableInfoList &inputs_ii,
-                                      const OutputsImmutableInfoList &outputs_ii, const ApplyRotaryPosEmbParam &param,
-                                      const std::string &op_name);
-AcmeOpPtr CreateRmsNormOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                          const NormParam &param, const std::string &op_name);
-AcmeOpPtr CreateMatmulAddRmsNormOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                                   const MatmulAddRmsNormParam &param, const std::string &op_name);
-AcmeOpPtr CreateMultiWeightMatmulOp(const InputsImmutableInfoList &inputs_ii,
-                                    const OutputsImmutableInfoList &outputs_ii, const MultiWeightMatmulParam &param,
-                                    const std::string &op_name);
-// param section 0
-AcmeOpPtr CreateGeLUOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                       const std::string &op_name);
-AcmeOpPtr CreateAddRmsNormOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                             const NormParam &param, const std::string &op_name);
-AcmeOpPtr CreateFlashAttentionScoreOp(const InputsImmutableInfoList &inputs_ii,
-                                      const OutputsImmutableInfoList &outputs_ii, const FlashAttentionScoreParam &param,
-                                      const std::string &op_name);
-AcmeOpPtr CreatePagedAttentionOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                                 const PagedAttentionParam &param, const std::string &op_name);
-// param section 1
-AcmeOpPtr CreateReshapeAndCacheOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                                  const std::string &op_name);
-AcmeOpPtr CreateReshapeAndCacheNzOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                                    const std::string &op_name);
-AcmeOpPtr CreateMulOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                      const std::string &op_name);
-AcmeOpPtr CreateSubOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                      const std::string &op_name);
-AcmeOpPtr CreateRealDivOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                          const std::string &op_name);
-
-// param section 2
-AcmeOpPtr CreateFastGeLUOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                           const std::string &op_name);
-AcmeOpPtr CreateTransDataOp(const InputsImmutableInfoList &inputs_ii, const OutputsImmutableInfoList &outputs_ii,
-                            const TransDataParam &param, const std::string &op_name);
-AcmeOpPtr CreateQuantLinearSparseOp(const InputsImmutableInfoList &inputs_ii,
-                                    const OutputsImmutableInfoList &outputs_ii, const std::string &op_name);
-// param section 3
-
-// param section 4
-
-// param section 5
-
-// param section 6
-
-// param section 7
-
-// param section 8
-
-// param section 9
-
-// param section 10
-
-bool IsAcmeKernelDtypesSupported(const std::string op_name, InputDataTypes in_dtypes, InputDataTypes out_dtypes);
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_OP_CREATOR_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/op_param.h

@@ -1,193 +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_OP_PARAM_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_OP_PARAM_H_
-
-#include <stdint.h>
-#include <vector>
-
-namespace mindspore {
-namespace acme {
-// matmul fused op
-constexpr auto kAcmeMatMulOpName = "MatMul";
-constexpr auto kAcmeMultiWeightMatmulOpName = "MultiWeightMatmul";
-constexpr auto kAcmeMatMulAddRmsNormOpName = "MatMulAddRmsNorm";
-// attention fused op
-constexpr auto kAcmeFlashAttentionScoreOpName = "FlashAttentionScore";
-constexpr auto kAcmePagedAttentionOpName = "PagedAttention";
-constexpr auto kAcmeReshapeAndCacheOpName = "ReshapeAndCache";
-constexpr auto kAcmeReshapeAndCacheNzOpName = "ReshapeAndCacheNz";
-constexpr auto kAcmeApplyRotaryPosEmbOpName = "ApplyRotaryPosEmb";
-constexpr auto kAcmeApplyRotaryPosEmbNzOpName = "ApplyRotaryPosEmbNz";
-// norm fused op
-constexpr auto kAcmeAddLayerNormOpName = "AddLayerNorm";
-constexpr auto kAcmeRmsNormOpName = "RmsNorm";
-constexpr auto kAcmeAddRmsNormOpName = "AddRmsNorm";
-constexpr auto kAcmeRmsNormQuantOpName = "RmsNormQuant";
-constexpr auto kAcmeAddRmsNormQuantOpName = "AddRmsNormQuantV2";
-// activation
-constexpr auto kAcmeReluOpName = "Relu";
-constexpr auto kAcmeGeLUOpName = "GeLU";
-constexpr auto kAcmeFastGeLUOpName = "FastGeLU";
-constexpr auto kAcmeSwishOpName = "Swish";
-constexpr auto kAcmeSwiGLUOpName = "SwiGLU";
-// elewise unary
-constexpr auto kAcmeCastOpName = "Cast";
-constexpr auto kAcmeExpOpName = "Exp";
-constexpr auto kAcmeLnOpName = "Ln";
-constexpr auto kAcmeRsqrtOpName = "Rsqrt";
-constexpr auto kAcmeSqrtOpName = "Sqrt";
-constexpr auto kAcmeAbsOpName = "Abs";
-constexpr auto kAcmeReciprocalOpName = "Reciprocal";
-// elewise binary
-constexpr auto kAcmeAddOpName = "Add";
-constexpr auto kAcmeSubOpName = "Sub";
-constexpr auto kAcmeMulOpName = "Mul";
-constexpr auto kAcmeDivOpName = "Div";
-constexpr auto kAcmeRealDivOpName = "RealDiv";
-constexpr auto kAcmeMaxOpName = "Max";
-constexpr auto kAcmeMinOpName = "Min";
-constexpr auto kAcmeNotOpName = "Not";
-constexpr auto kAcmeOrOpName = "Or";
-constexpr auto kAcmeAndOpName = "And";
-constexpr auto kAcmeEqualOpName = "Equal";
-constexpr auto kAcmeNotEqualOpName = "NotEqual";
-constexpr auto kAcmeLessOpName = "Less";
-constexpr auto kAcmeLessEqualOpName = "LessEqual";
-constexpr auto kAcmeGreaterOpName = "Greater";
-constexpr auto kAcmeGreaterEqualOpName = "GreaterEqual";
-constexpr auto kAcmeLogicalNotOpName = "LogicalNot";
-// others
-constexpr auto kAcmeGatherOpName = "Gather";
-constexpr auto kAcmeTransposeOpName = "Transpose";
-constexpr auto kAcmeTransDataOpName = "TransData";
-constexpr auto kAcmeQuantPerChannelOpName = "QuantPerChannel";
-constexpr auto kAcmeSoftmaxOpName = "Softmax";
-constexpr auto kAcmeReduceSumOpName = "ReduceSum";
-constexpr auto kAcmeQuantLinearSparseOpName = "QuantLinearSparse";
-
-struct AxesParam {
-  std::vector<int64_t> axes;
-};
-
-using TransposeParam = AxesParam;
-using SoftmaxParam = AxesParam;
-using ReduceSumParam = AxesParam;
-
-struct GatherParam {
-  int64_t batch_dims;
-  std::vector<int64_t> axes;
-};
-
-struct SwiGLUParam {
-  int64_t axis;
-};
-
-struct MatmulParam {
-  bool transpose_a{false};
-  bool transpose_b{false};
-  bool enable_dequant{false};
-  bool with_relu{false};
-  bool with_gelu{false};
-  bool with_bias{false};
-  bool with_bias_fastgelu{false};
-  bool enable_shuffle{false};
-};
-
-struct MatmulAddRmsNormParam {
-  bool transpose_a{false};
-  bool transpose_b{false};
-  float eps{1e-6};
-};
-
-struct MultiWeightMatmulParam {
-  uint32_t n0_len{0};
-  uint32_t n1_len{0};
-  uint32_t n2_len{0};
-  bool transpose_a;
-  bool transpose_b;
-  int32_t silu_position{-1};
-  bool with_bias{false};
-};
-
-struct NormParam {
-  float eps;
-  bool operator==(const NormParam &other) const { return this->eps == other.eps; }
-};
-
-struct ApplyRotaryPosEmbParam {
-  // cos_format=0  shape是[maxSeqLen, headDim]，    cos/sin不交替
-  // cos_format=1  shape是[maxSeqLen, headDim]，    cos/sin交替
-  // cos_format=2  shape是[batch*seqLen, headDim]， cos/sin不交替
-  // cos_format=3  shape是[batch*seqLen, headDim]， cos/sin交替
-  int32_t cos_format{0};
-  int32_t rotary_coeff{-1};
-  std::vector<int32_t> batch_valid_length;
-};
-
-struct TransDataParam {
-  enum TransdataType { UNDEFINED = 0, FRACTAL_NZ_TO_ND, ND_TO_FRACTAL_NZ };
-  TransdataType transdataType = UNDEFINED;
-  enum SpecialType { NORMAL = 0, ATTENTION_INPUT_QKV, ATTENTION_INPUT_MASK };
-  int64_t specialTransdata = NORMAL;
-};
-
-struct FlashAttentionScoreParam {
-  int32_t head_num = 0;
-  int32_t inner_precise = 0;
-  int32_t pre_tokens = 2147483647;
-  int32_t next_tokens = 0;
-  int32_t sparse_mode = 0;
-  int32_t mask_dtype = 0;
-  int32_t input_layout = 0;
-  std::vector<int64_t> mask_dims;
-  std::vector<int32_t> kv_seq_len;
-  std::vector<int32_t> q_seq_len;
-  float tor = 0;
-
-  enum InputLayoutMode : int64_t { 
-    BSH = 0,
-    BNSD = 1,
-    SBH = 2,
-    BSND = 3,
-    TND = 4,
-    TH = 5,
-    NSD = 6,
-    SH = 7
-  };
-};
-
-struct PagedAttentionParam {
-  int32_t inner_precise = 0;
-  int32_t head_num = 0;
-  int32_t kv_head_num = 0;
-  std::vector<int32_t> kv_seq_len;
-  std::vector<int32_t> q_seq_len;
-  float tor = 0;
-
-  enum MaskType : uint32_t {
-      kMaskTypeNone = 0,
-      kMaskTypeAlibi = 1,
-      kMaskTypeLookAhead = 2
-  };
-  MaskType mask_type = kMaskTypeNone;
-  int32_t kv_cache_quant_mode = 0;
-};
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_OP_PARAM_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/dtype_registry.h

@@ -1,90 +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_KERNEL_REGISTRY_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_KERNEL_REGISTRY_H_
-
-#include <map>
-
-#include "acme/include/base_type.h"
-
-namespace mindspore {
-namespace acme {
-class DtypeRegistry {
- public:
-  DtypeRegistry() = default;
-  virtual ~DtypeRegistry() = default;
-
-  static DtypeRegistry *GetInstance();
-  void RegisterOpName2NzOpName(const std::string &op_name, const std::string &nz_op_name);
-  std::string ConvertOpName2NzOpName(const std::string &op_name) const;
-  void RegisterAcmeDtypes(const std::string &op_name, const InOutDtypesList &dtype_list);
-  void RegisterAcmeDtypes(const std::string &op_name, const InOutDtypesTargetMap &dtype_map);
-  void RegisterAsdDtypes(const std::string &op_name, const InOutDtypesList &dtype_list);
-  void RegisterAsdDtypes(const std::string &op_name, const InOutDtypesTargetMap &dtype_map);
-  void PrintSupportDtypes(const std::string &op_name);
-  bool IsDtypesSupported(const std::string &op_name, const InputDataTypes &in_dtypes, const InputDataTypes &out_dtypes);
-  bool IsAcmeSupported(const std::string &op_name, const InputDataTypes &in_dtypes, const InputDataTypes &out_dtypes);
-  bool IsAsdSupported(const std::string &op_name, const InputDataTypes &in_dtypes, const InputDataTypes &out_dtypes);
-
- private:
-  std::map<std::string, InOutDtypesList> acme_dtype_support_map_;
-  std::map<std::string, InOutDtypesList> asd_dtype_support_map_;
-  std::map<std::string, std::string> op_name_2_nz_op_name_map_;
-};
-
-class OpNameRegistrar {
- public:
-  explicit OpNameRegistrar(const std::string &op_name, const std::string &nz_op_name) {
-    DtypeRegistry::GetInstance()->RegisterOpName2NzOpName(op_name, nz_op_name);
-  }
-  ~OpNameRegistrar() = default;
-};
-#define ACME_KERNEL_REG_OP_NAME(NAME, NZ_NAME) static const OpNameRegistrar g_##NAME##_acme_name_reg(NAME, NZ_NAME)
-
-class DtypeRegistrar {
- public:
-  explicit DtypeRegistrar(const std::string &op_name, const InOutDtypesList &dtype_list) {
-    DtypeRegistry::GetInstance()->RegisterAcmeDtypes(op_name, dtype_list);
-  }
-
-  explicit DtypeRegistrar(const std::string &op_name, const InOutDtypesTargetMap &dtype_map) {
-    DtypeRegistry::GetInstance()->RegisterAcmeDtypes(op_name, dtype_map);
-  }
-  ~DtypeRegistrar() = default;
-};
-
-#define ACME_KERNEL_REG_BY_DTYPES(NAME, DTYPES) static const DtypeRegistrar g_##NAME##_acme_reg(NAME, DTYPES)
-#define ACME_KERNEL_REG_BY_TARGET_DTYPES(NAME, TARGET_DTYPES) static const DtypeRegistrar g_##NAME##_target_acme_reg(NAME, TARGET_DTYPES)
-
-class AsdOpDtypeRegistrar {
- public:
-  explicit AsdOpDtypeRegistrar(const std::string &op_name, const InOutDtypesList &dtype_list) {
-    DtypeRegistry::GetInstance()->RegisterAsdDtypes(op_name, dtype_list);
-  }
-
-  explicit AsdOpDtypeRegistrar(const std::string &op_name, const InOutDtypesTargetMap &dtype_map) {
-    DtypeRegistry::GetInstance()->RegisterAsdDtypes(op_name, dtype_map);
-  }
-  ~AsdOpDtypeRegistrar() = default;
-};
-
-#define ASDOP_KERNEL_REG_BY_DTYPES(NAME, DTYPES) static const AsdOpDtypeRegistrar g_##NAME##_asd_reg(NAME, DTYPES)
-#define ASDOP_KERNEL_REG_BY_TARGET_DTYPES(NAME, TARGET_DTYPES) static const AsdOpDtypeRegistrar g_##NAME##_target_asd_reg(NAME, TARGET_DTYPES)
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_KERNEL_REGISTRY_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/kernel_register.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_ACME_SRC_CORE_KERNEL_REGISTER_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_CORE_KERNEL_REGISTER_H_
-
-#include <vector>
-
-namespace mindspore {
-namespace acme {
-
-extern const char *kMatmulKernelName;
-extern const char *kMultiWeightMatmulKernelName;
-extern const size_t kFlashAttentionScoreRegisterIndex;
-extern const size_t kPagedAttentionRegisterIndex;
-extern const std::vector<std::vector<const char *>> kKernelMaps;
-
-class KernelRegister {
- public:
-  KernelRegister();
-  ~KernelRegister() = default;
-
-  static const KernelRegister &GetInstance() {
-    static const KernelRegister kKernelRegister;
-    return kKernelRegister;
-  }
-
-  void Register();
-};
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_CORE_KERNEL_REGISTER_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/platform/platform_configs.h

@@ -1,89 +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_CORE_PLATFORM_CONFIGS_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_CORE_PLATFORM_CONFIGS_H_
-
-#include <string>
-#include <cstdint>
-
-namespace mindspore {
-namespace acme {
-class HardwareConfig {
- public:
-  HardwareConfig() = default;
-  ~HardwareConfig() = default;
-  HardwareConfig(uint32_t core_num, uint32_t l2_size, uint32_t l1_size, uint32_t l0a_size, uint32_t l0b_size,
-                 uint32_t l0c_size, uint32_t ub_size)
-      : core_num_(core_num),
-        l2_size_(l2_size),
-        l1_size_(l1_size),
-        l0a_size_(l0a_size),
-        l0b_size_(l0b_size),
-        l0c_size_(l0c_size),
-        ub_size_(ub_size) {}
-
-  uint32_t core_num_{0};
-  uint32_t l2_size_{0};
-  uint32_t l1_size_{0};
-  uint32_t l0a_size_{0};
-  uint32_t l0b_size_{0};
-  uint32_t l0c_size_{0};
-  uint32_t hbm_bandwidth_{1};
-  uint32_t l2_bandwidth_{5};
-  uint32_t ub_size_{0};
-};
-
-class PlatformConfigs {
- public:
-  PlatformConfigs();
-  ~PlatformConfigs() = default;
-
-  static const PlatformConfigs &GetInstance() {
-    static PlatformConfigs kPlatformConfigs;
-    return kPlatformConfigs;
-  }
-
-  inline uint32_t GetCoreNum() const { return hw_config_.core_num_; }
-
-  inline uint32_t GetL2Size() const { return hw_config_.l2_size_; }
-
-  inline uint32_t GetL1Size() const { return hw_config_.l1_size_; }
-
-  inline uint32_t GetL0aSize() const { return hw_config_.l0a_size_; }
-
-  inline uint32_t GetL0bSize() const { return hw_config_.l0b_size_; }
-
-  inline uint32_t GetL0cSize() const { return hw_config_.l0c_size_; }
-
-  inline uint32_t GetHbmBandwidth() const { return hw_config_.hbm_bandwidth_; }
-
-  inline uint32_t GetL2BandwidthSize() const { return hw_config_.l2_bandwidth_; }
-
-  inline uint32_t GetUbSize() const { return hw_config_.ub_size_; }
-
-  const HardwareConfig &GetConfigByVersion(const std::string &soc_version) const;
-
- private:
-  void Init();
-
-  HardwareConfig hw_config_;
-  std::string soc_version_;
-};
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_CORE_PLATFORM_CONFIGS_H_

mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/platform/rt_funcs.h

@@ -1,135 +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_CORE_PLATFORM_RT_FUNCS_H_
-#define MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_CORE_PLATFORM_RT_FUNCS_H_
-
-#include "acme/include/base_type.h"
-
-#ifdef __cplusplus
-extern "C" {
-#endif
-
-#define RT_DEV_BINARY_MAGIC_ELF 0x43554245U
-#define RT_DEV_BINARY_MAGIC_ELF_AIVEC 0x41415246U
-#define RT_DEV_BINARY_MAGIC_ELF_AICUBE 0x41494343U
-
-typedef void *rtStream_t;
-
-typedef enum {
-  INTERNAL_RTSUCCESS = 0,
-  INTERNAL_RTERROR_NOT_INITIALIZED = -1,
-  INTERNAL_RTERROR_NOT_IMPLMENT = -2,
-  INTERNAL_RTERROR_ASCEND_ENV_NOT_EXIST = -3,
-  INTERNAL_RTERROR_LOAD_RUNTIME_FAIL = -4,
-  INTERNAL_RTERROR_FUNC_NOT_EXIST = -5,
-  INTERNAL_RTERROR_OPEN_BIN_FILE_FAIL = -6,
-  INTERNAL_RTERROR_PARA_CHECK_FAIL = -7,
-} RtError;
-
-typedef enum tagRtError {
-  RT_ERROR_NONE = 0x0,                      // success
-  RT_ERROR_INVALID_VALUE = 0x1,             // invalid value
-  RT_ERROR_MEMORY_ALLOCATION = 0x2,         // memory allocation fail
-  RT_ERROR_INVALID_RESOURCE_HANDLE = 0x3,   // invalid handle
-  RT_ERROR_INVALID_DEVICE_POINTER = 0x4,    // invalid device point
-  RT_ERROR_INVALID_MEMCPY_DIRECTION = 0x5,  // invalid memory copy dirction
-  RT_ERROR_INVALID_DEVICE = 0x6,            // invalid device
-  RT_ERROR_NO_DEVICE = 0x7,                 // no valid device
-  RT_ERROR_CMD_OCCUPY_FAILURE = 0x8,        // command occpuy failure
-  RT_ERROR_SET_SIGNAL_FAILURE = 0x9,        // set signal failure
-  RT_ERROR_UNSET_SIGNAL_FAILURE = 0xA,      // unset signal failure
-  RT_ERROR_OPEN_FILE_FAILURE = 0xB,         // unset signal failure
-  RT_ERROR_WRITE_FILE_FAILURE = 0xC,
-  RT_ERROR_MEMORY_ADDRESS_UNALIGNED = 0xD,
-  RT_ERROR_DRV_ERR = 0xE,
-  RT_ERROR_LOST_HEARTBEAT = 0xF,
-  RT_ERROR_REPORT_TIMEOUT = 0x10,
-  RT_ERROR_NOT_READY = 0x11,
-  RT_ERROR_DATA_OPERATION_FAIL = 0x12,
-  RT_ERROR_INVALID_L2_INSTR_SIZE = 0x13,
-  RT_ERROR_DEVICE_PROC_HANG_OUT = 0x14,
-  RT_ERROR_DEVICE_POWER_UP_FAIL = 0x15,
-  RT_ERROR_DEVICE_POWER_DOWN_FAIL = 0x16,
-  RT_ERROR_FEATURE_NOT_SUPPROT = 0x17,
-  RT_ERROR_KERNEL_DUPLICATE = 0x18,             // register same kernel repeatly
-  RT_ERROR_MODEL_STREAM_EXE_FAILED = 0x91,      // the model stream failed
-  RT_ERROR_MODEL_LOAD_FAILED = 0x94,            // the model stream failed
-  RT_ERROR_END_OF_SEQUENCE = 0x95,              // end of sequence
-  RT_ERROR_NO_STREAM_CB_REG = 0x96,             // no callback register info for stream
-  RT_ERROR_DATA_DUMP_LOAD_FAILED = 0x97,        // data dump load info fail
-  RT_ERROR_CALLBACK_THREAD_UNSUBSTRIBE = 0x98,  // callback thread unsubstribe
-  RT_ERROR_RESERVED
-} rtError_t;
-
-// rt kernel
-typedef struct {
-  uint32_t magic{0};
-  uint32_t version{0};
-  const void *data{nullptr};
-  uint64_t length{0};
-} RtDevBinary_T;
-
-typedef void *rtStream_t;
-
-using RtDevBinaryRegisterFunc = rtError_t (*)(const RtDevBinary_T *bin, void **hdl);
-using RtFunctionRegisterFunc = rtError_t (*)(void *binHandle, const void *subFunc, const char *stubName,
-                                             const void *kernelInfoExt, uint32_t funcMode);
-using RtKernelLaunchFunc = rtError_t (*)(const void *stubFunc, uint32_t blockDim, void *args, uint32_t argsSize,
-                                         void *smDesc, rtStream_t sm);
-using RtGetC2cCtrlAddrFunc = rtError_t (*)(uint64_t *addr, uint32_t *len);
-
-using RtGetSocVersionFunc = int (*)(char *version, uint32_t maxLen);
-
-#ifdef __cplusplus
-}
-#endif
-
-namespace mindspore {
-namespace acme {
-class RtFuncs {
- public:
-  RtFuncs();
-  ~RtFuncs() = default;
-
-  static const RtFuncs &GetInstance() {
-    static RtFuncs kRtFuncs;
-    return kRtFuncs;
-  }
-
-  inline RtDevBinaryRegisterFunc GetRtDevBinaryRegisterFunc() const { return rt_dev_bin_reg_func_; }
-
-  inline RtFunctionRegisterFunc GetRtFunctionRegisterFunc() const { return rt_func_reg_func_; }
-
-  inline RtKernelLaunchFunc GetRtKernelLaunchFunc() const { return rt_kernel_launch_func_; }
-
-  inline RtGetC2cCtrlAddrFunc GetRtGetC2cCtrlAddrFunc() const { return rt_get_c2c_ctrl_addr_func_; }
-
-  std::string GetSocVersion() const;
-
- private:
-  void Init();
-
-  RtDevBinaryRegisterFunc rt_dev_bin_reg_func_{nullptr};
-  RtFunctionRegisterFunc rt_func_reg_func_{nullptr};
-  RtKernelLaunchFunc rt_kernel_launch_func_{nullptr};
-  RtGetC2cCtrlAddrFunc rt_get_c2c_ctrl_addr_func_{nullptr};
-  RtGetSocVersionFunc rt_get_soc_version_func_{nullptr};
-};
-}  // namespace acme
-}  // namespace mindspore
-
-#endif  // MS_KERNELS_INTERNAL_KERNEL_ACME_SRC_CORE_PLATFORM_RT_FUNCS_H_