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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
/*******************************************************************************
* Copyright 2017-2018 Intel Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*******************************************************************************/
#include <stdlib.h>
#include <stdio.h>
#include <float.h>
#include <math.h>
#include "mkldnn.h"
#include "mkldnn_common.hpp"
#include "mkldnn_memory.hpp"
#include "ip/ip.hpp"
namespace ip {
const dt_conf_t *cfg = conf_f32;
dir_t dir = FWD_B;
int mb = 0;
attr_t attr;
bool allow_unimpl = false;
const char *perf_template = "perf,%D,%n,%z,%q,%-t,%-Gp,%0t,%0Gp";
void reset_parameters() {
cfg = conf_f32;
dir = FWD_B;
mb = 0;
attr = attr_t();
allow_unimpl = false;
}
void check_correctness(const desc_t *c) {
const prb_t p(*c, mb, dir, cfg, attr);
char pstr[max_prb_len];
prb2str(&p, pstr);
res_t res{};
const int status = ip::doit(&p, &res);
bool want_perf_report = false;
parse_result(res, want_perf_report, allow_unimpl, status, pstr);
if (want_perf_report && bench_mode & PERF)
perf_report(&p, &res, pstr);
benchdnn_stat.tests++;
}
int bench(int argc, char **argv, bool main_bench) {
for (int arg = 0; arg < argc; ++arg) {
if (!strncmp("--batch=", argv[arg], 8))
SAFE(batch(argv[arg] + 8, bench), CRIT);
else if (!strncmp("--cfg=", argv[arg], 6))
cfg = str2cfg(argv[arg] + 6);
else if (!strncmp("--mb=", argv[arg], 5))
mb = atoi(argv[arg] + 5);
else if (!strncmp("--dir=", argv[arg], 6))
dir = str2dir(argv[arg] + 6);
else if (!strncmp("--attr=", argv[arg], 7))
SAFE(str2attr(&attr, argv[arg] + 7), CRIT);
else if (!strncmp("--allow-unimpl=", argv[arg], 15))
allow_unimpl = str2bool(argv[arg] + 15);
else if (!strncmp("--perf-template=", argv[arg], 16))
perf_template = argv[arg] + 16;
else if (!strcmp("--reset", argv[arg]))
reset_parameters();
else if (!strncmp("--mode=", argv[0], 7))
bench_mode = str2bench_mode(argv[0] + 7);
else if (!strncmp("-v", argv[arg], 2))
verbose = atoi(argv[arg] + 2);
else if (!strncmp("--verbose=", argv[arg], 10))
verbose = atoi(argv[arg] + 10);
else {
desc_t c;
if (str2desc(&c, argv[arg]) == FAIL) {
fprintf(stderr, "driver: unknown option: `%s`, exiting...\n",
argv[arg]);
exit(2);
}
check_correctness(&c);
}
}
return OK;
}
}
|
