[GitHub] [tvm] HungYangChang commented on a change in pull request #9802: [Runtime][ThreadPool]Refactor affinity function and support CPU affinity list setting.

[GitBox <gi...@apache.org>]

HungYangChang commented on a change in pull request #9802:
URL: https://github.com/apache/tvm/pull/9802#discussion_r776877238



##########
File path: include/tvm/runtime/threading_backend.h
##########
@@ -64,21 +85,24 @@ class ThreadGroup {
   enum AffinityMode : int {
     kBig = 1,
     kLittle = -1,
+    kSpecify = -2,
   };
 
   /*!
    * \brief configure the CPU id affinity
    *
    * \param mode The preferred CPU type (1 = big, -1 = little).
    * \param nthreads The number of threads to use (0 = use all).
+   * \param cpus A list of cpu to use for affinity setting.

Review comment:
       @huajsj Question regarding cpu list parameter "cpus":
   If I wanna split the network into two sub-graphs, then set the first graph → 4 small cores, second graph ->4 big cores. Should I set cpus =[4,4]?




-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org