OE版本3.7/3.4
J6B的TopK算子返回的indices的int64类型，这个类型后面接了乘以浮点类型的运算，会引入cast的CPU算子，查看手册J6B目前的cast BPU算子最多只支持int32，有什么方法能消除这个cpu 的cast算子？
Onnx算子如下图
专栏算法工具链J6B怎么消除int64类型的CPU的cast算子
J6B怎么消除int64类型的CPU的cast算子
已解决
刃木2025-12-30
60
6
算法工具链
征程6
评论2
0/1000
- YCJ
'/%3e%3cpath%20d='M8%200.5C12.1421%200.5%2015.5%203.85786%2015.5%208C15.5%2012.1421%2012.1421%2015.5%208%2015.5C3.85786%2015.5%200.5%2012.1421%200.5%208C0.5%203.85786%203.85786%200.5%208%200.5Z'%20fill='url(%23paint1_linear_0_45845)'%20stroke='white'/%3e%3cpath%20d='M10.0597%204.5332C10.031%204.5332%2010.0036%204.54817%209.98859%204.57313L8.09724%207.79442C8.02613%207.90046%207.84897%208.23731%207.5483%208.7875C7.37863%209.0969%207.25387%209.32771%207.17152%209.48615C7.1341%209.55727%207.0293%209.53855%207.01807%209.45871C6.97191%209.11936%206.89456%208.6066%206.78727%207.91793L6.2483%204.61804C6.24207%204.57812%206.20713%204.54942%206.16721%204.54942H4.28085C4.2297%204.54942%204.18977%204.59683%204.201%204.64798L5.71059%2012.1959C5.71808%2012.2346%205.75176%2012.2621%205.79044%2012.2621H7.46845C7.49715%2012.2621%207.52335%2012.2471%207.53832%2012.2234L12.1869%204.65796C12.2205%204.60307%2012.1806%204.5332%2012.117%204.5332H10.0597Z'%20fill='white'/%3e%3cdefs%3e%3clinearGradient%20id='paint0_linear_0_45845'%20x1='8.99711'%20y1='15.3526'%20x2='47.3575'%20y2='15.3526'%20gradientUnits='userSpaceOnUse'%3e%3cstop%20stop-color='%23AF86FF'/%3e%3cstop%20offset='1'%20stop-color='%23774EFF'/%3e%3c/linearGradient%3e%3clinearGradient%20id='paint1_linear_0_45845'%20x1='1.1152'%20y1='15.1368'%20x2='15.3888'%20y2='15.1368'%20gradientUnits='userSpaceOnUse'%3e%3cstop%20stop-color='%23AF86FF'/%3e%3cstop%20offset='1'%20stop-color='%23774EFF'/%3e%3c/linearGradient%3e%3c/defs%3e%3c/svg%3e)
Lv.4您好！您可以在yaml文件的模型参数组 remove_node_type 字段配置"Cast" ，这样在生成hbm模型的时候，工具链会自动帮您去除模型开头或者末尾 cast算子
2025-12-3004- 刃木回复YCJ:
cast算子不在模型开头和结尾，而是模型中间的topK算子的 索引得到的indices数据，和浮点相差的时候会引入cast转换
2025-12-300 - YCJ回复刃木:
您方便分享模型吗？您的onnx模型和yaml校准文件，我这边复现一下
2025-12-300 - YCJ:
方便的话可以发邮箱到873651674@qq.com
2025-12-300 - 刃木回复YCJ:
发送过去了，出问题的相关onnx算子的截图也添加在上面的问题描述里面了
2025-12-300
- HuanghuiLv.5
你好，注意到您是在 J6B 上进行业务开发，J6B应该是有专门工具链得同事进行对接支持得，我们对J6B使用较少，请联系一下接口人员获得帮助呢，感谢理解。
2025-12-3000