support custom internal path for debug. · Pull Request !39 · MindSpore/ms_custom_ops

What type of PR is this?

/kind task

What does this PR do / why do we need it:
ms_custom_ops support custom internal path setting.
Which issue(s) this PR fixes:

Fixes #ID26ZP

Test Plan and Test result：What scenarios were tested, and what were the verification results（Function, performance, reliability, etc.）：

Self-checklist:（请自检，在[ ]内打上x，我们将检视你的完成情况，否则会导致pr无法合入）
+ - [x ] **设计**：PR对应的方案是否已经经过Maintainer评审，方案检视意见是否均已答复并完成方案修改
+ - [x ] **测试**：PR中的代码是否已有UT/ST测试用例进行充分的覆盖，新增测试用例是否随本PR一并上库或已经上库
+ - [x ] **验证**：PR描述信息中是否已包含对该PR对应的Feature、Refactor、Bugfix的预期目标达成情况的详细验证结果描述
+ - [ ] **接口**：是否涉及对外接口变更，相应变更已得到接口评审组织的通过，API对应的注释信息已经刷新正确
+ - [x ] **文档**：是否涉及官网文档修改，如果涉及请及时提交资料到Doc仓

**功能概述** 
本次提交主要为MindSpore自定义算子开发支持自定义内部路径调试功能，通过引入环境变量MS_CUSTOM_INTERNAL_KERNEL_HOME，允许开发者指定自定义的内部内核路径，从而方便调试和测试。

**主要功能特性** 
 **1. 环境变量支持机制** 
 **问题描述** 
- 这里是列表文本原有代码硬编码使用固定的prebuild路径，不利于调试和定制化开发
- 这里是列表文本缺少灵活的内部内核路径配置机制

**解决方案** 
- 引入MS_CUSTOM_INTERNAL_KERNEL_HOME环境变量
- 当环境变量存在时，优先使用其指定的路径
- 不存在时回退到默认的prebuild路径

**2. 多模块统一适配** 
 **CMake构建系统**

- 修改CMakeLists.txt，支持环境变量路径检测
- 动态设置内部内核的包含路径和库路径
- 保持路径存在性检查和警告机制

**Python构建脚本**

- 在build_custom_with_ms.py中根据环境变量设置rpath标志
- 正确处理绝对路径和相对路径的格式差异

**运行时环境初始化**

- 修改_init_env函数，优先从环境变量路径加载内部内核库
- 增强错误处理，添加库文件存在性检查

**Shell预构建脚本**

- 优化prepare_prebuild.sh，仅在环境变量不存在时执行默认解压操作
- 改进架构检测逻辑，提高脚本健壮性

**技术实现细节** 
 **路径处理逻辑**

```bash
复制
# 环境变量优先
if [ -n "${MS_CUSTOM_INTERNAL_KERNEL_HOME}" ]; then
    # 使用自定义路径
else
    # 回退到默认路径
fi
```

**编译配置**

- 头文件路径：${MS_CUSTOM_INTERNAL_KERNEL_HOME}/internal_kernel/include
- 库文件路径：${MS_CUSTOM_INTERNAL_KERNEL_HOME}/internal_kernel/lib
- 运行时路径：正确处理rpath设置
 **测试验证** 
 **环境变量功能测试** 
- 验证环境变量存在时的路径正确性
- 验证环境变量不存在时的回退机制

**编译构建测试**

- 确认CMake配置正确识别环境变量路径
- 验证Python构建脚本生成正确的链接参数

**运行时测试**

- 测试环境变量路径下的库文件加载
- 验证错误路径的异常处理机制

**兼容性说明**

- 向后兼容：环境变量不存在时完全保持原有行为
- 路径结构：要求自定义路径保持与默认路径相同的目录结构
- 跨平台支持：Shell脚本支持x86_64和aarch64架构检测

**使用方式**

```bash
# 设置自定义内部内核路径
export MS_CUSTOM_INTERNAL_KERNEL_HOME=/ms_kernels_internal_repo_path/ms_kernels_internal

# 正常执行构建流程
python setup.py install
```

**影响范围**

- 构建系统：CMake配置和Python构建脚本
- 运行时：环境初始化和库加载机制
- 部署：预构建脚本的逻辑调整

类型	指派人员	状态
审查	fangwenyi	已完成 (0/0人)
测试	fangwenyi	已完成 (0/0人)

MindSpore/ms_custom_ops .gitee-modal { width: 500px !important; }

搜索帮助

MindSpore/ms_custom_ops