mindstudio-probe 8.3.0__tar.gz → 26.0.0a1__tar.gz

mindstudio_probe-26.0.0a1/LICENSE

@@ -0,0 +1,124 @@
+木兰宽松许可证， 第2版
+
+2020年1月 http://license.coscl.org.cn/MulanPSL2
+
+您对“软件”的复制、使用、修改及分发受木兰宽松许可证，第2版（“本许可证”）的如下条款的约束：
+
+0.   定义
+
+“软件” 是指由“贡献”构成的许可在“本许可证”下的程序和相关文档的集合。
+
+“贡献” 是指由任一“贡献者”许可在“本许可证”下的受版权法保护的作品。
+
+“贡献者” 是指将受版权法保护的作品许可在“本许可证”下的自然人或“法人实体”。
+
+“法人实体” 是指提交贡献的机构及其“关联实体”。
+
+“关联实体” 是指，对“本许可证”下的行为方而言，控制、受控制或与其共同受控制的机构，此处的控制是指有受控方或共同受控方至少50%直接或间接的投票权、资金或其他有价证券。
+
+1.   授予版权许可
+
+每个“贡献者”根据“本许可证”授予您永久性的、全球性的、免费的、非独占的、不可撤销的版权许可，您可以复制、使用、修改、分发其“贡献”，不论修改与否。
+
+2.   授予专利许可
+
+每个“贡献者”根据“本许可证”授予您永久性的、全球性的、免费的、非独占的、不可撤销的（根据本条规定撤销除外）专利许可，供您制造、委托制造、使用、许诺销售、销售、进口其“贡献”或以其他方式转移其“贡献”。前述专利许可仅限于“贡献者”现在或将来拥有或控制的其“贡献”本身或其“贡献”与许可“贡献”时的“软件”结合而将必然会侵犯的专利权利要求，不包括对“贡献”的修改或包含“贡献”的其他结合。如果您或您的“关联实体”直接或间接地，就“软件”或其中的“贡献”对任何人发起专利侵权诉讼（包括反诉或交叉诉讼）或其他专利维权行动，指控其侵犯专利权，则“本许可证”授予您对“软件”的专利许可自您提起诉讼或发起维权行动之日终止。
+
+3.   无商标许可
+
+“本许可证”不提供对“贡献者”的商品名称、商标、服务标志或产品名称的商标许可，但您为满足第4条规定的声明义务而必须使用除外。
+
+4.   分发限制
+
+您可以在任何媒介中将“软件”以源程序形式或可执行形式重新分发，不论修改与否，但您必须向接收者提供“本许可证”的副本，并保留“软件”中的版权、商标、专利及免责声明。
+
+5.   免责声明与责任限制
+
+“软件”及其中的“贡献”在提供时不带任何明示或默示的担保。在任何情况下，“贡献者”或版权所有者不对任何人因使用“软件”或其中的“贡献”而引发的任何直接或间接损失承担责任，不论因何种原因导致或者基于何种法律理论，即使其曾被建议有此种损失的可能性。
+
+6.   语言
+
+“本许可证”以中英文双语表述，中英文版本具有同等法律效力。如果中英文版本存在任何冲突不一致，以中文版为准。
+
+条款结束
+
+如何将木兰宽松许可证，第2版，应用到您的软件
+
+如果您希望将木兰宽松许可证，第2版，应用到您的新软件，为了方便接收者查阅，建议您完成如下三步：
+
+1， 请您补充如下声明中的空白，包括软件名、软件的首次发表年份以及您作为版权人的名字；
+
+2， 请您在软件包的一级目录下创建以“LICENSE”为名的文件，将整个许可证文本放入该文件中；
+
+3， 请将如下声明文本放入每个源文件的头部注释中。
+
+Copyright (c) [Year] [name of copyright holder]
+[Software Name] is licensed under Mulan PSL v2.
+You can use this software according to the terms and conditions of the Mulan PSL v2.
+You may obtain a copy of Mulan PSL v2 at:
+         http://license.coscl.org.cn/MulanPSL2
+THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+See the Mulan PSL v2 for more details.
+Mulan Permissive Software License，Version 2
+Mulan Permissive Software License，Version 2 (Mulan PSL v2)
+
+January 2020 http://license.coscl.org.cn/MulanPSL2
+
+Your reproduction, use, modification and distribution of the Software shall be subject to Mulan PSL v2 (this License) with the following terms and conditions:
+
+0. Definition
+
+Software means the program and related documents which are licensed under this License and comprise all Contribution(s).
+
+Contribution means the copyrightable work licensed by a particular Contributor under this License.
+
+Contributor means the Individual or Legal Entity who licenses its copyrightable work under this License.
+
+Legal Entity means the entity making a Contribution and all its Affiliates.
+
+Affiliates means entities that control, are controlled by, or are under common control with the acting entity under this License, ‘control’ means direct or indirect ownership of at least fifty percent (50%) of the voting power, capital or other securities of controlled or commonly controlled entity.
+
+1. Grant of Copyright License
+
+Subject to the terms and conditions of this License, each Contributor hereby grants to you a perpetual, worldwide, royalty-free, non-exclusive, irrevocable copyright license to reproduce, use, modify, or distribute its Contribution, with modification or not.
+
+2. Grant of Patent License
+
+Subject to the terms and conditions of this License, each Contributor hereby grants to you a perpetual, worldwide, royalty-free, non-exclusive, irrevocable (except for revocation under this Section) patent license to make, have made, use, offer for sale, sell, import or otherwise transfer its Contribution, where such patent license is only limited to the patent claims owned or controlled by such Contributor now or in future which will be necessarily infringed by its Contribution alone, or by combination of the Contribution with the Software to which the Contribution was contributed. The patent license shall not apply to any modification of the Contribution, and any other combination which includes the Contribution. If you or your Affiliates directly or indirectly institute patent litigation (including a cross claim or counterclaim in a litigation) or other patent enforcement activities against any individual or entity by alleging that the Software or any Contribution in it infringes patents, then any patent license granted to you under this License for the Software shall terminate as of the date such litigation or activity is filed or taken.
+
+3. No Trademark License
+
+No trademark license is granted to use the trade names, trademarks, service marks, or product names of Contributor, except as required to fulfill notice requirements in section 4.
+
+4. Distribution Restriction
+
+You may distribute the Software in any medium with or without modification, whether in source or executable forms, provided that you provide recipients with a copy of this License and retain copyright, patent, trademark and disclaimer statements in the Software.
+
+5. Disclaimer of Warranty and Limitation of Liability
+
+THE SOFTWARE AND CONTRIBUTION IN IT ARE PROVIDED WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED. IN NO EVENT SHALL ANY CONTRIBUTOR OR COPYRIGHT HOLDER BE LIABLE TO YOU FOR ANY DAMAGES, INCLUDING, BUT NOT LIMITED TO ANY DIRECT, OR INDIRECT, SPECIAL OR CONSEQUENTIAL DAMAGES ARISING FROM YOUR USE OR INABILITY TO USE THE SOFTWARE OR THE CONTRIBUTION IN IT, NO MATTER HOW IT’S CAUSED OR BASED ON WHICH LEGAL THEORY, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
+
+6. Language
+
+THIS LICENSE IS WRITTEN IN BOTH CHINESE AND ENGLISH, AND THE CHINESE VERSION AND ENGLISH VERSION SHALL HAVE THE SAME LEGAL EFFECT. IN THE CASE OF DIVERGENCE BETWEEN THE CHINESE AND ENGLISH VERSIONS, THE CHINESE VERSION SHALL PREVAIL.
+
+END OF THE TERMS AND CONDITIONS
+
+How to Apply the Mulan Permissive Software License，Version 2 (Mulan PSL v2) to Your Software
+
+To apply the Mulan PSL v2 to your work, for easy identification by recipients, you are suggested to complete following three steps:
+
+Fill in the blanks in following statement, including insert your software name, the year of the first publication of your software, and your name identified as the copyright owner;
+Create a file named "LICENSE" which contains the whole context of this License in the first directory of your software package;
+Attach the statement to the appropriate annotated syntax at the beginning of each source file.
+Copyright (c) [Year] [name of copyright holder]
+[Software Name] is licensed under Mulan PSL v2.
+You can use this software according to the terms and conditions of the Mulan PSL v2.
+You may obtain a copy of Mulan PSL v2 at:
+         http://license.coscl.org.cn/MulanPSL2
+THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+See the Mulan PSL v2 for more details.

