专栏算法工具链hb_mapper checker caffe模型的时候同一种算子有时候显示cpu有时候显示bpu

hb_mapper checker caffe模型的时候同一种算子有时候显示cpu有时候显示bpu

已解决
liuyang2020-10-22
168
4

如上图所示，整个网络里面有的转置卷积是BPU有的是CPU， concat层有的是BPU有的是CPU， relu有的是BPU有的是CPU.....这个是什么逻辑？

算法工具链
评论1
0/1000
  • anming
    Lv.1

    这是因为有些算子不被BPU所支持，工具会对计算图做优化，算子会被放到cpu

    2020-10-22
    0
    3
    • liuyang回复anming:

      我的意思是想确认同一种算子比如concat， 有的跑在BPU上， 有的跑在CPU上？

      2020-10-22
      4
    • liuyang回复anming:

      如上图所示， 想知道为什么同一种算子有的会在bpu上运行， 有的会在cpu上运行， 如果硬件支持这种算子为什么不全部跑在bpu上？ 如果是checker工具显示的是CPU运行，还需要自己手动写实际的实现代码吗？ 还是说已经有cpu的实现？

      2020-10-23
      0
    • anming回复liuyang:

      不需要手动实现，显示CPU 说明工具已经支持该算子 且 被分配到 cpu计算

      2020-10-23
      1