supersat commented on code in PR #11018:
URL: https://github.com/apache/tvm/pull/11018#discussion_r859118107
##########
src/runtime/threading_backend.cc:
##########
@@ -34,13 +34,63 @@
#endif
#if defined(__hexagon__)
#include <dlfcn.h>
+#include <qurt.h>
+#include <stdlib.h>
+#define HEXAGON_STACK_SIZE 65536
+#define HEXAGON_STACK_ALIGNMENT 32
#endif
#include <algorithm>
#include <thread>
#define CURRENT_THREAD_HANDLE (static_cast<std::thread::native_handle_type>(0))
namespace tvm {
namespace runtime {
namespace threading {
+#ifdef __hexagon__
+// pthreads are broken on older versions of qurt, so
+// we need to use native APIs instead of std::threads
+class QuRTThread {
+ typedef std::function<void()> Callback;
+
+ public:
+ explicit QuRTThread(Callback worker_callback) : f_(worker_callback) {
+ static int id = 1;
+ qurt_thread_attr_t attr;
+ char name[32];
+ int ret = posix_memalign(&stack_, HEXAGON_STACK_ALIGNMENT,
HEXAGON_STACK_SIZE);
+ CHECK_EQ(ret, 0);
+ qurt_thread_attr_init(&attr);
+ qurt_thread_attr_set_stack_size(&attr, HEXAGON_STACK_SIZE);
+ qurt_thread_attr_set_stack_addr(&attr, stack_);
+ snprintf(name, sizeof(name), "worker %d", id++);
+ qurt_thread_attr_set_name(&attr, name);
+ ret = qurt_thread_create(&thread_, &attr, (void (*)(void*))RunFunction,
this);
+ CHECK_EQ(ret, QURT_EOK);
+ }
+ QuRTThread(QuRTThread&& other) : thread_(other.thread_), f_(other.f_),
stack_(other.stack_) {
+ other.thread_ = 0;
Review Comment:
free(NULL) is defined to be a no-op, but I added an explicit check anyway.
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
