mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
7b8c7b11d2
Summary: After this, we should have contbuild guarding the Windows build both with and without CUDA. This includes a series of changes that are needed to make Windows build, specifically: (1) Various flags that are needed in the cmake system, specially dealing with /MD, /MT, cuda, cudnn, whole static linking, etc. (2) Contbuild scripts based on appveyo. (3) For Windows build, note that one will need to use "cmake --build" to build stuff so that the build type is consistent between configuration and actual build. see scripts\build_windows.bat for details. (4) In logging.h, ERROR is already defined by Windows. I don't have a good solution now, and as a result, LOG(ERROR) on windows is going to be LOG(INFO). (5) variable length array is not supported by MSVC (and it is not part of C++ standard). As a result I replaced them with vectors. (6) sched.h is not available on Windows, so akyrola 's awesome simple async net might encounter some slowdown due to no affinity setting on Windows. (7) MSVC has a Closes https://github.com/caffe2/caffe2/pull/183 Reviewed By: ajtulloch Differential Revision: D4657831 Pulled By: Yangqing fbshipit-source-id: 070ded372ed78a7e3e3919fdffa1d337640f146e
111 lines
3.2 KiB
C
111 lines
3.2 KiB
C
// pthreadpool header from https://github.com/Maratyszcza/pthreadpool
|
|
// for NNPACK
|
|
#ifndef CAFFE2_UTILS_PTHREADPOOL_H_
|
|
#define CAFFE2_UTILS_PTHREADPOOL_H_
|
|
|
|
#include "ThreadPoolCommon.h"
|
|
|
|
#ifndef CAFFE2_THREADPOOL_MOBILE
|
|
#error "mobile build state not defined"
|
|
#endif
|
|
|
|
// ThreadPool only used in mobile builds at the moment
|
|
#if CAFFE2_THREADPOOL_MOBILE
|
|
|
|
#include <stddef.h> // for size_t
|
|
|
|
typedef struct pthreadpool* pthreadpool_t;
|
|
|
|
typedef void (*pthreadpool_function_1d_t)(void*, size_t);
|
|
typedef void (*pthreadpool_function_1d_tiled_t)(void*, size_t, size_t);
|
|
typedef void (*pthreadpool_function_2d_t)(void*, size_t, size_t);
|
|
typedef void (*pthreadpool_function_2d_tiled_t)(void*, size_t, size_t, size_t, size_t);
|
|
typedef void (*pthreadpool_function_3d_t)(void*, size_t, size_t, size_t);
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/**
|
|
* Creates a thread pool with the specified number of threads.
|
|
*
|
|
* @param[in] threads_count The number of threads in the thread pool.
|
|
* A value of 0 has special interpretation: it creates a thread for each
|
|
* processor core available in the system.
|
|
*
|
|
* @returns A pointer to an opaque thread pool object.
|
|
* On error the function returns NULL and sets errno accordingly.
|
|
*/
|
|
pthreadpool_t pthreadpool_create(size_t threads_count);
|
|
|
|
/**
|
|
* Queries the number of threads in a thread pool.
|
|
*
|
|
* @param[in] threadpool The thread pool to query.
|
|
*
|
|
* @returns The number of threads in the thread pool.
|
|
*/
|
|
size_t pthreadpool_get_threads_count(pthreadpool_t threadpool);
|
|
|
|
|
|
/**
|
|
* Processes items in parallel using threads from a thread pool.
|
|
*
|
|
* When the call returns, all items have been processed and the thread pool is
|
|
* ready for a new task.
|
|
*
|
|
* @note If multiple threads call this function with the same thread pool, the
|
|
* calls are serialized.
|
|
*
|
|
* @param[in] threadpool The thread pool to use for parallelisation.
|
|
* @param[in] function The function to call for each item.
|
|
* @param[in] argument The first argument passed to the @a function.
|
|
* @param[in] items The number of items to process. The @a function
|
|
* will be called once for each item.
|
|
*/
|
|
void pthreadpool_compute_1d(
|
|
pthreadpool_t threadpool,
|
|
pthreadpool_function_1d_t function,
|
|
void* argument,
|
|
size_t range);
|
|
|
|
void pthreadpool_compute_1d_tiled(
|
|
pthreadpool_t threadpool,
|
|
pthreadpool_function_1d_tiled_t function,
|
|
void* argument,
|
|
size_t range,
|
|
size_t tile);
|
|
|
|
void pthreadpool_compute_2d(
|
|
pthreadpool_t threadpool,
|
|
pthreadpool_function_2d_t function,
|
|
void* argument,
|
|
size_t range_i,
|
|
size_t range_j);
|
|
|
|
void pthreadpool_compute_2d_tiled(
|
|
pthreadpool_t threadpool,
|
|
pthreadpool_function_2d_tiled_t function,
|
|
void* argument,
|
|
size_t range_i,
|
|
size_t range_j,
|
|
size_t tile_i,
|
|
size_t tile_j);
|
|
|
|
/**
|
|
* Terminates threads in the thread pool and releases associated resources.
|
|
*
|
|
* @warning Accessing the thread pool after a call to this function constitutes
|
|
* undefined behaviour and may cause data corruption.
|
|
*
|
|
* @param[in,out] threadpool The thread pool to destroy.
|
|
*/
|
|
void pthreadpool_destroy(pthreadpool_t threadpool);
|
|
|
|
#ifdef __cplusplus
|
|
} /* extern "C" */
|
|
#endif
|
|
|
|
#endif // CAFFE2_THREADPOOL_MOBILE
|
|
|
|
#endif // CAFFE2_UTILS_PTHREADPOOL_H_
|