tbb_test/Partition.h at master · jspark1105/tbb_test · GitHub

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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#pragma once

#include <algorithm>
#include <cassert>
#include <cmath>
#include <utility>

extern int nsockets;

int get_num_threads_per_socket();
int get_socket_num();
int get_thread_num_in_socket();

int get_work_per_socket(int work);
std::pair<int, int> get_partition(int work);
std::pair<int, int>
get_partition(int work, int nthreads_per_socket, int tid_in_socket);
std::pair<int, int> get_partition(
    int work,
    int socket_id,
    int nthreads_per_socket,
    int tid_in_socket);

void get_intra_socket_2dpartition(
    int* m_begin,
    int* m_end,
    int* n_begin,
    int* n_end,
    int m,
    int n,
    double aspect_ratio,
    bool m_align);
void get_intra_socket_2dpartition(
    int* m_begin,
    int* m_end,
    int* n_begin,
    int* n_end,
    int m,
    int n,
    double aspect_ratio,
    bool m_align,
    int nthreads_per_socket,
    int tid_in_socket);

void get_2dpartition(
    int* m_begin,
    int* m_end,
    int* n_begin,
    int* n_end,
    int m,
    int n,
    double aspect_ratio,
    bool m_align);
void get_2dpartition(
    int* m_begin,
    int* m_end,
    int* n_begin,
    int* n_end,
    int m,
    int n,
    double aspect_ratio,
    bool m_align,
    int nthreads_per_socket,
    int tid_in_socket);
void get_2dpartition(
    int* m_begin,
    int* m_end,
    int* n_begin,
    int* n_end,
    int m,
    int n,
    double aspect_ratio,
    bool m_align,
    int socket_id,
    int nthreads_per_socket,
    int tid_in_socket);