打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
集成到tf里,如果想要改成多核(更进一步,想要动态设定function_type,随意指定核),该怎么做?对于单算子,我们可以在`invokeKernel`时候指定,但集成的时候是在哪里指定有多少核的?
config = tf.ConfigProto()
config.mlu_options.core_num = 16 //执⾏运算的 MLU270 内核数量为 16
...
with tf.Session(config = config, graph = graph) as sess:
result = sess.run(...)
启动的Dim是几维的呢?
bangc的Dim和cuda是差不多的.这个是寒武纪的库自动实现的,会选择最优的方案,看不到源码的。
热门帖子
精华帖子