mindstudio_probe-26.0.0a1/MANIFEST.in

@@ -0,0 +1,6 @@
+include README.md
+include LICENSE
+recursive-include python/msprobe *
+recursive-exclude test *
+recursive-exclude ccsrc *
+

mindstudio_probe-26.0.0a1/PKG-INFO

@@ -0,0 +1,51 @@
+Metadata-Version: 2.4
+Name: mindstudio-probe
+Version: 26.0.0a1
+Summary: Ascend MindStudio Probe Utils
+Home-page: https://gitcode.com/Ascend/MindStudio-Probe
+Author: Ascend Team
+Author-email: pmail_mindstudio@huawei.com
+License: Mulan PSL v2
+Keywords: pytorch msprobe ascend
+Platform: Linux
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: C++
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.7
+License-File: LICENSE
+Requires-Dist: wheel
+Requires-Dist: einops
+Requires-Dist: numpy<2.0,>=1.23.0
+Requires-Dist: pandas<2.1,>=1.3.5
+Requires-Dist: pyyaml
+Requires-Dist: tqdm
+Requires-Dist: openpyxl>=3.0.6
+Requires-Dist: matplotlib
+Requires-Dist: tensorboard>=2.11.2
+Requires-Dist: protobuf<=3.20.2
+Requires-Dist: rich
+Requires-Dist: onnx>=1.14.0
+Requires-Dist: onnxruntime!=1.16.0,>=1.14.1
+Requires-Dist: skl2onnx>=1.14.1
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: platform
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+MindStudio-Probe is a set of tools for diagnosing and improving model accuracy on Ascend NPU.

