已合并
ut: add inductor a5 simt ut #31641
stonexxx创建于 3月11日
ut: add inductor a5 simt ut #31641
已合并
Pull Request已成功合入, 合并人@ascend-robot
（感谢 stonexxx 的贡献）
AtlasAccount
3月11日 评论：
3月11日 评论：
ascend-robot
3月11日 评论：
3月11日 评论：
Thanks for your pull-request.
The full list of commands accepted by me can be found at here。
You can get sig-info at here
PR Approval Progress
✅ Congratulations! All modules have met the lgtm and approve requirements.
Module Approval Details
| module | lgtm status | approve status |
|---|---|---|
| repo-Ascend/pytorch | ✅ zichun_ye, liangsongwei (2/2) | ✅ zichun_ye (1/1) |
| test | ✅ crazyDannyBoy, zichun_ye, weizhan4, liangsongwei (4/2) | ✅ crazyDannyBoy, zichun_ye (2/1) |
💡 Tip:
- Committer can comment
/approveor/lgtm- Commenting
/approveimplies both code review (lgtm) and intent to merge (approve)
3月11日 添加了label：ascend-cla/yes
AtlasAccount
3月11日 评论：
3月11日 评论：
问题/功能描述
本次PR主要为PyTorch Inductor的多个算子测试启用了SIMD/SIMT混合内存模式，并优化了测试配置。具体包括：在CI测试列表中新增了四个SIMT相关算子的测试项；为embedding、gather、index_select、index四个算子的测试文件设置环境变量以激活特定的内存访问模式；同时，对部分耗时较长的性能基准测试添加了跳过标记，以缩短常规测试的执行时间。
修改方案描述
- 更新CI测试列表：在
ci/a5_test_list.yaml中新增了test_embedding_simt、test_index_simt、test_gather_simt、test_index_select_simt四个测试用例，确保新增的SIMT测试能被CI系统执行。 - 启用SIMD/SIMT混合模式：在四个对应的Python测试文件开头，通过设置环境变量
INDUCTOR_INDIRECT_MEMORY_MODE为”simd_simt_mix”，强制相关算子在Inductor编译时使用特定的内存访问策略进行测试。 - 优化测试执行：将
test_embedding_simt.py、test_index_select_simt.py和test_index_simt.py中的性能基准测试方法标记为@unittest.skip，避免其长时间运行影响测试效率。同时修正了test_embedding_simt.py中test_embedding_slice_cat函数的一个逻辑错误。
此处折叠了62条消息 查看更多
梁松伟
3月12日 评论：
3月12日 评论：
/lgtm
3月12日 添加了label：lgtm
ascend-robot
3月12日 评论：
3月12日 评论：
Review Guide
This pull-request passes review.
Committers who wrote a comment of /approve are: crazyDannyBoy, zichun_ye.
Reviewers who wrote a comment of /lgtm are: zichun_ye, liangsongwei, crazyDannyBoy, weizhan4.
3月12日 合入了pull request
【合入来源】
【修改方案】
【资料变更】
【接口变更】
【功能验证】
【CheckList】