Star122
162
代码介绍
代码
Issues36
Pull Requests13
流水线
Actions
讨论
Wiki
项目成员75
分析
项目设置
Star122
162
  1. RecSDK
  2. /
  3. cust_op
  4. /
  5. test
  6. /
  7. permute_pooled_embs
  8. /
  9. torch
ascend-robotascend-robot【FEAT】【ops】新增permute_pooled_embs_auto_grad算子测试脚本
bd200549创建于 2月13日历史提交
文件最后提交记录最后更新时间
common.py
【FEAT】【ops】新增permute_pooled_embs_auto_grad算子测试脚本3 个月前
permute_pooled_embedding_modules.py
【FEAT】【ops】新增permute_pooled_embs_auto_grad算子测试脚本3 个月前
permute_pooled_embedding_test.py
【FEAT】【ops】新增permute_pooled_embs_auto_grad算子测试脚本3 个月前
test_permute_pooled_embs.py
【feat】新增permute_pooled_embs算子SIMD kernel实现,基于Datacopy非连续搬运,采用按total_dim维度均匀分核策略,补充测试用例4 个月前