mindstudio_probe-26.0.0a1/README.md

@@ -0,0 +1,231 @@
+# MindStudio Probe
+
+## 简介
+
+MindStudio Probe（MindStudio精度调试工具，msProbe）是针对昇腾提供的全场景精度工具链，专为模型开发的精度调试环节设计，可显著提升用户定位模型精度问题的效率。
+
+msProbe主要包括精度数据采集（dump）、精度预检、训练状态监测和精度比对等功能，这些功能侧重不同的训练或推理场景，可以帮助定位模型训练或推理中的精度问题。
+
+## 环境部署
+
+### 环境和依赖
+
+使用msProbe工具前，要求已存在可执行的用户AI应用，其中要求昇腾环境：
+
+- 可正常运行用户AI应用，详细设备型号请参见《[昇腾产品形态说明](https://www.hiascend.com/document/detail/zh/AscendFAQ/ProduTech/productform/hardwaredesc_0001.html)》。
+- 已安装配套版本的CANN Toolkit开发套件包或算子包并配置环境变量，详情请参见《CANN 软件安装指南》中”[选择安装场景](https://www.hiascend.com/document/detail/zh/canncommercial/83RC1/softwareinst/instg/instg_0000.html?Mode=PmIns&InstallType=local&OS=openEuler&Software=cannToolKit)“章节的”训练&推理&开发调试“场景。
+
+### 工具安装
+
+安装msProbe工具，当前仅支持编译安装方式，具体请参见《[msProbe工具安装指南](docs/zh/msprobe_install_guide.md)》。
+
+## 快速入门
+
+msProbe工具快速入门当前提供在PyTorch和MindSpore训练场景中，通过一个可执行样例，串联msProbe工具的训练前配置检查、精度数据采集、精度预检、训练状态监测及精度比对功能，帮助用户快速上手。详细快速入门可参见《训练场景工具快速入门》中的“[模型精度调试](https://www.hiascend.com/document/detail/zh/mindstudio/82RC1/msquickstart/atlasquick_train_0023.html?framework=pytorch)（PyTorch场景）”或“[模型精度调试](https://www.hiascend.com/document/detail/zh/mindstudio/82RC1/msquickstart/atlasquick_train_0006.html?framework=mindspore)（MindSpore场景）。
+
+## 工具限制与注意事项
+
+- 工具读写的所有路径，如`config_path`、`dump_path`等，只允许包含大小写字母、数字、下划线、斜杠、点和短横线。
+
+- 出于安全性及权限最小化角度考虑，本工具不应使用root等高权限账户，建议使用普通用户权限安装执行。
+
+- 使用本工具前请确保执行用户的umask值大于等于0027，否则可能会导致工具生成的精度数据文件和目录权限过大。
+
+- 用户须自行保证使用最小权限原则，如给工具输入的文件要求other用户不可写，在一些对安全要求更严格的功能场景下还需确保输入的文件group用户不可写。
+
+- 使用工具前，建议先浏览[工具功能模块简介、适用场景和当前版本局限性](docs/zh/limitations_and_precautions.md)，了解功能特性。
+
+- msProbe建议执行用户与安装用户保持一致，如果使用root执行，请自行关注root高权限触及的安全风险。
+
+
+## 功能介绍
+
+### vLLM推理场景
+
+#### aclgraph图模式
+1. [数据采集](docs/zh/dump/aclgraph_dump_instruct.md)
+
+   通过acl_save接口完成精度数据采集操作。
+
+#### torchair图模式
+
+1. [数据采集](docs/zh/dump/torchair_dump_instruct.md)
+
+   通过set_ge_dump_config接口完成精度数据采集操作。
+
+2. [精度比对](docs/zh/accuracy_compare/torchair_compare_instruct.md)
+
+   将msProbe工具dump的精度数据进行精度比对，进而定位精度问题。
+
+
+### ATB推理场景
+
+1. [数据采集](docs/zh/dump/atb_data_dump_instruct.md)
+
+   通过在ATB模型运行前，加载ATB dump模块的方式，实现对ATB模型运行过程中的精度数据的采集。
+
+2. [精度比对](docs/zh/accuracy_compare/atb_data_compare_instruct.md)
+
+   将ATB dump的精度数据进行精度比对，进而定位精度问题。
+
+
+### 离线模型推理场景
+
+1. [数据采集](docs/zh/dump/infer_offline_dump_instruct.md)
+
+   完成msProbe精度数据采集操作。
+
+2. [精度比对](docs/zh/accuracy_compare/infer_compare_offline_model_instruct.md)
+
+   提供一键式离线模型比对功能，仅需输入模型即可完成比对，无需提前采集数据，快速输出结果。
+
+3. [离线模型数据精度比对](docs/zh/accuracy_compare/offlline_data_compare_instruct.md)
+   提供离线模型数据比对功能，输入离线模型的dump数据进行精度比对。
+
+
+### PyTorch训练场景
+
+1. [训练前配置检查](docs/zh/config_check_instruct.md)
+
+   训练前或精度比对前，对比两个环境下可能影响训练精度的配置差异。
+
+2. [数据采集](docs/zh/dump/pytorch_data_dump_instruct.md)
+
+   通过config.json配置，完成msProbe精度数据采集操作。
+
+   config.json配置文件详细介绍请参见[配置文件介绍](docs/zh/dump/config_json_introduct.md)和[config.json配置样例](docs/zh/dump/config_json_examples.md)。
+
+3. [精度预检](docs/zh/accuracy_checker/pytorch_accuracy_checker_instruct.md)
+
+   在昇腾NPU上扫描训练模型中的所有API，给出精度情况的诊断和分析。
+
+4. [分级可视化构图比对](docs/zh/accuracy_compare/pytorch_visualization_instruct.md)
+
+   将msProbe工具dump的精度数据进行解析，还原模型图结构，实现模型各个层级的精度数据比对，方便用户理解模型结构、分析精度问题。
+
+5. [精度比对](docs/zh/accuracy_compare/pytorch_accuracy_compare_instruct.md)
+
+   将msProbe工具dump的精度数据进行精度比对，进而定位精度问题。
+
+6. [训练状态监测](docs/zh/monitor_instruct.md)
+
+   收集和聚合模型训练过程中的网络层，优化器，通信算子的中间值，帮助诊断模型训练过程中计算，通信，优化器各部分出现的异常情况。
+
+7. [checkpoint比对](docs/zh/checkpoint_compare_instruct.md)
+
+   训练过程中或结束后，比较两个不同的checkpoint，评估模型相似度。
+
+8. [整网首个溢出节点分析](docs/zh/overflow_check/overflow_check_instruct.md)
+
+   多rank场景下通过dump数据找到首个出现Nan或Inf的节点。
+
+9. [趋势可视化](docs/zh/accuracy_compare/trend_visualization_instruct.md)
+
+   将msProbe工具数据采集或训练状态监测的统计量数据从迭代步数、节点rank和张量目标三个维度进行趋势可视化，方便用户从整体的趋势分布观测精度数据。
+
+
+### MindSpore训练场景
+
+1. [训练前配置检查](docs/zh/config_check_instruct.md)
+
+   训练前或精度比对前，对比两个环境下可能影响训练精度的配置差异。
+
+2. [数据采集](docs/zh/dump/mindspore_data_dump_instruct.md)
+
+   通过config.json配置，完成msProbe精度数据采集操作。
+
+   config.json配置文件详细介绍请参见[配置文件介绍](docs/zh/dump/config_json_introduct.md)和[config.json配置样例](docs/zh/dump/config_json_examples.md)。
+
+3. [精度预检](docs/zh/accuracy_checker/mindspore_accuracy_checker_instruct.md)
+
+   在昇腾NPU上扫描训练模型中的所有API，给出精度情况的诊断和分析。
+
+4. [分级可视化构图比对](docs/zh/accuracy_compare/mindspore_visualization_instruct.md)
+
+   将msProbe工具dump的精度数据进行解析，还原模型图结构，实现模型各个层级的精度数据比对，方便用户理解模型结构、分析精度问题。
+
+5. [精度比对](docs/zh/accuracy_compare/mindspore_accuracy_compare_instruct.md)
+
+   将msProbe工具dump的精度数据进行精度比对，进而定位精度问题。
+
+6. [训练状态监测](docs/zh/monitor_instruct.md)
+
+   收集和聚合模型训练过程中的网络层，优化器，通信算子的中间值，帮助诊断模型训练过程中计算，通信，优化器各部分出现的异常情况。
+
+7. [溢出检测与解析](docs/zh/overflow_check/mindspore_overflow_check_instruct.md)
+
+   溢出检测用于采集溢出API或模块的精度数据，而溢出解析则是通过对溢出数据的分析，进一步判断是否为正常溢出。
+
+   推荐直接使用[数据采集](#数据采集-1)功能采集统计量信息，检测溢出问题，具体请参见。
+
+8. [checkpoint比对](docs/zh/checkpoint_compare_instruct.md)
+
+   训练过程中或结束后，比较两个不同的checkpoint，评估模型相似度。
+
+9. [趋势可视化](docs/zh/accuracy_compare/trend_visualization_instruct.md)
+
+   将msProbe工具数据采集或训练状态监测的统计量数据从迭代步数、节点rank和张量目标三个维度进行趋势可视化，方便用户从整体的趋势分布观测精度数据。
+
+
+### MSAdapter场景
+
+1. [数据采集](docs/zh/dump/msadapter_data_dump_instruct.md)
+
+   通过config.json配置，完成msProbe精度数据采集操作。
+
+   config.json配置文件详细介绍请参见[配置文件介绍](docs/zh/dump/config_json_introduct.md)和[config.json配置样例](docs/zh/dump/config_json_examples.md)。
+
+2. [checkpoint比对](docs/zh/checkpoint_compare_instruct.md)
+
+   训练过程中或结束后，比较两个不同的checkpoint，评估模型相似度。
+
+
+## 补充材料
+
+- [PyTorch场景的精度数据采集基线报告](docs/zh/baseline/pytorch_data_dump_perf_baseline.md)
+
+- [MindSpore场景的精度预检基线报告](docs/zh/baseline/mindspore_accuracy_checker_perf_baseline.md)
+
+- [MindSpore场景的精度数据采集基线报告](docs/zh/baseline/mindspore_data_dump_perf_baseline.md)
+
+- [训练状态监测工具标准性能基线报告](docs/zh/baseline/monitor_perf_baseline.md)
+
+
+## FAQ
+
+FAQ汇总了在使用msProbe工具过程中可能遇到的问题，具体请参见[FAQ](docs/zh/faq.md)。
+
+## 贡献指导
+
+介绍如何向msProbe反馈问题、需求以及为msProbe贡献的代码开发流程，具体请参见[为MindStudio Probe贡献](CONTRIBUTING.md)。
+
+## 联系我们
+
+<div>
+  <a href="https://raw.gitcode.com/kali20gakki1/Imageshack/raw/main/CDC0BEE2-8F11-477D-BD55-77A15417D7D1_4_5005_c.jpeg">
+    <img src="https://img.shields.io/badge/WeChat-07C160?style=for-the-badge&logo=wechat&logoColor=white"></a>
+</div>
+
+## 免责声明
+
+- 本工具仅供调试和开发之用，使用者需自行承担使用风险，并理解以下内容：
+  - 数据处理及删除：用户在使用本工具过程中产生的数据属于用户责任范畴。建议用户在使用完毕后及时删除相关数据，以防信息泄露。
+  - 数据保密与传播：使用者了解并同意不得将通过本工具产生的数据随意外发或传播。对于由此产生的信息泄露、数据泄露或其他不良后果，本工具及其开发者概不负责。
+  - 用户输入安全性：用户需自行保证输入的命令行的安全性，并承担因输入不当而导致的任何安全风险或损失。对于由于输入命令行不当所导致的问题，本工具及其开发者概不负责。
+- 免责声明范围：本免责声明适用于所有使用本工具的个人或实体。使用本工具即表示您同意并接受本声明的内容，并愿意承担因使用该功能而产生的风险和责任，如有异议请停止使用本工具。
+- 在使用本工具之前，请谨慎阅读并理解以上免责声明的内容。对于使用本工具所产生的任何问题或疑问，请及时联系开发者。
+
+## License
+
+介绍msProbe产品的使用许可证，具体请参见[LICENSE](LICENSE)文件。
+
+介绍msProbe工具docs目录下的文档适用CC-BY 4.0许可证，具体请参见[LICENSE](docs/LICENSE)文件。
+
+## 致谢
+
+msProbe由华为公司的下列部门联合贡献：
+
+- 昇腾计算MindStudio开发部
+- 分布式并行计算实验室
+
+感谢来自社区的每一个PR，欢迎贡献msProbe！

mindstudio_probe-26.0.0a1/plugins/tb_graph_ascend/secure_build.py

@@ -0,0 +1,175 @@
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------
+
+import os
+import shutil
+import subprocess
+import sys
+import tempfile
+import glob
+import zipfile
+import logging
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+
+class SecureBuildManager:
+    """安全构建管理器"""
+    
+    def __init__(self, source_dir):
+        self.source_dir = source_dir
+        self.original_cwd = os.getcwd()
+    
+    @staticmethod
+    def _fix_whl_permissions(whl_file_path):
+        """修复whl文件内权限"""
+        temp_file_path = whl_file_path + ".new"
+        
+        try:
+            with zipfile.ZipFile(whl_file_path, 'r') as source_zip:
+                with zipfile.ZipFile(temp_file_path, 'w') as target_zip:
+                    for zip_item in source_zip.infolist():
+                        file_data = source_zip.read(zip_item.filename)
+                        
+                        new_zip_item = zipfile.ZipInfo(zip_item.filename)
+                        new_zip_item.date_time = zip_item.date_time
+                        new_zip_item.compress_type = zip_item.compress_type
+                        
+                        # 设置文件权限
+                        permission = 0o550 << 16 if zip_item.filename.endswith('/') else 0o440 << 16
+                        new_zip_item.external_attr = permission
+                        
+                        target_zip.writestr(new_zip_item, file_data)
+            
+            os.replace(temp_file_path, whl_file_path)
+            
+        except Exception as error:
+            if os.path.exists(temp_file_path):
+                os.remove(temp_file_path)
+            raise error
+    
+    @staticmethod
+    def _transfer_artifacts(temp_dist_dir, original_dist_dir):
+        """将构建产物传输回原目录"""
+        os.makedirs(original_dist_dir, exist_ok=True)
+        
+        for whl_file in glob.glob(os.path.join(temp_dist_dir, '*.whl')):
+            target_file = os.path.join(original_dist_dir, os.path.basename(whl_file))
+            shutil.copy2(whl_file, target_file)
+
+    def execute_secure_build(self):
+        """执行安全构建流程"""
+        with tempfile.TemporaryDirectory(prefix="secure_build_") as temp_dir:
+            logger.info("创建安全构建环境")
+            
+            self._copy_required_files(temp_dir)
+            os.chdir(temp_dir)
+            
+            try:
+                logger.info("开始构建wheel包")
+                build_result = subprocess.run(
+                    [sys.executable, 'setup.py', 'bdist_wheel'],
+                    capture_output=True, text=True, check=True
+                )
+                
+                if build_result.stderr:
+                    logger.warning("构建过程中出现警告")
+                
+                # 在安全环境中修复权限
+                temp_dist_dir = os.path.join(temp_dir, 'dist')
+                if os.path.exists(temp_dist_dir):
+                    logger.info("执行权限修复")
+                    for whl_file in glob.glob(os.path.join(temp_dist_dir, '*.whl')):
+                        SecureBuildManager._fix_whl_permissions(whl_file)
+                
+                # 传输构建产物
+                original_dist_dir = os.path.join(self.source_dir, 'dist')
+                SecureBuildManager._transfer_artifacts(temp_dist_dir, original_dist_dir)
+                
+                logger.info("安全构建完成")
+                
+            except subprocess.CalledProcessError as error:
+                logger.error("构建过程失败")
+                raise error
+            finally:
+                os.chdir(self.original_cwd)
+
+    def _copy_required_files(self, target_dir):
+        """复制必要的文件到目标目录"""
+        required_items = ['setup.py', 'server']
+        
+        for item in required_items:
+            source_path = os.path.join(self.source_dir, item)
+            target_path = os.path.join(target_dir, item)
+            
+            if not os.path.exists(source_path):
+                continue
+                
+            if os.path.isfile(source_path):
+                shutil.copy2(source_path, target_path)
+            else:
+                shutil.copytree(source_path, target_path, dirs_exist_ok=True)
+
+
+class BuildCleaner:
+    """构建清理器"""
+    
+    def __init__(self, base_dir):
+        self.base_dir = base_dir
+    
+    def clean_all(self):
+        """清理所有构建产物"""
+        cleanup_patterns = [
+            'build',
+            'dist',
+            '*.egg-info',
+            '__pycache__',
+            '*.pyc',
+            '*.pyo'
+        ]
+        
+        for pattern in cleanup_patterns:
+            for path in glob.glob(os.path.join(self.base_dir, pattern)):
+                if not os.path.exists(path):
+                    continue
+                    
+                try:
+                    if os.path.isfile(path):
+                        os.remove(path)
+                    else:
+                        shutil.rmtree(path)
+                except Exception as error:
+                    logger.warning(f"清理失败: {error}")
+
+
+def main():
+    """主执行函数"""
+    if len(sys.argv) > 1 and sys.argv[1] == 'clean':
+        cleaner = BuildCleaner(os.path.dirname(os.path.abspath(__file__)))
+        cleaner.clean_all()
+    else:
+        source_directory = os.path.dirname(os.path.abspath(__file__))
+        build_manager = SecureBuildManager(source_directory)
+        build_manager.execute_secure_build()
+
+
+if __name__ == "__main__":
+    main()

mindstudio_probe-26.0.0a1/plugins/tb_graph_ascend/server/__init__.py

@@ -0,0 +1,15 @@
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------

mindstudio_probe-26.0.0a1/plugins/tb_graph_ascend/server/app/__init__.py

@@ -0,0 +1,15 @@
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------

mindstudio_probe-26.0.0a1/plugins/tb_graph_ascend/server/app/model/__init__.py

@@ -0,0 +1,15 @@
+# -------------------------------------------------------------------------
+#  This file is part of the MindStudio project.
+# Copyright (c) 2025 Huawei Technologies Co.,Ltd.
+#
+# MindStudio is licensed under Mulan PSL v2.
+# You can use this software according to the terms and conditions of the Mulan PSL v2.
+# You may obtain a copy of Mulan PSL v2 at:
+#
+#          http://license.coscl.org.cn/MulanPSL2
+#
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+# See the Mulan PSL v2 for more details.
+# -------------------------------------------------------------------------