parallel.h 1.93 KB
Newer Older
Gerard Ryan's avatar
Gerard Ryan committed
1 2
/**
 * @file parallel.h This file contains the functionality for parallel operation
Gerard Ryan's avatar
Gerard Ryan committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
 * @author  TPOC: palisade@njit.edu
 *
 * @copyright Copyright (c) 2017, New Jersey Institute of Technology (NJIT)
 * All rights reserved.
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 * 1. Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice, this
 * list of conditions and the following disclaimer in the documentation and/or other
 * materials provided with the distribution.
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

Gerard Ryan's avatar
Gerard Ryan committed
27 28 29 30 31 32 33 34 35 36 37 38 39

#ifndef SRC_CORE_LIB_UTILS_PARALLEL_H_
#define SRC_CORE_LIB_UTILS_PARALLEL_H_

#include "omp.h"

namespace lbcrypto {

class ParallelControls {
	int machineThreads;
public:
	ParallelControls() {
		machineThreads = omp_get_max_threads();
Gerard Ryan's avatar
Gerard Ryan committed
40
		Enable();
Gerard Ryan's avatar
Gerard Ryan committed
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
	}

	void Enable() {
		omp_set_num_threads(machineThreads);
	}

	void Disable() {
		omp_set_num_threads(0);
	}
};

extern ParallelControls PalisadeParallelControls;

}

#endif /* SRC_CORE_LIB_UTILS_PARALLEL_H_ */