Use std::thread in compiler on all platforms.
PiperOrigin-RevId: 651005412 Change-Id: I8dfb142dc2d87ab0af79583f2df168c105b50138
This commit is contained in:
committed by
Copybara-Service
parent
7e97caaafd
commit
b483ce608e
+1
-2
@@ -737,8 +737,7 @@ Automatic computation of actuator length ranges is done at compile time, and the
|
|||||||
mjModel.actuator_lengthrange of the compiled model. If the model is then saved (either as XML or MJB), the computation
|
mjModel.actuator_lengthrange of the compiled model. If the model is then saved (either as XML or MJB), the computation
|
||||||
does not need to be repeated at the next load. This is important because the computation can slow down the model
|
does not need to be repeated at the next load. This is important because the computation can slow down the model
|
||||||
compiler with large musculo-skeletal models. Indeed we have made the compiler multi-threaded just to speed up this
|
compiler with large musculo-skeletal models. Indeed we have made the compiler multi-threaded just to speed up this
|
||||||
operation (different actuators are processed in parallel in different threads). Incidentally, this is why the flag
|
operation (different actuators are processed in parallel in different threads).
|
||||||
'-pthread' is now needed when linking user code against the MuJoCo library on Linux and macOS.
|
|
||||||
|
|
||||||
Automatic computation relies on modified physics simulation. For each actuator we apply force (negative when computing
|
Automatic computation relies on modified physics simulation. For each actuator we apply force (negative when computing
|
||||||
the minimum, positive when computing the maximum) through the actuator's transmission, advance the simulation in a
|
the minimum, positive when computing the maximum) through the actuator's transmission, advance the simulation in a
|
||||||
|
|||||||
+4
-38
@@ -24,6 +24,7 @@
|
|||||||
#include <map>
|
#include <map>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
|
#include <thread>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include <mujoco/mjdata.h>
|
#include <mujoco/mjdata.h>
|
||||||
@@ -52,25 +53,6 @@ using std::string;
|
|||||||
using std::vector;
|
using std::vector;
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
// pthread on Linux, std::thread on Mac and Windows
|
|
||||||
#if defined(__APPLE__) || defined(_WIN32)
|
|
||||||
#include <thread>
|
|
||||||
using std::thread;
|
|
||||||
|
|
||||||
int getnumproc(void) {
|
|
||||||
return thread::hardware_concurrency();
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
#include <pthread.h>
|
|
||||||
#include <sys/sysinfo.h>
|
|
||||||
|
|
||||||
int getnumproc(void) {
|
|
||||||
return get_nprocs();
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//---------------------------------- CONSTRUCTOR AND DESTRUCTOR ------------------------------------
|
//---------------------------------- CONSTRUCTOR AND DESTRUCTOR ------------------------------------
|
||||||
|
|
||||||
// constructor
|
// constructor
|
||||||
@@ -1550,7 +1532,7 @@ void mjCModel::LengthRange(mjModel* m, mjData* data) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// number of threads available, max 16
|
// number of threads available, max 16
|
||||||
const int nthread = mjMIN(16, getnumproc()/2);
|
const int nthread = mjMIN(16, std::thread::hardware_concurrency()/2);
|
||||||
|
|
||||||
// count actuators that need computation
|
// count actuators that need computation
|
||||||
int cnt = 0;
|
int cnt = 0;
|
||||||
@@ -1609,12 +1591,10 @@ void mjCModel::LengthRange(mjModel* m, mjData* data) {
|
|||||||
err[i][0] = 0;
|
err[i][0] = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
// use std::thread
|
|
||||||
#if defined(_WIN32) || defined(__APPLE__)
|
|
||||||
// launch threads
|
// launch threads
|
||||||
thread th[16];
|
std::thread th[16];
|
||||||
for (int i=0; i<nthread; i++) {
|
for (int i=0; i<nthread; i++) {
|
||||||
th[i] = thread(LRfunc, arg+i);
|
th[i] = std::thread(LRfunc, arg+i);
|
||||||
}
|
}
|
||||||
|
|
||||||
// wait for threads to finish
|
// wait for threads to finish
|
||||||
@@ -1622,20 +1602,6 @@ void mjCModel::LengthRange(mjModel* m, mjData* data) {
|
|||||||
th[i].join();
|
th[i].join();
|
||||||
}
|
}
|
||||||
|
|
||||||
// use pthread
|
|
||||||
#else
|
|
||||||
// launch threads
|
|
||||||
pthread_t th[16];
|
|
||||||
for (int i=0; i<nthread; i++) {
|
|
||||||
pthread_create(th+i, NULL, LRfunc, arg+i);
|
|
||||||
}
|
|
||||||
|
|
||||||
// wait for threads to finish
|
|
||||||
for (int i=0; i<nthread; i++) {
|
|
||||||
pthread_join(th[i], NULL);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// free mjData allocated here
|
// free mjData allocated here
|
||||||
for (int i=1; i<nthread; i++) {
|
for (int i=1; i<nthread; i++) {
|
||||||
mj_deleteData(pdata[i]);
|
mj_deleteData(pdata[i]);
|
||||||
|
|||||||
Reference in New Issue
Block a user