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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of mindspore might be problematic. Click here for more details.

Files changed (866) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/Third_Party_Open_Source_Software_Notice +39 -0
  3. mindspore/__init__.py +8 -3
  4. mindspore/_akg/akg/composite/build_module.py +6 -2
  5. mindspore/_akg/akg/utils/kernel_exec.py +2 -2
  6. mindspore/_c_dataengine.cpython-310-x86_64-linux-gnu.so +0 -0
  7. mindspore/_c_expression.cpython-310-x86_64-linux-gnu.so +0 -0
  8. mindspore/_c_mindrecord.cpython-310-x86_64-linux-gnu.so +0 -0
  9. mindspore/_checkparam.py +0 -5
  10. mindspore/_extends/parallel_compile/akg_compiler/gen_custom_op_files.py +1 -1
  11. mindspore/_extends/parse/compile_config.py +64 -0
  12. mindspore/_extends/parse/deprecated/__init__.py +0 -0
  13. mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +375 -0
  14. mindspore/_extends/parse/parser.py +23 -5
  15. mindspore/_extends/parse/standard_method.py +123 -27
  16. mindspore/_extends/pijit/pijit_func_white_list.py +1 -1
  17. mindspore/amp.py +7 -1
  18. mindspore/boost/boost_cell_wrapper.py +136 -41
  19. mindspore/common/__init__.py +3 -1
  20. mindspore/common/_register_for_tensor.py +0 -1
  21. mindspore/common/_stub_tensor.py +25 -4
  22. mindspore/common/_tensor_cpp_method.py +17 -0
  23. mindspore/common/_tensor_docs.py +6132 -0
  24. mindspore/common/api.py +99 -25
  25. mindspore/common/dtype.py +34 -34
  26. mindspore/common/dump.py +2 -1
  27. mindspore/common/file_system.py +8 -1
  28. mindspore/common/generator.py +2 -0
  29. mindspore/common/hook_handle.py +3 -1
  30. mindspore/common/initializer.py +3 -4
  31. mindspore/common/lazy_inline.py +8 -2
  32. mindspore/common/mindir_util.py +10 -2
  33. mindspore/common/parameter.py +30 -27
  34. mindspore/common/tensor.py +713 -1337
  35. mindspore/communication/__init__.py +1 -1
  36. mindspore/communication/_comm_helper.py +10 -0
  37. mindspore/communication/comm_func.py +215 -173
  38. mindspore/communication/management.py +23 -20
  39. mindspore/context.py +292 -193
  40. mindspore/dataset/__init__.py +23 -19
  41. mindspore/dataset/callback/ds_callback.py +2 -1
  42. mindspore/dataset/core/config.py +84 -3
  43. mindspore/dataset/engine/cache_admin.py +3 -3
  44. mindspore/dataset/engine/cache_client.py +5 -4
  45. mindspore/dataset/engine/datasets.py +192 -149
  46. mindspore/dataset/engine/datasets_audio.py +14 -0
  47. mindspore/dataset/engine/datasets_standard_format.py +28 -11
  48. mindspore/dataset/engine/datasets_text.py +38 -1
  49. mindspore/dataset/engine/datasets_user_defined.py +125 -65
  50. mindspore/dataset/engine/datasets_vision.py +81 -8
  51. mindspore/dataset/engine/iterators.py +281 -63
  52. mindspore/dataset/engine/obs/util.py +8 -0
  53. mindspore/dataset/engine/queue.py +40 -0
  54. mindspore/dataset/engine/samplers.py +26 -2
  55. mindspore/dataset/engine/serializer_deserializer.py +1 -1
  56. mindspore/dataset/engine/validators.py +43 -11
  57. mindspore/dataset/transforms/py_transforms_util.py +17 -0
  58. mindspore/dataset/transforms/transforms.py +29 -12
  59. mindspore/dataset/vision/validators.py +1 -2
  60. mindspore/device_context/__init__.py +21 -0
  61. mindspore/device_context/ascend/__init__.py +25 -0
  62. mindspore/device_context/ascend/device.py +72 -0
  63. mindspore/device_context/ascend/op_debug.py +94 -0
  64. mindspore/device_context/ascend/op_precision.py +193 -0
  65. mindspore/device_context/ascend/op_tuning.py +127 -0
  66. mindspore/device_context/cpu/__init__.py +25 -0
  67. mindspore/device_context/cpu/device.py +62 -0
  68. mindspore/device_context/cpu/op_tuning.py +43 -0
  69. mindspore/device_context/gpu/__init__.py +21 -0
  70. mindspore/device_context/gpu/device.py +70 -0
  71. mindspore/device_context/gpu/op_precision.py +67 -0
  72. mindspore/device_context/gpu/op_tuning.py +175 -0
  73. mindspore/device_manager.py +134 -0
  74. mindspore/experimental/llm_boost/__init__.py +3 -2
  75. mindspore/experimental/llm_boost/ascend_native/__init__.py +22 -0
  76. mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +211 -0
  77. mindspore/experimental/llm_boost/ascend_native/llm_boost.py +52 -0
  78. mindspore/experimental/llm_boost/atb/boost_base.py +239 -64
  79. mindspore/experimental/llm_boost/atb/llama_boost.py +52 -30
  80. mindspore/experimental/llm_boost/atb/qwen_boost.py +47 -24
  81. mindspore/experimental/llm_boost/register.py +1 -0
  82. mindspore/experimental/optim/adadelta.py +26 -22
  83. mindspore/experimental/optim/adam.py +3 -0
  84. mindspore/experimental/optim/lr_scheduler.py +33 -24
  85. mindspore/experimental/optim/radam.py +33 -30
  86. mindspore/hal/device.py +28 -0
  87. mindspore/hal/event.py +17 -0
  88. mindspore/hal/memory.py +94 -3
  89. mindspore/hal/stream.py +91 -6
  90. mindspore/include/api/context.h +1 -2
  91. mindspore/include/dataset/constants.h +2 -2
  92. mindspore/lib/libavcodec.so.59 +0 -0
  93. mindspore/lib/libavdevice.so.59 +0 -0
  94. mindspore/lib/libavfilter.so.8 +0 -0
  95. mindspore/lib/libavformat.so.59 +0 -0
  96. mindspore/lib/libavutil.so.57 +0 -0
  97. mindspore/lib/libdnnl.so.2 +0 -0
  98. mindspore/lib/libmindspore_backend.so +0 -0
  99. mindspore/lib/libmindspore_common.so +0 -0
  100. mindspore/lib/libmindspore_core.so +0 -0
  101. mindspore/lib/libmindspore_glog.so.0 +0 -0
  102. mindspore/lib/libmindspore_gpr.so.15 +0 -0
  103. mindspore/lib/libmindspore_grpc++.so.1 +0 -0
  104. mindspore/lib/libmindspore_grpc.so.15 +0 -0
  105. mindspore/lib/libmindspore_ops.so +0 -0
  106. mindspore/lib/libmpi_adapter.so +0 -0
  107. mindspore/lib/libmpi_collective.so +0 -0
  108. mindspore/lib/libnnacl.so +0 -0
  109. mindspore/lib/libopencv_core.so.4.5 +0 -0
  110. mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
  111. mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
  112. mindspore/lib/libps_cache.so +0 -0
  113. mindspore/lib/libswresample.so.4 +0 -0
  114. mindspore/lib/libswscale.so.6 +0 -0
  115. mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910_93/aic-ascend910_93-ops-info.json +2048 -0
  116. mindspore/lib/plugin/ascend/custom_aicore_ops/op_proto/libop_proto.so +0 -0
  117. mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_cpu_kernels.so +0 -0
  118. mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.so +0 -0
  119. mindspore/lib/plugin/ascend/custom_ascendc_910/framework/npu_supported_ops.json +10 -0
  120. mindspore/lib/plugin/ascend/custom_ascendc_910/op_api/lib/libcust_opapi.so +0 -0
  121. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json +182 -0
  122. mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl}/dynamic/decoder_kv_cache.py +51 -16
  123. mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl}/dynamic/prompt_kv_cache.py +51 -16
  124. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +158 -0
  125. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
  126. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +158 -0
  127. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
  128. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +158 -0
  129. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
  130. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +158 -0
  131. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
  132. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +158 -0
  133. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
  134. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +158 -0
  135. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
  136. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +158 -0
  137. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
  138. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +158 -0
  139. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
  140. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +167 -0
  141. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
  142. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +167 -0
  143. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
  144. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +167 -0
  145. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
  146. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +167 -0
  147. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
  148. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +167 -0
  149. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
  150. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +167 -0
  151. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
  152. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +167 -0
  153. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
  154. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +167 -0
  155. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
  156. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/config/ascend910/binary_info_config.json +302 -0
  157. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/config/ascend910/decoder_kv_cache.json +892 -0
  158. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/config/ascend910/prompt_kv_cache.json +892 -0
  159. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/op_tiling/lib/linux/x86_64/libcust_opmaster_rt2.0.so +0 -0
  160. mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
  161. mindspore/lib/plugin/ascend/custom_ascendc_910/op_proto/inc/op_proto.h +33 -0
  162. mindspore/lib/plugin/ascend/custom_ascendc_910/op_proto/lib/linux/x86_64/libcust_opsproto_rt2.0.so +0 -0
  163. mindspore/lib/plugin/ascend/custom_ascendc_910/version.info +1 -0
  164. mindspore/lib/plugin/ascend/custom_ascendc_910b/framework/npu_supported_ops.json +14 -0
  165. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/include/aclnn_decoder_kv_cache.h +59 -0
  166. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/include/aclnn_prompt_kv_cache.h +59 -0
  167. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/lib/libcust_opapi.so +0 -0
  168. mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl}/dynamic/all_finite.py +51 -16
  169. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.cpp +192 -0
  170. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.py +215 -0
  171. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.cpp +274 -0
  172. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.py +215 -0
  173. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +80 -0
  174. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.o +0 -0
  175. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.json +80 -0
  176. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.o +0 -0
  177. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.json +80 -0
  178. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.o +0 -0
  179. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +158 -0
  180. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
  181. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +158 -0
  182. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
  183. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +158 -0
  184. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
  185. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +158 -0
  186. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
  187. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +158 -0
  188. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
  189. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +158 -0
  190. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
  191. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +158 -0
  192. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
  193. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +158 -0
  194. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
  195. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +167 -0
  196. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
  197. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +167 -0
  198. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
  199. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +167 -0
  200. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
  201. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +167 -0
  202. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
  203. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +167 -0
  204. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
  205. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +167 -0
  206. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
  207. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +167 -0
  208. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
  209. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +167 -0
  210. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
  211. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +78 -0
  212. mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_f55e0ebaad1f2f572e43677336992fa0.o → custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.o} +0 -0
  213. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.json +78 -0
  214. mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_576ceaeef5870c451cab59af55ea46ad.o → custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.o} +0 -0
  215. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.json +78 -0
  216. mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_86a73ff6e28d734c96bb8d3054f7dd18.o → custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.o} +0 -0
  217. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +156 -0
  218. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
  219. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +156 -0
  220. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
  221. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +156 -0
  222. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
  223. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +156 -0
  224. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
  225. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +156 -0
  226. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
  227. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +156 -0
  228. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
  229. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +156 -0
  230. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
  231. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +156 -0
  232. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
  233. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +165 -0
  234. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
  235. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +165 -0
  236. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
  237. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +165 -0
  238. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
  239. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +165 -0
  240. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
  241. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +165 -0
  242. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
  243. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +165 -0
  244. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
  245. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +165 -0
  246. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
  247. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +165 -0
  248. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
  249. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +78 -0
  250. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.o +0 -0
  251. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.json +78 -0
  252. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.o +0 -0
  253. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.json +78 -0
  254. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.o +0 -0
  255. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +156 -0
  256. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
  257. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +156 -0
  258. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
  259. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +156 -0
  260. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
  261. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +156 -0
  262. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
  263. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +156 -0
  264. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
  265. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +156 -0
  266. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
  267. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +156 -0
  268. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
  269. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +156 -0
  270. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
  271. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +165 -0
  272. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
  273. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +165 -0
  274. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
  275. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +165 -0
  276. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
  277. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +165 -0
  278. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
  279. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +165 -0
  280. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
  281. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +165 -0
  282. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
  283. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +165 -0
  284. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
  285. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +165 -0
  286. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
  287. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/all_finite.json +139 -0
  288. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/binary_info_config.json +361 -0
  289. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/decoder_kv_cache.json +892 -0
  290. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/prompt_kv_cache.json +892 -0
  291. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/all_finite.json +139 -0
  292. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/binary_info_config.json +361 -0
  293. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/decoder_kv_cache.json +892 -0
  294. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/prompt_kv_cache.json +892 -0
  295. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/all_finite.json +139 -0
  296. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/binary_info_config.json +361 -0
  297. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/decoder_kv_cache.json +892 -0
  298. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/prompt_kv_cache.json +892 -0
  299. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/op_tiling/lib/linux/x86_64/libcust_opmaster_rt2.0.so +0 -0
  300. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
  301. mindspore/lib/plugin/ascend/custom_ascendc_910b/op_proto/lib/linux/x86_64/libcust_opsproto_rt2.0.so +0 -0
  302. mindspore/lib/plugin/ascend/custom_ascendc_910b/version.info +1 -0
  303. mindspore/lib/plugin/ascend/custom_compiler/setup.py +1 -1
  304. mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
  305. mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
  306. mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
  307. mindspore/lib/plugin/ascend/liblowlatency_collective.so +0 -0
  308. mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
  309. mindspore/lib/plugin/ascend/libmindspore_internal_kernels.so +0 -0
  310. mindspore/lib/plugin/ascend/libms_ascend_native_boost.so +0 -0
  311. mindspore/lib/plugin/ascend/libms_atb_boost.so +0 -0
  312. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/bin/PkgInspect +0 -0
  313. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/bin/op_man +0 -0
  314. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/device/ascend910b/bin/ascend910b.bin +960 -958
  315. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/host/libasdops_cann_host.so +0 -0
  316. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/host/libasdops_host.so +0 -0
  317. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libasdops.so +0 -0
  318. mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libasdops_static.a +0 -0
  319. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/{acme/include/base_type.h → base_type.h} +25 -20
  320. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/{cast/cast_tiling.h → internal.h} +6 -4
  321. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_op.h +114 -0
  322. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/boost_kernel.h +70 -0
  323. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/llama_impl.h +85 -0
  324. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/model_interface.h +52 -0
  325. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/tensor.h +81 -0
  326. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_creator.h +123 -0
  327. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_param.h +155 -110
  328. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/{acme/include/tiling_info.h → tiling_info.h} +12 -9
  329. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tiling_utils.h +178 -0
  330. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_layer_norm_op.so +0 -0
  331. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_op.so +0 -0
  332. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_quant_op.so +0 -0
  333. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_310p_op.so +0 -0
  334. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_op.so +0 -0
  335. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libcast_op.so +0 -0
  336. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libcompare_op.so +0 -0
  337. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libgelu_op.so +0 -0
  338. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libllama_op.so +0 -0
  339. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmatmul_op.so +0 -0
  340. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libms_kernels_internal.so +0 -0
  341. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libms_optiling.so +0 -0
  342. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmulti_weight_matmul_kernel_op.so +0 -0
  343. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_nz_op.so +0 -0
  344. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_op.so +0 -0
  345. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/librms_norm_op.so +0 -0
  346. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_f16_nz/internal_pp_matmul_f16_nz.o +0 -0
  347. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_f16_nz/internal_pp_matmul_f16_nz_0.o +0 -0
  348. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_i8_nz_compress/internal_pp_matmul_i8_nz_compress.o +0 -0
  349. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_i8_nz_compress/internal_pp_matmul_i8_nz_compress_0.o +0 -0
  350. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_int8_nz/internal_pp_matmul_int8_nz.o +0 -0
  351. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_int8_nz/internal_pp_matmul_int8_nz_0.o +0 -0
  352. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libadd_rms_norm_quant_ascend310p.so +0 -0
  353. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libapply_rotary_pos_emb_310p_impl.so → op_kernels/ascend310p/so_kernels/libapply_rotary_pos_emb_310p_ascend310p.so} +0 -0
  354. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libcast_ascend310p.so +0 -0
  355. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libcompare_ascend310p.so +0 -0
  356. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libgelu_ascend310p.so +0 -0
  357. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libmatmul_ascend310p.so +0 -0
  358. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libreshape_and_cache_nz_ascend310p.so +0 -0
  359. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_4b60f88cdc28b25a36bad2d8b0a88092.json +163 -0
  360. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_4b60f88cdc28b25a36bad2d8b0a88092.o +0 -0
  361. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_cde61da2bd6fededcb1ba310a6ad16ee.json +163 -0
  362. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_cde61da2bd6fededcb1ba310a6ad16ee.o +0 -0
  363. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bnsd_full_mix.o +0 -0
  364. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bnsd_tri_mix.o +0 -0
  365. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bsh_full_mix.o +0 -0
  366. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bsh_tri_mix.o +0 -0
  367. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bnsd_full_mix.o +0 -0
  368. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bnsd_tri_mix.o +0 -0
  369. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bsh_full_mix.o +0 -0
  370. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bsh_tri_mix.o +0 -0
  371. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_matmul_postfusion_mix/internal_matmul_postfusion_mix.o +0 -0
  372. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_matmul_postfusion_mix/internal_matmul_postfusion_mix_mix_aic_0.o +0 -0
  373. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_matmul_postfusion_mix/internal_matmul_postfusion_mix_mix_aiv_0.o +0 -0
  374. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_multi_weight_matmul_postfusion_mix/internal_multi_weight_matmul_postfusion_mix.o +0 -0
  375. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_multi_weight_matmul_postfusion_mix/internal_multi_weight_matmul_postfusion_mix_mix_aic_0.o +0 -0
  376. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_multi_weight_matmul_postfusion_mix/internal_multi_weight_matmul_postfusion_mix_mix_aiv_0.o +0 -0
  377. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/matmul_add_rmsnorm/matmul_add_rmsnorm_bf16_bf16.o +0 -0
  378. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/matmul_add_rmsnorm/matmul_add_rmsnorm_bf16_fp16.o +0 -0
  379. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/matmul_add_rmsnorm/matmul_add_rmsnorm_bf16_fp32.o +0 -0
  380. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/matmul_add_rmsnorm/matmul_add_rmsnorm_fp16_bf16.o +0 -0
  381. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/matmul_add_rmsnorm/matmul_add_rmsnorm_fp16_fp16.o +0 -0
  382. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/matmul_add_rmsnorm/matmul_add_rmsnorm_fp16_fp32.o +0 -0
  383. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/paged_attention_v2/paged_attention_v2.o +0 -0
  384. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/paged_attention_v2/paged_attention_v2_mix_aic_0.o +0 -0
  385. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/paged_attention_v2/paged_attention_v2_mix_aiv_0.o +0 -0
  386. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libadd_layer_norm_ascend910b.so +0 -0
  387. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libadd_rms_norm_impl.so → op_kernels/ascend910b/so_kernels/libadd_rms_norm_ascend910b.so} +0 -0
  388. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libadd_rms_norm_quant_ascend910b.so +0 -0
  389. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libapply_rotary_pos_emb_impl.so → op_kernels/ascend910b/so_kernels/libapply_rotary_pos_emb_ascend910b.so} +0 -0
  390. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libcast_impl.so → op_kernels/ascend910b/so_kernels/libcast_ascend910b.so} +0 -0
  391. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libnot_equal_impl.so → op_kernels/ascend910b/so_kernels/libcompare_ascend910b.so} +0 -0
  392. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libgelu_impl.so → op_kernels/ascend910b/so_kernels/libgelu_ascend910b.so} +0 -0
  393. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libllama_ascend910b.so +0 -0
  394. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libmatmul_impl.so → op_kernels/ascend910b/so_kernels/libmatmul_ascend910b.so} +0 -0
  395. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libmulti_weight_matmul_kernel_impl.so → op_kernels/ascend910b/so_kernels/libmulti_weight_matmul_kernel_ascend910b.so} +0 -0
  396. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libreshape_and_cache_impl.so → op_kernels/ascend910b/so_kernels/libreshape_and_cache_ascend910b.so} +0 -0
  397. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/librms_norm_impl.so → op_kernels/ascend910b/so_kernels/librms_norm_ascend910b.so} +0 -0
  398. mindspore/lib/plugin/ascend/ms_kernels_internal/lccl/lib/liblccl_wrapper.so +0 -0
  399. mindspore/lib/plugin/gpu/libcuda_ops.so.10 +0 -0
  400. mindspore/lib/plugin/gpu/libcuda_ops.so.11 +0 -0
  401. mindspore/lib/plugin/gpu10.1/libnccl.so.2 +0 -0
  402. mindspore/lib/plugin/gpu10.1/libnvidia_collective.so +0 -0
  403. mindspore/lib/plugin/gpu11.1/libnccl.so.2 +0 -0
  404. mindspore/lib/plugin/gpu11.1/libnvidia_collective.so +0 -0
  405. mindspore/lib/plugin/gpu11.6/libnccl.so.2 +0 -0
  406. mindspore/lib/plugin/gpu11.6/libnvidia_collective.so +0 -0
  407. mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
  408. mindspore/lib/plugin/libmindspore_gpu.so.10.1 +0 -0
  409. mindspore/lib/plugin/libmindspore_gpu.so.11.1 +0 -0
  410. mindspore/lib/plugin/libmindspore_gpu.so.11.6 +0 -0
  411. mindspore/log.py +12 -0
  412. mindspore/mindrecord/__init__.py +1 -1
  413. mindspore/mindrecord/config.py +17 -316
  414. mindspore/mindrecord/filereader.py +1 -9
  415. mindspore/mindrecord/filewriter.py +5 -15
  416. mindspore/mindrecord/mindpage.py +1 -9
  417. mindspore/mint/__init__.py +824 -218
  418. mindspore/mint/distributed/__init__.py +66 -4
  419. mindspore/mint/distributed/distributed.py +2594 -44
  420. mindspore/mint/linalg/__init__.py +6 -0
  421. mindspore/mint/nn/__init__.py +473 -14
  422. mindspore/mint/nn/functional.py +486 -11
  423. mindspore/mint/nn/layer/__init__.py +17 -4
  424. mindspore/mint/nn/layer/_functions.py +330 -0
  425. mindspore/mint/nn/layer/activation.py +169 -1
  426. mindspore/mint/nn/layer/basic.py +123 -0
  427. mindspore/mint/nn/layer/conv.py +727 -0
  428. mindspore/mint/nn/layer/normalization.py +215 -19
  429. mindspore/mint/nn/layer/padding.py +797 -0
  430. mindspore/mint/nn/layer/pooling.py +170 -0
  431. mindspore/mint/optim/__init__.py +2 -1
  432. mindspore/mint/optim/adam.py +223 -0
  433. mindspore/mint/optim/adamw.py +26 -19
  434. mindspore/mint/special/__init__.py +2 -1
  435. mindspore/multiprocessing/__init__.py +5 -0
  436. mindspore/nn/__init__.py +2 -0
  437. mindspore/nn/cell.py +142 -21
  438. mindspore/nn/dynamic_lr.py +2 -1
  439. mindspore/nn/layer/activation.py +6 -6
  440. mindspore/nn/layer/basic.py +35 -25
  441. mindspore/nn/layer/channel_shuffle.py +3 -3
  442. mindspore/nn/layer/conv.py +3 -0
  443. mindspore/nn/layer/embedding.py +3 -3
  444. mindspore/nn/layer/normalization.py +8 -7
  445. mindspore/nn/layer/padding.py +4 -3
  446. mindspore/nn/layer/pooling.py +55 -23
  447. mindspore/nn/layer/rnn_cells.py +1 -1
  448. mindspore/nn/layer/rnns.py +2 -1
  449. mindspore/nn/layer/timedistributed.py +5 -5
  450. mindspore/nn/layer/transformer.py +48 -26
  451. mindspore/nn/learning_rate_schedule.py +5 -3
  452. mindspore/nn/loss/loss.py +31 -36
  453. mindspore/nn/optim/ada_grad.py +1 -0
  454. mindspore/nn/optim/adadelta.py +2 -2
  455. mindspore/nn/optim/adam.py +1 -1
  456. mindspore/nn/optim/lars.py +1 -4
  457. mindspore/nn/optim/optimizer.py +1 -1
  458. mindspore/nn/optim/rprop.py +2 -2
  459. mindspore/nn/optim/thor.py +2 -1
  460. mindspore/nn/utils/__init__.py +22 -0
  461. mindspore/nn/utils/init.py +73 -0
  462. mindspore/nn/wrap/cell_wrapper.py +4 -6
  463. mindspore/nn/wrap/loss_scale.py +3 -4
  464. mindspore/numpy/array_creations.py +60 -62
  465. mindspore/numpy/array_ops.py +148 -143
  466. mindspore/numpy/logic_ops.py +41 -42
  467. mindspore/numpy/math_ops.py +361 -359
  468. mindspore/numpy/utils.py +16 -16
  469. mindspore/numpy/utils_const.py +4 -4
  470. mindspore/ops/__init__.py +2 -1
  471. mindspore/ops/_grad_experimental/grad_comm_ops.py +107 -8
  472. mindspore/ops/_grad_experimental/grad_debug_ops.py +6 -1
  473. mindspore/ops/_grad_experimental/grad_inner_ops.py +9 -0
  474. mindspore/ops/_grad_experimental/grad_math_ops.py +2 -1
  475. mindspore/ops/_op_impl/cpu/__init__.py +1 -0
  476. mindspore/ops/_op_impl/cpu/raise_op.py +28 -0
  477. mindspore/ops/_vmap/vmap_array_ops.py +20 -19
  478. mindspore/ops/_vmap/vmap_base.py +0 -2
  479. mindspore/ops/_vmap/vmap_grad_nn_ops.py +19 -13
  480. mindspore/ops/_vmap/vmap_math_ops.py +11 -9
  481. mindspore/ops/_vmap/vmap_nn_ops.py +20 -34
  482. mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +149 -12
  483. mindspore/ops/auto_generate/gen_arg_handler.py +0 -61
  484. mindspore/ops/auto_generate/gen_extend_func.py +554 -60
  485. mindspore/ops/auto_generate/gen_ops_def.py +1621 -115
  486. mindspore/ops/auto_generate/gen_ops_prim.py +8027 -3411
  487. mindspore/ops/auto_generate/pyboost_inner_prim.py +183 -79
  488. mindspore/ops/composite/base.py +1 -1
  489. mindspore/ops/composite/multitype_ops/_compile_utils.py +229 -30
  490. mindspore/ops/composite/multitype_ops/pow_impl.py +0 -29
  491. mindspore/ops/function/__init__.py +12 -0
  492. mindspore/ops/function/array_func.py +561 -159
  493. mindspore/ops/function/clip_func.py +64 -0
  494. mindspore/ops/function/debug_func.py +28 -20
  495. mindspore/ops/function/image_func.py +1 -1
  496. mindspore/ops/function/linalg_func.py +5 -4
  497. mindspore/ops/function/math_func.py +1664 -294
  498. mindspore/ops/function/nn_func.py +988 -317
  499. mindspore/ops/function/parameter_func.py +3 -56
  500. mindspore/ops/function/random_func.py +243 -33
  501. mindspore/ops/function/sparse_unary_func.py +1 -1
  502. mindspore/ops/functional.py +18 -5
  503. mindspore/ops/functional_overload.py +897 -0
  504. mindspore/ops/operations/__init__.py +3 -2
  505. mindspore/ops/operations/_embedding_cache_ops.py +4 -4
  506. mindspore/ops/operations/_grad_ops.py +2 -34
  507. mindspore/ops/operations/_infer_ops.py +2 -1
  508. mindspore/ops/operations/_inner_ops.py +38 -8
  509. mindspore/ops/operations/array_ops.py +45 -303
  510. mindspore/ops/operations/comm_ops.py +23 -17
  511. mindspore/ops/operations/custom_ops.py +7 -49
  512. mindspore/ops/operations/debug_ops.py +42 -47
  513. mindspore/ops/operations/inner_ops.py +6 -4
  514. mindspore/ops/operations/linalg_ops.py +3 -2
  515. mindspore/ops/operations/manually_defined/ops_def.py +185 -104
  516. mindspore/ops/operations/math_ops.py +11 -216
  517. mindspore/ops/operations/nn_ops.py +153 -310
  518. mindspore/ops/primitive.py +23 -21
  519. mindspore/ops/tensor_method.py +1669 -0
  520. mindspore/ops_generate/aclnn_kernel_register_auto_cc_generator.py +110 -0
  521. mindspore/ops_generate/add_tensor_docs_generator.py +54 -0
  522. mindspore/ops_generate/arg_handler.py +0 -61
  523. mindspore/ops_generate/auto_grad_impl_cc_generator.py +135 -0
  524. mindspore/ops_generate/auto_grad_reg_cc_generator.py +93 -0
  525. mindspore/ops_generate/base_generator.py +11 -0
  526. mindspore/ops_generate/cpp_create_prim_instance_helper_generator.py +108 -0
  527. mindspore/ops_generate/functional_map_cpp_generator.py +491 -0
  528. mindspore/ops_generate/functional_overload_py_generator.py +110 -0
  529. mindspore/ops_generate/functions_cc_generator.py +233 -0
  530. mindspore/ops_generate/gen_aclnn_implement.py +110 -114
  531. mindspore/ops_generate/gen_constants.py +157 -3
  532. mindspore/ops_generate/gen_ops.py +245 -990
  533. mindspore/ops_generate/gen_pyboost_func.py +97 -998
  534. mindspore/ops_generate/gen_utils.py +119 -33
  535. mindspore/ops_generate/lite_ops_cpp_generator.py +155 -0
  536. mindspore/ops_generate/op_api_proto.py +206 -0
  537. mindspore/ops_generate/op_def_py_generator.py +131 -0
  538. mindspore/ops_generate/op_prim_py_generator.py +480 -0
  539. mindspore/ops_generate/op_proto.py +373 -108
  540. mindspore/ops_generate/op_template_parser.py +436 -0
  541. mindspore/ops_generate/ops_def_cc_generator.py +288 -0
  542. mindspore/ops_generate/ops_def_h_generator.py +74 -0
  543. mindspore/ops_generate/ops_name_h_generator.py +68 -0
  544. mindspore/ops_generate/ops_primitive_h_generator.py +81 -0
  545. mindspore/ops_generate/pyboost_functions_cpp_generator.py +370 -0
  546. mindspore/ops_generate/pyboost_functions_h_generator.py +68 -0
  547. mindspore/ops_generate/pyboost_functions_py_generator.py +148 -0
  548. mindspore/ops_generate/pyboost_grad_function_cpp_generator.py +154 -0
  549. mindspore/ops_generate/pyboost_inner_prim_generator.py +131 -0
  550. mindspore/ops_generate/pyboost_native_grad_functions_generator.py +268 -0
  551. mindspore/ops_generate/pyboost_op_cpp_code_generator.py +851 -0
  552. mindspore/ops_generate/pyboost_overload_functions_cpp_generator.py +344 -0
  553. mindspore/ops_generate/pyboost_utils.py +92 -33
  554. mindspore/ops_generate/template.py +294 -44
  555. mindspore/ops_generate/tensor_func_reg_cpp_generator.py +422 -0
  556. mindspore/parallel/__init__.py +3 -3
  557. mindspore/parallel/_auto_parallel_context.py +44 -34
  558. mindspore/parallel/_cell_wrapper.py +22 -3
  559. mindspore/parallel/_parallel_serialization.py +13 -2
  560. mindspore/parallel/_utils.py +4 -2
  561. mindspore/parallel/algo_parameter_config.py +1 -1
  562. mindspore/parallel/checkpoint_transform.py +44 -0
  563. mindspore/parallel/cluster/process_entity/_api.py +131 -37
  564. mindspore/parallel/cluster/process_entity/_utils.py +41 -6
  565. mindspore/parallel/cluster/run.py +20 -3
  566. mindspore/parallel/parameter_broadcast.py +1 -1
  567. mindspore/parallel/shard.py +3 -0
  568. mindspore/parallel/transform_safetensors.py +119 -253
  569. mindspore/profiler/__init__.py +17 -4
  570. mindspore/profiler/analysis/__init__.py +0 -0
  571. mindspore/profiler/analysis/parser/__init__.py +0 -0
  572. mindspore/profiler/analysis/parser/ascend_cann_parser.py +166 -0
  573. mindspore/profiler/analysis/parser/base_parser.py +158 -0
  574. mindspore/profiler/analysis/parser/framework_cann_relation_parser.py +45 -0
  575. mindspore/profiler/analysis/parser/ms_framework_parser.py +142 -0
  576. mindspore/profiler/analysis/parser/ms_minddata_parser.py +145 -0
  577. mindspore/profiler/analysis/parser/timeline_assembly_factory/__init__.py +0 -0
  578. mindspore/profiler/analysis/parser/timeline_assembly_factory/ascend_timeline_assembler.py +261 -0
  579. mindspore/profiler/analysis/parser/timeline_assembly_factory/base_timeline_assembler.py +40 -0
  580. mindspore/profiler/analysis/parser/timeline_assembly_factory/trace_view_container.py +84 -0
  581. mindspore/profiler/analysis/parser/timeline_creator/__init__.py +0 -0
  582. mindspore/profiler/analysis/parser/timeline_creator/base_timeline_creator.py +44 -0
  583. mindspore/profiler/analysis/parser/timeline_creator/cpu_op_timeline_creator.py +90 -0
  584. mindspore/profiler/analysis/parser/timeline_creator/fwk_timeline_creator.py +76 -0
  585. mindspore/profiler/analysis/parser/timeline_creator/msprof_timeline_creator.py +103 -0
  586. mindspore/profiler/analysis/parser/timeline_creator/scope_layer_timeline_creator.py +134 -0
  587. mindspore/profiler/analysis/parser/timeline_event/__init__.py +0 -0
  588. mindspore/profiler/analysis/parser/timeline_event/base_event.py +233 -0
  589. mindspore/profiler/analysis/parser/timeline_event/cpu_op_event.py +47 -0
  590. mindspore/profiler/analysis/parser/timeline_event/flow_event.py +36 -0
  591. mindspore/profiler/analysis/parser/timeline_event/fwk_event.py +260 -0
  592. mindspore/profiler/analysis/parser/timeline_event/msprof_event.py +73 -0
  593. mindspore/profiler/analysis/parser/timeline_event/scope_layer_event.py +53 -0
  594. mindspore/profiler/analysis/parser/timeline_event/timeline_event_pool.py +146 -0
  595. mindspore/profiler/analysis/task_manager.py +131 -0
  596. mindspore/profiler/analysis/time_converter.py +84 -0
  597. mindspore/profiler/analysis/viewer/__init__.py +0 -0
  598. mindspore/profiler/analysis/viewer/ascend_communication_viewer.py +333 -0
  599. mindspore/profiler/analysis/viewer/ascend_integrate_viewer.py +87 -0
  600. mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +252 -0
  601. mindspore/profiler/analysis/viewer/ascend_memory_viewer.py +313 -0
  602. mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +322 -0
  603. mindspore/profiler/analysis/viewer/ascend_step_trace_time_viewer.py +265 -0
  604. mindspore/profiler/analysis/viewer/ascend_timeline_viewer.py +58 -0
  605. mindspore/profiler/analysis/viewer/base_viewer.py +26 -0
  606. mindspore/profiler/analysis/viewer/ms_dataset_viewer.py +97 -0
  607. mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +581 -0
  608. mindspore/profiler/analysis/work_flow.py +73 -0
  609. mindspore/profiler/common/ascend_msprof_exporter.py +138 -0
  610. mindspore/profiler/common/command_executor.py +90 -0
  611. mindspore/profiler/common/constant.py +174 -3
  612. mindspore/profiler/common/file_manager.py +208 -0
  613. mindspore/profiler/common/log.py +130 -0
  614. mindspore/profiler/common/msprof_cmd_tool.py +202 -0
  615. mindspore/profiler/common/path_manager.py +371 -0
  616. mindspore/profiler/common/process_bar.py +168 -0
  617. mindspore/profiler/common/process_pool.py +9 -3
  618. mindspore/profiler/common/profiler_context.py +476 -0
  619. mindspore/profiler/common/profiler_info.py +304 -0
  620. mindspore/profiler/common/profiler_output_path.py +284 -0
  621. mindspore/profiler/common/profiler_parameters.py +210 -0
  622. mindspore/profiler/common/profiler_path_manager.py +120 -0
  623. mindspore/profiler/common/record_function.py +76 -0
  624. mindspore/profiler/common/tlv_decoder.py +76 -0
  625. mindspore/profiler/common/util.py +75 -2
  626. mindspore/profiler/dynamic_profiler.py +270 -37
  627. mindspore/profiler/envprofiler.py +138 -0
  628. mindspore/profiler/mstx.py +199 -0
  629. mindspore/profiler/platform/__init__.py +21 -0
  630. mindspore/profiler/platform/base_profiler.py +40 -0
  631. mindspore/profiler/platform/cpu_profiler.py +124 -0
  632. mindspore/profiler/platform/gpu_profiler.py +74 -0
  633. mindspore/profiler/platform/npu_profiler.py +309 -0
  634. mindspore/profiler/profiler.py +580 -93
  635. mindspore/profiler/profiler_action_controller.py +187 -0
  636. mindspore/profiler/profiler_interface.py +114 -0
  637. mindspore/profiler/schedule.py +208 -0
  638. mindspore/rewrite/api/symbol_tree.py +1 -2
  639. mindspore/run_check/_check_version.py +18 -13
  640. mindspore/runtime/__init__.py +37 -0
  641. mindspore/runtime/device.py +27 -0
  642. mindspore/runtime/event.py +209 -0
  643. mindspore/runtime/executor.py +148 -0
  644. mindspore/runtime/memory.py +392 -0
  645. mindspore/runtime/stream.py +460 -0
  646. mindspore/runtime/thread_bind_core.py +401 -0
  647. mindspore/train/__init__.py +2 -2
  648. mindspore/train/_utils.py +53 -18
  649. mindspore/train/amp.py +8 -4
  650. mindspore/train/callback/_checkpoint.py +32 -18
  651. mindspore/train/callback/_early_stop.py +1 -1
  652. mindspore/train/callback/_flops_collector.py +105 -69
  653. mindspore/train/callback/_history.py +1 -1
  654. mindspore/train/callback/_summary_collector.py +44 -6
  655. mindspore/train/callback/_tft_register.py +37 -15
  656. mindspore/train/dataset_helper.py +11 -11
  657. mindspore/train/metrics/precision.py +4 -5
  658. mindspore/train/mind_ir_pb2.py +167 -46
  659. mindspore/train/model.py +13 -14
  660. mindspore/train/serialization.py +461 -72
  661. mindspore/train/summary/summary_record.py +1 -2
  662. mindspore/train/train_thor/model_thor.py +1 -1
  663. mindspore/utils/__init__.py +4 -2
  664. mindspore/utils/bin/dataset-cache +0 -0
  665. mindspore/utils/bin/dataset-cache-server +0 -0
  666. mindspore/utils/dryrun.py +138 -0
  667. mindspore/utils/runtime_execution_order_check.py +550 -0
  668. mindspore/version.py +1 -1
  669. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/METADATA +3 -4
  670. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/RECORD +683 -490
  671. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/entry_points.txt +1 -1
  672. mindspore/_data_dump.cpython-310-x86_64-linux-gnu.so +0 -0
  673. mindspore/bin/cache_admin +0 -0
  674. mindspore/bin/cache_server +0 -0
  675. mindspore/common/_tensor_overload.py +0 -139
  676. mindspore/lib/libmindspore_np_dtype.so +0 -0
  677. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_api/lib/libcust_opapi.so +0 -0
  678. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_576ceaeef5870c451cab59af55ea46ad.json +0 -58
  679. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_86a73ff6e28d734c96bb8d3054f7dd18.json +0 -58
  680. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_f55e0ebaad1f2f572e43677336992fa0.json +0 -58
  681. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/kernel/config/ascend910b/all_finite.json +0 -109
  682. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/kernel/config/ascend910b/binary_info_config.json +0 -38
  683. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/op_tiling/lib/linux/x86_64/libcust_opmaster_rt2.0.so +0 -0
  684. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
  685. mindspore/lib/plugin/ascend/custom_ascendc_ops/op_proto/lib/linux/x86_64/libcust_opsproto_rt2.0.so +0 -0
  686. mindspore/lib/plugin/ascend/custom_ascendc_ops/version.info +0 -1
  687. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/acme.h +0 -24
  688. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/acme_op.h +0 -82
  689. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/op_creator.h +0 -113
  690. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/op_param.h +0 -193
  691. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/dtype_registry.h +0 -90
  692. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/kernel_register.h +0 -46
  693. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/platform/platform_configs.h +0 -89
  694. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/platform/rt_funcs.h +0 -135
  695. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/add_layer_norm_op.h +0 -60
  696. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/add_rms_norm_op.h +0 -50
  697. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/add_rms_norm_quant_op.h +0 -50
  698. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/apply_rotary_pos_emb_nz_op.h +0 -42
  699. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/apply_rotary_pos_emb_op.h +0 -55
  700. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/asd_elewise_op.h +0 -34
  701. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/asd_only_ops.h +0 -94
  702. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/asd_op_base.h +0 -97
  703. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/cast_op.h +0 -52
  704. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/flash_attention_score_op.h +0 -92
  705. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/gelu_op.h +0 -44
  706. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/matmul_add_rmsnorm_op.h +0 -73
  707. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/matmul_op.h +0 -108
  708. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/multi_impls_op.h +0 -64
  709. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/multi_weight_matmul_op.h +0 -91
  710. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/paged_attention_op.h +0 -99
  711. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/reshape_and_cache_nz_op.h +0 -44
  712. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/reshape_and_cache_op.h +0 -44
  713. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/rms_norm_op.h +0 -64
  714. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/asd_utils.h +0 -179
  715. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/comm_utils.h +0 -69
  716. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/profiling_util.h +0 -366
  717. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/add/add_impl.h +0 -56
  718. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/add/kernel/add.h +0 -21
  719. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/add/tiling/add_tiling.h +0 -43
  720. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/apply_rotary_pos_emb_impl.h +0 -46
  721. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb.h +0 -23
  722. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_base.h +0 -456
  723. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_bf16.h +0 -217
  724. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp.h +0 -391
  725. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp16.h +0 -126
  726. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp32.h +0 -230
  727. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_tiling.h +0 -43
  728. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_value.h +0 -27
  729. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/apply_rotary_pos_emb_nz_impl.h +0 -34
  730. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz.h +0 -23
  731. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_base.h +0 -460
  732. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_fp16.h +0 -116
  733. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_fp32.h +0 -230
  734. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_tiling.h +0 -43
  735. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_value.h +0 -27
  736. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/asdop/asd_op_impl.h +0 -74
  737. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/backend_param.h +0 -74
  738. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/cast/cast_impl.h +0 -48
  739. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/cast/kernel/cast_kernel.h +0 -21
  740. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/compare/compare_impl.h +0 -55
  741. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/compare/compare_tiling.h +0 -27
  742. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/compare/kernel/compare_kernel.h +0 -23
  743. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/and_impl.h +0 -29
  744. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/div_impl.h +0 -29
  745. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/elewise_binary_impl.h +0 -48
  746. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/elewise_binary_tiling.h +0 -25
  747. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/and_kernel.h +0 -46
  748. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/div_kernel.h +0 -46
  749. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/elewise_binary_base.h +0 -260
  750. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/elewise_binary_kernel.h +0 -35
  751. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/max_kernel.h +0 -66
  752. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/min_kernel.h +0 -66
  753. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/mul_kernel.h +0 -66
  754. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/or_kernel.h +0 -46
  755. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/max_impl.h +0 -29
  756. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/min_impl.h +0 -29
  757. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/mul_impl.h +0 -29
  758. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/or_impl.h +0 -29
  759. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/abs_impl.h +0 -29
  760. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/elewise_unary_impl.h +0 -47
  761. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/elewise_unary_tiling.h +0 -24
  762. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/exp_impl.h +0 -29
  763. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/abs_kernel.h +0 -45
  764. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/elewise_unary_base.h +0 -148
  765. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/elewise_unary_kernel.h +0 -31
  766. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/exp_kernel.h +0 -45
  767. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/ln_kernel.h +0 -45
  768. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/not_kernel.h +0 -45
  769. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/reciprocal_kernel.h +0 -45
  770. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/relu_kernel.h +0 -55
  771. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/rsqrt_kernel.h +0 -45
  772. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/sqrt_kernel.h +0 -45
  773. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/ln_impl.h +0 -29
  774. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/not_impl.h +0 -29
  775. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/reciprocal_impl.h +0 -29
  776. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/relu_impl.h +0 -29
  777. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/rsqrt_impl.h +0 -29
  778. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/sqrt_impl.h +0 -29
  779. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/flash_attention_score/flash_attention_score_impl.h +0 -68
  780. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_kernel.h +0 -99
  781. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_rtbackend.h +0 -21
  782. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/lccl/lccl_wrapper.h +0 -58
  783. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/ms_int_types.h +0 -91
  784. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/ms_int_utils.h +0 -108
  785. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/paged_attention/paged_attention_impl.h +0 -64
  786. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/add_param.h +0 -68
  787. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/attention_param.h +0 -40
  788. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/cast_param.h +0 -30
  789. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/compare_param.h +0 -31
  790. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/elewise_param.h +0 -41
  791. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/grouped_matmul_param.h +0 -40
  792. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/matmul_ext_param.h +0 -38
  793. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/matmul_qkv_param.h +0 -42
  794. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/sub_param.h +0 -33
  795. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/profiling_util.h +0 -377
  796. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/reshape_and_cache_nz/kernel/reshape_and_cache_nz.h +0 -24
  797. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/reshape_and_cache_nz/reshape_and_cache_nz_impl.h +0 -42
  798. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/reshape_and_cache_nz/reshape_and_cache_nz_tiling.h +0 -27
  799. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/rms_norm/rms_norm_impl.h +0 -46
  800. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/sub/kernel/sub_kernel.h +0 -20
  801. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/sub/sub_impl.h +0 -48
  802. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/sub/sub_tiling.h +0 -25
  803. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tune_repo/matmul_table.h +0 -399
  804. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tune_repo/utils.h +0 -41
  805. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/backend.h +0 -45
  806. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/elewise_tiling.h +0 -29
  807. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/elewise_utils.h +0 -30
  808. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log.h +0 -69
  809. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_core.h +0 -43
  810. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_entity.h +0 -38
  811. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_sink.h +0 -69
  812. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_stream.h +0 -41
  813. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_tiling.h +0 -71
  814. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_utils.h +0 -165
  815. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/math.h +0 -20
  816. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/register/kernel_creator.h +0 -39
  817. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/register/kernel_registry.h +0 -121
  818. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/utils.h +0 -106
  819. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libAdd_impl.so +0 -0
  820. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libSub_impl.so +0 -0
  821. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_layer_norm_impl.so +0 -0
  822. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_quant_acme_impl.so +0 -0
  823. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_310p_old_impl.so +0 -0
  824. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_old_impl.so +0 -0
  825. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_nz_impl.so +0 -0
  826. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_nz_old_impl.so +0 -0
  827. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix.json +0 -19
  828. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix.o +0 -0
  829. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix_mix_aic_0.o +0 -0
  830. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix_mix_aiv_0.o +0 -0
  831. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix.json +0 -19
  832. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix.o +0 -0
  833. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix_mix_aic_0.o +0 -0
  834. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix_mix_aiv_0.o +0 -0
  835. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bnsd_full_mix.o +0 -0
  836. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bnsd_tri_mix.o +0 -0
  837. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bsh_full_mix.o +0 -0
  838. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bsh_tri_mix.o +0 -0
  839. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bnsd_full_mix.o +0 -0
  840. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bnsd_tri_mix.o +0 -0
  841. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bsh_full_mix.o +0 -0
  842. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bsh_tri_mix.o +0 -0
  843. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/matmul_add_rmsnorm/matmul_add_rmsnorm_bf16_bf16.o +0 -0
  844. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/matmul_add_rmsnorm/matmul_add_rmsnorm_bf16_fp16.o +0 -0
  845. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/matmul_add_rmsnorm/matmul_add_rmsnorm_bf16_fp32.o +0 -0
  846. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/matmul_add_rmsnorm/matmul_add_rmsnorm_fp16_bf16.o +0 -0
  847. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/matmul_add_rmsnorm/matmul_add_rmsnorm_fp16_fp16.o +0 -0
  848. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/matmul_add_rmsnorm/matmul_add_rmsnorm_fp16_fp32.o +0 -0
  849. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_bf16_bnsd_mix.o +0 -0
  850. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_bf16_bsh_mix.o +0 -0
  851. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_fp16_bnsd_mix.o +0 -0
  852. mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_fp16_bsh_mix.o +0 -0
  853. mindspore/profiler/envprofiling.py +0 -254
  854. mindspore/profiler/profiling.py +0 -1926
  855. /mindspore/lib/plugin/ascend/{custom_ascendc_ops → custom_ascendc_910}/op_api/include/aclnn_decoder_kv_cache.h +0 -0
  856. /mindspore/lib/plugin/ascend/{custom_ascendc_ops → custom_ascendc_910}/op_api/include/aclnn_prompt_kv_cache.h +0 -0
  857. /mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl}/dynamic/decoder_kv_cache.cpp +0 -0
  858. /mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl}/dynamic/prompt_kv_cache.cpp +0 -0
  859. /mindspore/lib/plugin/ascend/{custom_ascendc_ops → custom_ascendc_910b}/op_api/include/aclnn_all_finite.h +0 -0
  860. /mindspore/lib/plugin/ascend/{custom_ascendc_ops → custom_ascendc_910b}/op_impl/ai_core/tbe/config/ascend310p/aic-ascend310p-ops-info.json +0 -0
  861. /mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json → custom_ascendc_910b/op_impl/ai_core/tbe/config/ascend910_93/aic-ascend910_93-ops-info.json} +0 -0
  862. /mindspore/lib/plugin/ascend/{custom_ascendc_ops → custom_ascendc_910b}/op_impl/ai_core/tbe/config/ascend910b/aic-ascend910b-ops-info.json +0 -0
  863. /mindspore/lib/plugin/ascend/{custom_ascendc_ops/op_impl/ai_core/tbe/custom_ascendc_ops_impl → custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl}/dynamic/all_finite.cpp +0 -0
  864. /mindspore/lib/plugin/ascend/{custom_ascendc_ops → custom_ascendc_910b}/op_proto/inc/op_proto.h +0 -0
  865. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/WHEEL +0 -0
  866. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,97 @@
