×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

集成到tf里,如果想要改成多核该怎么做? liujing2020-05-26 17:30:11 回复 查看 高校课程
集成到tf里,如果想要改成多核该怎么做?
分享到:

集成到tf里,如果想要改成多核(更进一步,想要动态设定function_type,随意指定核),该怎么做?对于单算子,我们可以在`invokeKernel`时候指定,但集成的时候是在哪里指定有多少核的?


  1. config = tf.ConfigProto()

  2. config.mlu_options.core_num = 16 //执⾏运算的 MLU270 内核数量为 16

  3. ... 

  4. with tf.Session(config = config, graph = graph) as sess:

  5. result = sess.run(...)





启动的Dim是几维的呢?

bangc的Dim和cuda是差不多的.这个是寒武纪的库自动实现的,会选择最优的方案,看不到源码的。

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