0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
ascend-robotascend-robotfix: lstm fallback to acl_op for bf16 and mixed floating dtypes
90f9a798创建于 4月2日历史提交
文件最后提交记录最后更新时间
.keep
!875 op_plugin ut run with torch_npu version. Merge pull request !875 from 李金鑫/ut_master 2 年前
test___ior__.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test___or__.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test__aminmax.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test__iRshift__.py
rshift apply aclnn Co-authored-by: SCh-zx<1325467101@qq.com> # message auto-generated for no-merge-commit merge: !3897 merge master into master rshift apply aclnn Created-by: SCh_zx Commit-by: SCh-zx Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature **What does this PR do / why do we need it**: 添加aclnn路径 修复用例没有测试标量的问题 **Special notes for your reviewers**: See merge request: Ascend/op-plugin!38975 个月前
test__ilshift__.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test__lshift__.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test__npu_quant_rms_norm.py
!2176 [Task] add quant_rms_norm op. Merge pull request !2176 from will-devil/master 1 年前
test__rshift__.py
rshift apply aclnn Co-authored-by: SCh-zx<1325467101@qq.com> # message auto-generated for no-merge-commit merge: !3897 merge master into master rshift apply aclnn Created-by: SCh_zx Commit-by: SCh-zx Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature **What does this PR do / why do we need it**: 添加aclnn路径 修复用例没有测试标量的问题 **Special notes for your reviewers**: See merge request: Ascend/op-plugin!38975 个月前
test__svd_helper.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_abs.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_absolute.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_acos.py
!1781 Move part opapi kernel to autogen from a to b Merge pull request !1781 from 王广斌/new_struct 1 年前
test_acosh.py
!2142 Add device_check testcase Merge pull request !2142 from wgb/bugfix 1 年前
test_adaptive_avg_pool1d.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_adaptive_avg_pool2d_backward.py
!2210 [Fix] Fix special case and add check for adaptive_avg_pool2d. Merge pull request !2210 from 刘嘉巍/dev 1 年前
test_adaptive_avg_pool3d.py
!2777 fix fuzz add check_adaptive_avg_pool3d Merge pull request !2777 from zhangqiongwen/master_check_adaptive_avg_pool3d 10 个月前
test_adaptive_avg_pool3d_backward.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_adaptive_max_pool2d.py
!1781 Move part opapi kernel to autogen from a to b Merge pull request !1781 from 王广斌/new_struct 1 年前
test_adaptive_max_pool2d_backward.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_adaptive_max_pool3d.py
!2355 adaptivemaxpool3d fallback Merge pull request !2355 from 詹昊/master 1 年前
test_adaptive_max_pool3d_backward.py
!2355 adaptivemaxpool3d fallback Merge pull request !2355 from 詹昊/master 1 年前
test_add.py
!2748 Fuzz fix add and addr Merge pull request !2748 from wgb/bugfix 10 个月前
test_add_relu.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_addbmm.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_addcdiv.py
!2770 fix fuzz addcdiv Merge pull request !2770 from zhangqiongwen/master_addcdiv_fuzz 10 个月前
test_addcmul.py
!2770 fix fuzz addcdiv Merge pull request !2770 from zhangqiongwen/master_addcdiv_fuzz 10 个月前
test_addmm.py
!1524 【Addmm】API fix precision loss Merge pull request !1524 from 陈志敏/master 2 年前
test_addmm_activation_out.py
注册_addmm_activation_out接口 Co-authored-by: jjt_3<1926550527@qq.com> # message auto-generated for no-merge-commit merge: !4164 merge addmmactivationout into master 注册_addmm_activation_out接口 Created-by: jjt_3 Commit-by: jjt_3 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature **What does this PR do / why do we need it**: **Special notes for your reviewers**: See merge request: Ascend/op-plugin!41643 个月前
test_addmv.py
修改torch.addmv算子输出类型：在alpha或beta不等于1时，输出类型改为int64 Co-authored-by: xiaoqi-zhou<zhouxiaoqi9@h-partners.com> # message auto-generated for no-merge-commit merge: !4263 merge addmv_master into master 修改torch.addmv算子输出类型：在alpha或beta不等于1时，输出类型改为int64 Created-by: xiaoqi-zhou Commit-by: xiaoqi-zhou Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 https://gitcode.com/Ascend/op-plugin/issues/35 - [ ] 需求 - [ ] 问题单 - [x] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 修改AddmvKernelNpuOpApi.cpp文件的addmv方法，加判断条件： 在alpha或beta不等于1时，输出类型改为int64 # 【资料变更】 > 请确认是否涉及资料变更。如涉及，需要在PR中体现，并简要说明修改内容。如不涉及，需填写“不涉及” # 【接口变更】 > 请确认是否涉及跨代码仓或者客户面可见的接口变更。如涉及，需要详细说明接口以及对应的变更内容，同时需要在资料中体现。如不涉及，需填写“不涉及” # 【功能验证】 > 说明测试场景，测试方法。如果本次测试方式与常规单元测试不同，请详细说明您的测试步骤\ > 新增/变更内容是否已新增/适配UT测试用例看护，并补充测试自验证截图 测试代码如下： import torch import torch_npu mat = torch.tensor([[True, True], [True, True], [True, True]]).npu() # 3x2 矩阵 vec = torch.tensor([True, True]).npu() # 2维向量 input_vec = torch.tensor([True, True, True]).npu() # 3维向量 result = torch.addmv(input_vec, mat, vec, beta=2, alpha=2) print("基本用法结果:") print(result) 输出结果正常： ![image.png](https://raw.gitcode.com/user-images/assets/7403085/cdd58f6b-01c7-4cef-8aca-946fc342f058/image.png 'image.png') 新增UT：test_addmv.py中的test_addmv_bool_promotion_with_non_one_scalars UT正常跑通： ![image.png](https://raw.gitcode.com/user-images/assets/7403085/c3646293-1654-46ce-82e7-9f8b5944dcbb/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [ ] 代码注释完备，正确记录错误日志 - [ ] 代码实现进行了返回值、空指针等校验 - [ ] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [ ] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!42633 个月前
test_addr.py
!2748 Fuzz fix add and addr Merge pull request !2748 from wgb/bugfix 10 个月前
test_affine_grid_generator.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_affine_grid_generator_backward.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_all.py
!1873 make_dim_mask新增ndim = 0判断 Merge pull request !1873 from 彭一鸣/reducesum 1 年前
test_amp_foreach_non_finite_check_and_unscale_.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_anchorresponseflags.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_angle.py
!1598 [Feature] Support named tensor [1/3]. Merge pull request !1598 from 刘嘉巍/master-add 1 年前
test_any.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_api_common.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_apply_adam.py
!1506 Fix the adapter and ut of apply_adam Merge pull request !1506 from 张向龙3/apply_adam_0416 2 年前
test_apply_adam_w.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_arange.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_arccosh.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_arcsin.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_arcsinh.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_arctan.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_arctanh.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_argmax.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_argmax_out.py
!2108 The argmax operator adapts to the null tensor and null dim check. Merge pull request !2108 from 史佳馨/master 1 年前
test_argmin.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_argsort.py
!1675 [adpt]add argsort.stable in opapi Merge pull request !1675 from zhangqiongwen/master_argsort_stable_adpt 1 年前
test_as_strided.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_asin.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_asinh.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_atan.py
!1781 Move part opapi kernel to autogen from a to b Merge pull request !1781 from 王广斌/new_struct 1 年前
test_atan2.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_atanh.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_avg_pool2d.py
Fix avg_pool2d infershape bug Co-authored-by: wang-guangbin<wgb_strive@163.com> # message auto-generated for no-merge-commit merge: !3593 merge bugfix into master Fix avg_pool2d infershape bug Created-by: wang-guangbin Commit-by: wang-guangbin Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature **What does this PR do / why do we need it**: **Special notes for your reviewers**: See merge request: Ascend/op-plugin!35936 个月前
test_avg_pool2d_backward.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_avg_pool3d.py
!1846 Add AvgPool3d op api Merge pull request !1846 from 冯彤/avg_pool3d 1 年前
test_avg_pool3d_backward.py
!1845 增加avgpool3dbackward的路径五调用 Merge pull request !1845 from 唐燕锋/avgpool3d 1 年前
test_baddbmm.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batch_nms.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batch_norm.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batch_norm_elemt.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batch_norm_reduce.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batch_norm_stats.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batchnorm3d_backward.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_batchnorm_backward.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_batchnorm_backward_elemt.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batchnorm_backward_eval.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batchnorm_backward_reduce.py
!1477 Move UTs on PTA to OP-Plugin Part1 Merge pull request !1477 from 张向龙3/mv_ut 2 年前
test_batchnorm_gather_stats_update.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_batchnorm_gather_stats_with_counts.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bernoulli.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_binary_cross_entropy.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_binary_cross_entropy_backward.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_binary_cross_entropy_with_logits.py
fix ut Co-authored-by: DaiFu<daifu2@huawei.com> # message auto-generated for no-merge-commit merge: merge 2509ut into master fix ut Created-by: daifu1234567 Commit-by: DaiFu Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** fix test_binary_cross_with_logits_float16 by setting random seed > Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature **What does this PR do / why do we need it**: **Special notes for your reviewers**: See merge request: Ascend/op-plugin!32648 个月前
test_binary_cross_entropy_with_logits_backward.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bincount.py
!2142 Add device_check testcase Merge pull request !2142 from wgb/bugfix 1 年前
test_bitwise_and.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bitwise_not.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bitwise_or.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bitwisexor.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bmm.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bmmV2.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bounding_box_decode.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bounding_box_encode.py
!1988 Fix some UTs on atlas_a2 part 3 Merge pull request !1988 from 张向龙3/fix_910b_ut_3 1 年前
test_broadcast.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_bucketize.py
!1514 auto-gdd into open source community Merge pull request !1514 from 王育才/master 1 年前
test_cast.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_cat.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_cdist.py
fix the bug in cdist and add the test instance Co-authored-by: jjt_3<1926550527@qq.com> # message auto-generated for no-merge-commit merge: !4085 merge fix_cdist_bug_master into master fix the bug in cdist and add the test instance Created-by: jjt_3 Commit-by: jjt_3 Merged-by: ascend-robot Description: <!-- Thanks for sending a pull request! --> **What type of PR is this?** > Uncomment only one /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line: > > /kind bug > /kind task > /kind feature **What does this PR do / why do we need it**: **Special notes for your reviewers**: See merge request: Ascend/op-plugin!40854 个月前
test_cdist_backward.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_ceil.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_celu.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_channel_shuffle.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_chunk_cat_out.py
add new api _chunk_cat Co-authored-by: jjt_3<1926550527@qq.com> # message auto-generated for no-merge-commit merge: !4509 merge chunkcat into master add new api _chunk_cat Created-by: jjt_3 Commit-by: jjt_3 Merged-by: ascend-robot Description: <!-- PR描述模板更新日期：20260203 --> # 【合入来源】 > <font color="red">**如有社区issue，请关联issue链接**</font>\ > <font color="red">**请勿携带内部流程信息（需求链接、问题单、内部issue等）**</font> - [x] 需求 - [ ] 问题单 - [ ] issue/工单 - [ ] 重构优化 - [ ] 资料更新 # 【修改方案】 > _chunk_cat和_chunk_cat.out算子适配 > 符合以下条件之一走入执行aclnnchunkcat算子分支： > a.输入tensor数据类型是bfloat16/float16,输出tensor数据类型是bfloat16/float16/float32,dim=0,contiguous > b.输入tensor数据类型是float32,输出tensor数据类型是float32,dim=0,contiguous > 其余情况走native实现，小算子拼接 # 【资料变更】 > 不涉及 # 【接口变更】 > 不涉及 # 【功能验证】 ![image.png](https://raw.gitcode.com/user-images/assets/7403085/89d57d5b-1499-4987-aa6b-691d2a3ae40a/image.png 'image.png') # 【CheckList】 > PR提交人对以下CheckList自检项进行全量自检，自检通过或不涉及，均修改 [ ] 为 [x] - [x] 代码注释完备，正确记录错误日志 - [x] 代码实现进行了返回值、空指针等校验 - [x] PR标题正确使用类型标签，如：feat、fix、refactor、docs、test等 - [x] PR持续集成流水线（CI）执行通过，代码检查无异常 See merge request: Ascend/op-plugin!45092 个月前
test_ciou.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前
test_ciou_backward.py
!1513 Move UTs on PTA to OP-Plugin Part2 Merge pull request !1513 from 张向龙3/mv_ut_2 2 年前