1
+ # Copyright 2024 Huawei Technologies Co., Ltd
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ============================================================================
15
+ """ms dataset viewer"""
16
+ import os
17
+ from collections import defaultdict
18
+ from typing import List, Dict, Any
19
+
20
+ from mindspore import log as logger
21
+ from mindspore.profiler.common.constant import FileConstant, EventConstant
22
+ from mindspore.profiler.common.file_manager import FileManager
23
+ from mindspore.profiler.analysis.viewer.base_viewer import BaseViewer
24
+ from mindspore.profiler.analysis.parser.timeline_event.fwk_event import (
25
+ FwkCompleteEvent,
26
+ OpRangeStructField,
27
+ FwkArgsDecoder
28
+ )
29
+ from mindspore.profiler.common.log import ProfilerLogger
30
+
31
+
32
+ class MsDatasetViewer(BaseViewer):
33
+ """Viewer for MindSpore dataset profiling data."""
34
+
35
+ _DATASET_FILE_NAME = 'dataset.csv'
36
+ _DATASET_OP_PREFIX = 'Dataset'
37
+ _COL_NAMES = ['Operation', 'Stage', 'Occurrences', 'Avg. time (us)', 'Custom Info']
38
+
39
+ def __init__(self, **kwargs):
40
+ super().__init__()
41
+ self._save_path = os.path.join(
42
+ kwargs.get("ascend_profiler_output_path"),
43
+ self._DATASET_FILE_NAME
44
+ )
45
+ self._ascend_ms_dir = kwargs.get("ascend_ms_dir")
46
+ ProfilerLogger.init(self._ascend_ms_dir)
47
+ self._logger = ProfilerLogger.get_instance()
48
+
49
+ def save(self, data: Dict[str, Any]) -> None:
50
+ """Process and save dataset profiling data."""
51
+ self._logger.info("MsDatasetViewer start")
52
+ try:
53
+ op_range_list = data.get("mindspore_op_list", [])
54
+ dataset_statistics = self._calculate_data(op_range_list)
55
+ self._save_data(dataset_statistics)
56
+ except Exception as e: # pylint: disable=W0703
57
+ self._logger.error("Failed to save dataset.csv: %s", e, exc_info=True)
58
+ self._logger.info("MsDatasetViewer end")
59
+
60
+ def _save_data(self, dataset_statistics: List[List[Any]]) -> None:
61
+ """Save dataset statistics to a CSV file."""
62
+ if not dataset_statistics:
63
+ return
64
+ self._logger.info("Save dataset statistics start")
65
+ FileManager.create_csv_file(self._save_path, dataset_statistics, self._COL_NAMES)
66
+ self._logger.info("Save dataset statistics done")
67
+
68
+ def _calculate_data(self, fwk_tlv_data: List[Dict]) -> List[List[Any]]:
69
+ """Calculate statistics for dataset operations."""
70
+ dataset_op_data = []
71
+ for data in fwk_tlv_data:
72
+ if (data[FileConstant.FIX_SIZE_DATA][OpRangeStructField.START_NS.value] <
73
+ data[FileConstant.FIX_SIZE_DATA][OpRangeStructField.END_NS.value]): # dur > 0
74
+ name = data.get(FwkArgsDecoder.TLV_TYPES.get(EventConstant.OP_NAME), "")
75
+ if name.split('::')[0] == self._DATASET_OP_PREFIX:
76
+ dataset_op_data.append(FwkCompleteEvent(data))
77
+
78
+ dataset_op_stats = defaultdict(list)
79
+ for op_data in dataset_op_data:
80
+ op_name_list = op_data.name.split('::')
81
+ if len(op_name_list) != 3:
82
+ logger.warning(f"Invalid dataset op name: {op_data.name}")
83
+ continue
84
+ _, event, stage = op_data.name.split('::')
85
+ key = f"{event}::{stage}::{op_data.custom_info}"
86
+ dataset_op_stats[key].append(op_data.dur)
87
+
88
+ dataset_statistics = []
89
+ for key, durations in dataset_op_stats.items():
90
+ event, stage, custom_info = key.split('::')
91
+ occurrence_count = len(durations)
92
+ if occurrence_count == 0:
93
+ continue
94
+ average_duration = round(float(sum(durations) / occurrence_count), 2)
95
+ dataset_statistics.append([event, stage, occurrence_count, average_duration, custom_info])
96
+
97
+ return dataset_statistics
@@ -0,0 +1,581 @@
1
+ # Copyright 2024 Huawei Technologies Co., Ltd
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # http://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ # ============================================================================
15
+ """mind data viewer"""
16
+ import os
17
+ from queue import Queue
18
+ from collections import defaultdict
19
+ from typing import List, Dict, Any
20
+
21
+ from mindspore import log as logger
22
+ from mindspore.profiler.common.constant import ProfilerActivity
23
+ from mindspore.profiler.analysis.viewer.base_viewer import BaseViewer
24
+ from mindspore.profiler.common.file_manager import FileManager
25
+ from mindspore.profiler.common.log import ProfilerLogger
26
+ from mindspore.profiler.common.exceptions.exceptions import (
27
+ ProfilerRawFileException,
28
+ )
29
+
30
+
31
+ class MindDataPipelineRawViewer(BaseViewer):
32
+ """
33
+ MindData Pipeline Raw Viewer for parsing and saving raw pipeline profiling data.
34
+ This class processes the raw pipeline information and saves it in a CSV format.
35
+ """
36
+ _FILE_NAME = 'minddata_pipeline_raw_{}.csv'
37
+ _COL_NAMES = [
38
+ 'op_id', 'op_type', 'num_workers', 'output_queue_size',
39
+ 'output_queue_average_size', 'output_queue_length',
40
+ 'output_queue_usage_rate', 'sample_interval', 'parent_id', 'children_id'
41
+ ]
42
+
43
+ def __init__(self, **kwargs):
44
+ super().__init__()
45
+ self._device_id = kwargs.get("rank_id") if (ProfilerActivity.NPU.value in
46
+ kwargs.get("activities")) else kwargs.get("device_id")
47
+ self._save_path = os.path.join(
48
+ kwargs.get("ascend_profiler_output_path"),
49
+ self._FILE_NAME.format(self._device_id)
50
+ )
51
+ self._ascend_ms_dir = kwargs.get("ascend_ms_dir")
52
+ ProfilerLogger.init(self._ascend_ms_dir)
53
+ self._logger = ProfilerLogger.get_instance()
54
+
55
+ def save(self, data: Dict[str, Any]) -> None:
56
+ if not data.get("pipeline_info"):
57
+ return
58
+ try:
59
+ dict_op_id_info, sampling_interval = data["pipeline_info"]
60
+ op_info_list = self._analyse_data(dict_op_id_info, sampling_interval)
61
+ self._logger.info("Analyse minddata pipeline raw data done")
62
+ self._save_data(op_info_list)
63
+ self._logger.info("Save minddata pipeline raw data done")
64
+ except Exception as e: # pylint: disable=W0703
65
+ self._logger.error("Failed to save minddata %s", e, exc_info=True)
66
+
67
+ def _analyse_data(self, dict_op_id_info: Dict[int, Dict[str, Any]], sample_interval: float) -> List[List[Any]]:
68
+ """
69
+ Create a list of operator information from the op id info dict.
70
+ Args:
71
+ dict_op_id_info (dict): dict of operator information indexed by op_id.
72
+ sample_interval (int): Sampling interval for the profiling data.
73
+ Returns:
74
+ list: List of operator information for each node in the pipeline.
75
+ Raises:
76
+ ProfilerRawFileException: If the root operator (id=0) does not exist.
77
+ """
78
+ root_node = dict_op_id_info.get(0)
79
+ if not root_node:
80
+ raise ProfilerRawFileException(
81
+ 'The format of minddata pipeline raw file is wrong, '
82
+ 'the operator that id is 0 does not exist.'
83
+ )
84
+ root_node['parent_id'] = None
85
+ queue = Queue()
86
+ queue.put_nowait(root_node)
87
+ op_info_list = []
88
+
89
+ while not queue.empty():
90
+ node = queue.get_nowait()
91
+ self._update_child_node(node, dict_op_id_info)
92
+ op_info_list.append(self._get_op_info(node, sample_interval))
93
+ for child_op_id in node.get('children', []):
94
+ sub_node = dict_op_id_info.get(child_op_id)
95
+ sub_node['parent_id'] = node['op_id']
96
+ queue.put_nowait(sub_node)
97
+
98
+ return op_info_list
99
+
100
+ def _save_data(self, op_info_list: List[List[Any]]) -> None:
101
+ self._logger.info("Save minddata pipeline raw data start")
102
+ FileManager.create_csv_file(self._save_path, op_info_list, self._COL_NAMES)
103
+ self._logger.info(
104
+ "Save minddata pipeline raw data done, %d rows saved, save path: %s",
105
+ len(op_info_list),
106
+ self._save_path,
107
+ )
108
+
109
+ @staticmethod
110
+ def _update_child_node(node: Dict[str, Any], dict_op_id_info: Dict[int, Dict[str, Any]]) -> None:
111
+ """
112
+ Update the children information for a given node in the pipeline.
113
+ Args:
114
+ node (dict): The current node (operator) being processed.
115
+ dict_op_id_info (dict): Dict of operator information indexed by op_id.
116
+ """
117
+ child_op_ids = node.get('children', [])
118
+ if not child_op_ids:
119
+ return
120
+
121
+ queue = Queue()
122
+ queue.queue.extend(child_op_ids)
123
+
124
+ new_child_op_ids = []
125
+ while not queue.empty():
126
+ child_op_id = queue.get_nowait()
127
+ child_node = dict_op_id_info.get(child_op_id)
128
+ if not child_node:
129
+ continue
130
+ metrics = child_node.get('metrics')
131
+ if not metrics or not metrics.get('output_queue'):
132
+ children = child_node.get('children', [])
133
+ if children:
134
+ queue.queue.extend(children)
135
+ else:
136
+ new_child_op_ids.append(child_op_id)
137
+
138
+ node['children'] = new_child_op_ids
139
+
140
+ @staticmethod
141
+ def _get_op_info(op_node: Dict[str, Any], sample_interval: float) -> List[Any]:
142
+ """
143
+ Get the operator information.
144
+ Args:
145
+ op_node (dict): The node represents an operator.
146
+ sample_interval (int): Sampling interval.
147
+ Returns:
148
+ list: A list containing the extracted operator information.
149
+ Raises:
150
+ ValueError: If the queue size is None or the queue length is 0.
151
+ """
152
+ metrics = op_node.get('metrics')
153
+ output_queue_info = metrics.get('output_queue', {}) if metrics else {}
154
+ queue_size = output_queue_info.get('size')
155
+ queue_length = output_queue_info.get('length')
156
+
157
+ if output_queue_info and queue_size is None:
158
+ raise ValueError("The queue size cannot be None.")
159
+ if output_queue_info and queue_length == 0:
160
+ raise ValueError("The length of queue cannot be 0.")
161
+
162
+ queue_average_size = sum(queue_size) / len(queue_size) if queue_size else None
163
+ queue_usage_rate = queue_average_size / queue_length if queue_average_size is not None else None
164
+
165
+ children_id = op_node.get('children')
166
+ return [
167
+ op_node.get('op_id'),
168
+ op_node.get('op_type'),
169
+ op_node.get('num_workers'),
170
+ queue_size,
171
+ queue_average_size,
172
+ queue_length,
173
+ queue_usage_rate,
174
+ sample_interval,
175
+ op_node.get('parent_id'),
176
+ children_id if children_id else None
177
+ ]
178
+
179
+
180
+ class MindDataPiplineSummaryViewer(BaseViewer):
181
+ """
182
+ MindData Pipeline Summary Viewer for processing and saving pipeline profiling summary data.
183
+ This class calculates various metrics and saves the summary in both JSON and CSV formats.
184
+ """
185
+ _FILE_NAMES = {
186
+ 'json_file': 'minddata_pipeline_summary_{}.json',
187
+ 'csv_file': 'minddata_pipeline_summary_{}.csv'
188
+ }
189
+
190
+ def __init__(self, **kwargs):
191
+ super().__init__()
192
+ self._device_id = kwargs.get("rank_id") if (ProfilerActivity.NPU.value in
193
+ kwargs.get("activities")) else kwargs.get("device_id")
194
+ self._device_queue_file_found = None
195
+ self._save_paths = {
196
+ file_type: os.path.join(kwargs.get("ascend_profiler_output_path"), file_name.format(self._device_id))
197
+ for file_type, file_name in self._FILE_NAMES.items()
198
+ }
199
+ self._ascend_ms_dir = kwargs.get("ascend_ms_dir")
200
+ ProfilerLogger.init(self._ascend_ms_dir, "MindDataPiplineSummaryViewer")
201
+ self._logger = ProfilerLogger.get_instance()
202
+
203
+ def save(self, data: Dict[str, Any]) -> None:
204
+ # If there are errors in the data during the parsing phase, the data will be set to empty
205
+ if not (data.get("pipeline_info") and data.get('cpu_util_info') and data.get("device_trace_info")):
206
+ return
207
+ try:
208
+ self._device_queue_file_found = data["device_queue_file_found"]
209
+ summary_dict = self._analyse_data(data)
210
+ self._logger.info("Analyse minddata pipeline summary data done")
211
+ self._save_data(summary_dict)
212
+ self._logger.info("Save minddata pipeline summary data done")
213
+ except Exception as e: # pylint: disable=W0703
214
+ self._logger.error("Failed to save minddata %s", e, exc_info=True)
215
+
216
+ def _analyse_data(self, data: Dict[str, Any]) -> Dict[str, Any]:
217
+ """
218
+ Calculate the summary information from the raw profiling data.
219
+ Args:
220
+ data (dict): Raw profiling data.
221
+ Returns:
222
+ dict: Calculated summary information.
223
+ """
224
+ summary_dict = {}
225
+ summary_dict.update(self._analyse_pipeline_info(data["pipeline_info"]))
226
+ summary_dict.update(self._analyse_cpu_util_info(data["cpu_util_info"]))
227
+ summary_dict.update(self._analyse_device_trace_info(data["device_trace_info"]))
228
+ self._check_and_update_summary(summary_dict)
229
+ return summary_dict
230
+
231
+ def _save_data(self, summary_dict: Dict[str, Any]) -> None:
232
+ self._logger.info("Save minddata pipeline summary data start")
233
+ FileManager.create_json_file(self._save_paths['json_file'], summary_dict)
234
+ FileManager.create_csv_file(
235
+ self._save_paths['csv_file'],
236
+ [[key] + value for key, value in summary_dict.items()]
237
+ )
238
+ self._logger.info("Save minddata pipeline summary data done")
239
+
240
+ def _check_and_update_summary(self, summary_dict: Dict[str, Any]) -> None:
241
+ """
242
+ Check for consistency in the summary data and update with composite information.
243
+ Args:
244
+ summary_dict (dict): Summary dictionary to be checked and updated.
245
+ """
246
+ num_pipeline_ops = len(summary_dict.get('pipeline_ops', []))
247
+ num_cpu_util_ops = len(summary_dict.get('avg_cpu_pct', []))
248
+ # Check if both pipeline data and CPU utilization data have the same number of ops
249
+ if num_pipeline_ops == num_cpu_util_ops:
250
+ summary_dict.update(self._analyse_composite_info(summary_dict))
251
+ bottleneck_dict = self._analyse_bottleneck_op(summary_dict)
252
+ if bottleneck_dict:
253
+ summary_dict.update(bottleneck_dict)
254
+ else:
255
+ logger.warning(f'Number of ops mismatch: pipeline data ({num_pipeline_ops}) '
256
+ f'vs CPU utilization data ({num_cpu_util_ops})')
257
+
258
+ def _analyse_pipeline_info(self, pipeline_info):
259
+ """
260
+ Calculate pipeline information from raw pipeline data.
261
+ Args:
262
+ pipeline_info (dict): Raw pipeline information.
263
+ Returns:
264
+ dict: Processed pipeline information.
265
+ Raises:
266
+ ProfilerRawFileException: If the format of the input is wrong.
267
+ """
268
+ # Since there may be non-linear pipelines, the processed op info needs to be sorted before final output is
269
+ # produced and saved.
270
+ op_id_info_list = sorted(pipeline_info[0].items(), key=lambda item: item[0])
271
+ return_dict = defaultdict(list)
272
+ dict_opid_parent_id = {}
273
+
274
+ for op_id, op_info in op_id_info_list:
275
+ op_name = op_info.get('op_type')[0:-2]
276
+ queue_info_items = self._get_pipeline_metrics_info(op_info.get('metrics', {}))
277
+ children_ids = op_info.get('children', [])
278
+ for child_op_id in children_ids:
279
+ dict_opid_parent_id[child_op_id] = op_id
280
+
281
+ return_dict['op_ids'].append(op_id)
282
+ return_dict['op_names'].append(op_name)
283
+ return_dict['pipeline_ops'].append(f'{op_name}(id={op_id})')
284
+ return_dict['num_workers'].append(op_info.get('num_workers'))
285
+ return_dict['queue_average_size'].append(queue_info_items[2]) # output queue average size
286
+ return_dict['queue_utilization_pct'].append(queue_info_items[3]) # output queue utilization percentage
287
+ return_dict['queue_empty_freq_pct'].append(queue_info_items[4]) # output queue empty frequency percentage
288
+ return_dict['children_ids'].append(children_ids)
289
+
290
+ return_dict['parent_id'] = [dict_opid_parent_id.get(idx, -1) for idx, _ in op_id_info_list]
291
+ return return_dict
292
+
293
+ def _analyse_device_trace_info(self, device_trace_info):
294
+ """
295
+ Calculate device trace information.
296
+ Args:
297
+ device_trace_info (list): Raw device trace information.
298
+ Returns:
299
+ Dictionary consists of:
300
+ per_batch_time: Average per batch time for pipeline in milliseconds
301
+ per_pipeline_time: Average per pipeline time in milliseconds
302
+ per_push_queue_time: Average per queue push time in milliseconds
303
+ """
304
+ q_time = [[], [], []] # pipeline time, push TDT time, batch time
305
+ prev_time = 0
306
+
307
+ for line_data in device_trace_info:
308
+ if len(line_data.split(" ")) < 5:
309
+ logger.warning("Invalid device trace data: %s", line_data)
310
+ continue
311
+ record = [int(d) for d in line_data.split(" ")][0:5]
312
+ if record[2] < 2: # skip 1st batch
313
+ prev_time = record[4]
314
+ continue
315
+ if record[0] == 0: # type 0: time record
316
+ q_time[record[1]].append(record[3])
317
+ elif record[0] == 1 and not self._device_queue_file_found: # type 1: connector size record
318
+ q_time[2].append(record[4] - prev_time)
319
+ prev_time = record[4]
320
+
321
+ avg_times = [sum(t) / len(t) if t else -1 for t in q_time]
322
+ return {
323
+ 'per_pipeline_time': [round(avg_times[0], 3)],
324
+ 'per_push_queue_time': [round(avg_times[1], 3)],
325
+ 'per_batch_time': [round(avg_times[2], 3)]
326
+ }
327
+
328
+ @staticmethod
329
+ def _get_pipeline_metrics_info(metrics: Dict[str, Any]) -> List[Any]:
330
+ """
331
+ Parse and process the pipeline profiling metrics information for a given op.
332
+ Args:
333
+ metrics (dict): The pipeline profiling metrics information for a given op.
334
+ Returns:
335
+ list: A list containing output queue size, length, average size, utilization percentage,
336
+ and empty frequency percentage.
337
+ """
338
+ queue_metrics = metrics.get('output_queue', {}) if metrics else {}
339
+ queue_size = queue_metrics.get('size', [])
340
+ queue_length = queue_metrics.get('length', 0)
341
+ if not queue_size:
342
+ return [-1] * 5
343
+ queue_average_size = round(sum(queue_size) / len(queue_size), 2)
344
+ queue_utilization_pct = round(100 * queue_average_size / queue_length, 2) if queue_length else -1
345
+ # Compute percentage of time queue is empty
346
+ queue_empty_freq_pct = round(100 * queue_size.count(0) / len(queue_size), 2)
347
+
348
+ return [queue_size, queue_length, queue_average_size, queue_utilization_pct, queue_empty_freq_pct]
349
+
350
+ @staticmethod
351
+ def _analyse_cpu_util_info(cpu_util_info: Dict[str, Any]) -> Dict[str, List[float]]:
352
+ """
353
+ Calculate CPU utilization information.
354
+ Args:
355
+ cpu_util_info (dict): The CPU utilization profiling information.
356
+ Returns:
357
+ dict: Average CPU utilization percentage for each op, a list ordered by increasing op id
358
+ Raises:
359
+ ProfilerRawFileException: If the format of the CPU utilization file is incorrect.
360
+ """
361
+ cpu_processor_num = cpu_util_info.get('cpu_processor_num')
362
+ cpu_op_info = cpu_util_info.get('op_info')
363
+ if cpu_processor_num is None or not cpu_op_info:
364
+ raise ProfilerRawFileException('The format of MindData CPU utilization JSON file is wrong.')
365
+ for item in cpu_op_info:
366
+ if not item:
367
+ raise ProfilerRawFileException('The contents of MindData CPU utilization JSON file is wrong.')
368
+ # Note: The CPU utilization data may have an extra entry with op_id=-1
369
+ # Omit info for op_id=1
370
+ dict_opid_cpuutil = {
371
+ op["op_id"]: [op_sys + op_usr for op_sys, op_usr in
372
+ zip(op["metrics"]["sys_utilization"], op["metrics"]["user_utilization"])]
373
+ for op in cpu_op_info if op and op["op_id"] != -1
374
+ }
375
+ # Initialize oplist_avg_cpu_pct with -1 for each pipeline op, since
376
+ # CPU utilization data may not have information for each pipeline op
377
+ oplist_avg_cpu_pct = [-1] * len(dict_opid_cpuutil)
378
+ for op_id, cpu in dict_opid_cpuutil.items():
379
+ op_avg_cpu_pct = sum(cpu) / len(cpu) if cpu else 0
380
+ oplist_avg_cpu_pct[op_id] = round(op_avg_cpu_pct, 2)
381
+
382
+ return {'avg_cpu_pct': oplist_avg_cpu_pct}
383
+
384
+ @staticmethod
385
+ def _analyse_composite_info(summary_dict: Dict[str, Any]) -> Dict[str, List[float]]:
386
+ """
387
+ Compute composite analysis information from the current summary pipeline data.
388
+ Args:
389
+ summary_dict (dict): Input summary pipeline information.
390
+ Returns:
391
+ dict: Average CPU utilization percentage per worker
392
+ """
393
+ # Build list: average CPU utilization percentage per worker - for each op
394
+ avg_cpu_pct_per_worker = [
395
+ round(c / n if (n != 0 and c >= 0) else -1, 2)
396
+ for c, n in zip(summary_dict.get('avg_cpu_pct', []), summary_dict.get('num_workers', []))
397
+ ]
398
+ return {'avg_cpu_pct_per_worker': avg_cpu_pct_per_worker}
399
+
400
+ @staticmethod
401
+ def _analyse_bottleneck_op(summary_dict: Dict[str, Any]) -> Dict[str, Any]:
402
+ """
403
+ Calculate the bottleneck operation using the BottleneckCalculator.
404
+ Args:
405
+ summary_dict (dict): Summary dictionary containing pipeline information.
406
+ Returns:
407
+ Dictionary with the following information, if applicable:
408
+ - CPU utilization analysis
409
+ - queue utilization analysis
410
+ - bottleneck warning: Information on the bottleneck op
411
+ (This is returned only if a potential bottleneck is identified.)
412
+ - bottleneck suggestion: Reason why the subject op is it is identified as
413
+ a potential bottleneck, plus suggestion on how to resolve the bottleneck.
414
+ (This is returned only if a potential bottleneck is identified.)
415
+ """
416
+ try:
417
+ bottleneck_analyzer = BottleneckAnalyzer(summary_dict)
418
+ return bottleneck_analyzer.analyze()
419
+ except IndexError:
420
+ return {}
421
+
422
+
423
+ class BottleneckAnalyzer:
424
+ """ analyzer for bottleneck """
425
+
426
+ # These are the threshold values used in the pipeline bottleneck analyzer algorithm
427
+ _THRESHOLDS = {
428
+ '_AVG_CPU_UTIL_PCT_PER_WORKER_MAXIMUM': 75.0,
429
+ '_AVG_CPU_UTIL_PCT_PER_WORKER_MINIMUM': 20.0,
430
+ '_LEAF_OUTPUT_QUEUE_EMPTY_FREQ_PCT_MAXIMUM': 50,
431
+ '_DEVICEQUEUE_INPUT_QUEUE_EMPTY_FREQ_PCT_MAXIMUM': 60,
432
+ '_IN_OUT_QUEUE_UTIL_PCT_DIFF_MAXIMUM': 50,
433
+ '_IN_QUEUE_UTIL_PCT_MAXIMUM': 10
434
+ }
435
+
436
+ _NON_MULTITHREADED_OPS = {
437
+ "Barrier", "Concat", "EpochCtrl", "Rename", "Repeat",
438
+ "Shuffle", "Skip", "Take", "Zip"
439
+ }
440
+
441
+ def __init__(self, summary_dict):
442
+ self.pipeline_ops = summary_dict["pipeline_ops"]
443
+ self.op_names = summary_dict["op_names"]
444
+ self.op_ids = summary_dict["op_ids"]
445
+ self.num_workers = summary_dict["num_workers"]
446
+ self.queue_average_size = summary_dict["queue_average_size"]
447
+ self.queue_utilization_pct = summary_dict["queue_utilization_pct"]
448
+ self.queue_empty_freq_pct = summary_dict["queue_empty_freq_pct"]
449
+ self.children_ids = summary_dict["children_ids"]
450
+ self.parent_id = summary_dict["parent_id"]
451
+ self.avg_cpu_pct = summary_dict["avg_cpu_pct"]
452
+ self.avg_cpu_pct_per_worker = summary_dict["avg_cpu_pct_per_worker"]
453
+ self.op_id_not_exist = -1
454
+ self.queue_usage_not_exist = -1
455
+
456
+ def analyze(self):
457
+ """ analyze all op's usage """
458
+ detailed_analysis = {}
459
+ cpu_analysis = self._analyze_cpu_usage()
460
+ queue_analysis = self._analyze_queue_usage()
461
+
462
+ if cpu_analysis:
463
+ detailed_analysis["cpu_analysis_details"] = cpu_analysis
464
+ if queue_analysis:
465
+ detailed_analysis["queue_analysis_details"] = queue_analysis
466
+
467
+ bottleneck, suggestion = self._analyze_bottleneck()
468
+ if bottleneck[0]:
469
+ detailed_analysis["bottleneck_warning"] = bottleneck
470
+ detailed_analysis["bottleneck_suggestion"] = suggestion
471
+
472
+ return detailed_analysis
473
+
474
+ def _analyze_cpu_usage(self):
475
+ """ analyze cpu usage of each op """
476
+ cpu_usage_analysis = []
477
+ for op_id in self.op_ids:
478
+ if op_id == self.op_id_not_exist or self.op_names[op_id] in self._NON_MULTITHREADED_OPS:
479
+ continue
480
+
481
+ cpu_pct = self.avg_cpu_pct_per_worker[op_id]
482
+ if cpu_pct > self._THRESHOLDS['_AVG_CPU_UTIL_PCT_PER_WORKER_MAXIMUM'] and self.op_names[op_id]:
483
+ cpu_usage_analysis.append(self._format_high_cpu_usage_suggestion(op_id, cpu_pct))
484
+ elif cpu_pct < self._THRESHOLDS['_AVG_CPU_UTIL_PCT_PER_WORKER_MINIMUM'] and self.num_workers[op_id] > 1:
485
+ cpu_usage_analysis.append(self._format_low_cpu_usage_suggestion(op_id, cpu_pct))
486
+ return cpu_usage_analysis
487
+
488
+ def _analyze_queue_usage(self):
489
+ """ analyze queue usage of each op """
490
+ queue_usage_analysis = []
491
+ for op_id in self.op_ids:
492
+ if op_id == self.op_id_not_exist or self.op_names[op_id] in self._NON_MULTITHREADED_OPS:
493
+ continue
494
+ if self.op_names[op_id] == "Batch":
495
+ continue
496
+ in_op_id, out_q = self._get_non_inline_child_recur(op_id), self.queue_utilization_pct[op_id]
497
+ # This is a leaf node since input queue does not exist and output queue exists
498
+ if in_op_id == self.op_id_not_exist and out_q != self.queue_usage_not_exist:
499
+ if out_q < self._THRESHOLDS['_LEAF_OUTPUT_QUEUE_EMPTY_FREQ_PCT_MAXIMUM']:
500
+ queue_usage_analysis.append(self._format_leaf_node_suggestion(op_id, out_q))
501
+ # This is device_queue op
502
+ elif self.op_names[op_id] == "DeviceQueue" and in_op_id != self.op_id_not_exist:
503
+ if (self.queue_empty_freq_pct[in_op_id] >
504
+ self._THRESHOLDS['_DEVICEQUEUE_INPUT_QUEUE_EMPTY_FREQ_PCT_MAXIMUM']):
505
+ queue_usage_analysis.append(self._format_device_queue_suggestion(op_id, in_op_id))
506
+ elif in_op_id != self.op_id_not_exist and out_q != self.queue_usage_not_exist:
507
+ in_q = self.queue_utilization_pct[in_op_id]
508
+ if (in_q != self.queue_usage_not_exist and
509
+ in_q - out_q > self._THRESHOLDS['_IN_OUT_QUEUE_UTIL_PCT_DIFF_MAXIMUM']):
510
+ queue_usage_analysis.append(self._format_internal_node_suggestion(op_id, in_op_id, out_q, in_q))
511
+ return queue_usage_analysis
512
+
513
+ def _analyze_bottleneck(self):
514
+ """ analyze bottleneck by using both cpu and queue usage """
515
+ bottleneck, suggestion = "", ""
516
+ for op_id in reversed(self.op_ids):
517
+ if self._should_skip_op(op_id):
518
+ continue
519
+
520
+ in_op_id, out_q = self._get_non_inline_child_recur(op_id), self.queue_utilization_pct[op_id]
521
+ wkr_cpu = self.avg_cpu_pct_per_worker[op_id]
522
+
523
+ if wkr_cpu > self._THRESHOLDS['_AVG_CPU_UTIL_PCT_PER_WORKER_MAXIMUM']:
524
+ bottleneck = self.pipeline_ops[op_id]
525
+ suggestion = self._format_high_cpu_usage_suggestion(op_id, wkr_cpu)
526
+
527
+ elif wkr_cpu < self._THRESHOLDS['_AVG_CPU_UTIL_PCT_PER_WORKER_MINIMUM']:
528
+ in_q_usage = self.queue_utilization_pct[in_op_id]
529
+ if in_op_id != self.op_id_not_exist and (in_q_usage < self._THRESHOLDS['_IN_QUEUE_UTIL_PCT_MAXIMUM']
530
+ or out_q - in_q_usage > self._THRESHOLDS[
531
+ '_IN_OUT_QUEUE_UTIL_PCT_DIFF_MAXIMUM']):
532
+ bottleneck = self.pipeline_ops[op_id]
533
+ suggestion = self._format_queue_bottleneck_suggestion(op_id)
534
+
535
+ return [bottleneck], [suggestion]
536
+
537
+ def _should_skip_op(self, op_id):
538
+ return (op_id == -1 or
539
+ self.op_names[op_id] in self._NON_MULTITHREADED_OPS or
540
+ self.op_names[op_id] == "DeviceQueue")
541
+
542
+ def _get_non_inline_child_recur(self, cur_op_id):
543
+ """ get the child id of cur op which isn't an inline op """
544
+ if cur_op_id == self.op_id_not_exist or not self.children_ids[cur_op_id]:
545
+ return self.op_id_not_exist
546
+ cur_child_id = self.children_ids[cur_op_id][0]
547
+ if self.queue_average_size[cur_child_id] != -1:
548
+ return cur_child_id
549
+ return self._get_non_inline_child_recur(cur_child_id)
550
+
551
+ def _format_high_cpu_usage_suggestion(self, op_id, cpu_pct):
552
+ return (f"{self.pipeline_ops[op_id]} is using {cpu_pct}% CPU per worker. "
553
+ f"Setting num_parallel_workers>{self.num_workers[op_id]} might bring extra performance.")
554
+
555
+ def _format_low_cpu_usage_suggestion(self, op_id, cpu_pct):
556
+ return (f"{self.pipeline_ops[op_id]} is using {cpu_pct}% CPU per worker. "
557
+ f"Using num_parallel_workers={self.num_workers[op_id]} might not bring as much benefit "
558
+ f"due to low CPU usage per worker.")
559
+
560
+ def _format_leaf_node_suggestion(self, op_id, out_q):
561
+ return (f"Leaf op {self.pipeline_ops[op_id]} is using {out_q}% of its output queue. "
562
+ f"Setting num_parallel_workers>{self.num_workers[op_id]} might speed up I/O.")
563
+
564
+ def _format_device_queue_suggestion(self, op_id, in_op_id):
565
+ return (f"{self.pipeline_ops[op_id]}'s input queue is empty "
566
+ f"{self.queue_empty_freq_pct[in_op_id]}% of the time. "
567
+ f"This might indicate dataset bottlenecks. Hence host cannot keep up with "
568
+ f"the device {self.queue_empty_freq_pct[in_op_id]}% of the time. "
569
+ f"Device waits whenever input queue is empty.")
570
+
571
+ def _format_internal_node_suggestion(self, op_id, in_op_id, out_q, in_q):
572
+ return (f"{self.pipeline_ops[op_id]}'s input queue usage={in_q}% is greater than output queue "
573
+ f"usage={out_q}%. This indicates child op {self.pipeline_ops[in_op_id]} "
574
+ f"might be producing faster than its parent {self.pipeline_ops[op_id]} can consume. "
575
+ f"If this op has low CPU utilization, try increasing "
576
+ f"prefetch_size or increasing num_workers.")
577
+
578
+ def _format_queue_bottleneck_suggestion(self, op_id):
579
+ return (f"{self.pipeline_ops[op_id]} has low CPU utilization per worker of "
580
+ f"{self.avg_cpu_pct_per_worker[op_id]}% and abnormal queue usage. "
581
+ f"Try increasing prefetch_size.")