File: benl_parallel.ml

package info (click to toggle)
ben 0.7.0%2Bdeb8u1
  • links: PTS, VCS
  • area: main
  • in suites: jessie
  • size: 460 kB
  • ctags: 776
  • sloc: ml: 3,584; makefile: 86; ansic: 39
file content (46 lines) | stat: -rw-r--r-- 1,854 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
(**************************************************************************)
(*  Copyright © 2014 Mehdi Dogguy <mehdi@dogguy.org>                      *)
(*                                                                        *)
(*  This program is free software: you can redistribute it and/or modify  *)
(*  it under the terms of the GNU Affero General Public License as        *)
(*  published by the Free Software Foundation, either version 3 of the    *)
(*  License, or (at your option) any later version, with the additional   *)
(*  exemption that compiling, linking, and/or using OpenSSL is allowed.   *)
(*                                                                        *)
(*  This program is distributed in the hope that it will be useful, but   *)
(*  WITHOUT ANY WARRANTY; without even the implied warranty of            *)
(*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU     *)
(*  Affero General Public License for more details.                       *)
(*                                                                        *)
(*  You should have received a copy of the GNU Affero General Public      *)
(*  License along with this program.  If not, see                         *)
(*  <http://www.gnu.org/licenses/>.                                       *)
(**************************************************************************)

let level = ref (Parmap.get_default_ncores ())

let set_level l =
  if l > 0 then begin
    Parmap.set_default_ncores l;
    level := l
  end

let get_level () =
  !level

let map ?(level = !level) f l =
  Parmap.parmap ~ncores:level
    f
    (Parmap.L l)

let iter ?(level = !level) f l =
  Parmap.pariter ~ncores:level
    f
    (Parmap.L l)

let fold ?(level = !level) f a l m =
  Parmap.parfold ~ncores:level
    (fun e a -> f a e)
    (Parmap.L l)
    a
    m