How to specify which DLA to use?

Hello,

My question is how to specify the usage of DLA? for example:

  1. use DLA0 for inference 0, and use DLA1 for inference 1?
  2. If I have inference 0/1/2/3 to run together, use DLA0 for inference 0/1, and use DLA1 for inference 2/3 ?

Thanks,
-Owen

Hi Owen, the function you want to use is IBuilder::setDLACore() and IRuntime::setDLACore()

These will allow you to distribute your networks between the DLA engines as you desire.

Thanks for the replying! It works.