"git@developer.sourcefind.cn:yuhai/deepks_pytorch.git" did not exist on "47af8be9072b26f85c445b90df3c759a52bd7f73"
profiler.cpp 2.09 KB
Newer Older
1
2
3
4
5
6
7
#include <iostream>
#include <numeric>
#include <initializer_list>
#include <cstdlib>
#include <stdlib.h>
#include <half.hpp>

Chao Liu's avatar
Chao Liu committed
8
int profile_gemm(int, char*[]);
zjing14's avatar
zjing14 committed
9
int profile_batched_gemm(int, char*[]);
Chao Liu's avatar
Chao Liu committed
10
11
int profile_gemm_bias_relu(int, char*[]);
int profile_gemm_bias_relu_add(int, char*[]);
Chao Liu's avatar
Chao Liu committed
12
13
14
15
int profile_conv_fwd(int, char*[]);
int profile_conv_fwd_bias_relu(int, char*[]);
int profile_conv_fwd_bias_relu_add(int, char*[]);
int profile_conv_fwd_bias_relu_atomic_add(int, char*[]);
16
17
18
19
20

int main(int argc, char* argv[])
{
    if(strcmp(argv[1], "gemm") == 0)
    {
Chao Liu's avatar
Chao Liu committed
21
        return profile_gemm(argc, argv);
22
    }
zjing14's avatar
zjing14 committed
23
    else if(strcmp(argv[1], "gemm_bias_relu") == 0)
Chao Liu's avatar
Chao Liu committed
24
25
26
    {
        return profile_gemm_bias_relu(argc, argv);
    }
zjing14's avatar
zjing14 committed
27
    else if(strcmp(argv[1], "gemm_bias_relu_add") == 0)
Chao Liu's avatar
Chao Liu committed
28
29
30
    {
        return profile_gemm_bias_relu_add(argc, argv);
    }
zjing14's avatar
zjing14 committed
31
32
33
34
    else if(strcmp(argv[1], "batched_gemm") == 0)
    {
        return profile_batched_gemm(argc, argv);
    }
Chao Liu's avatar
Chao Liu committed
35
    else if(strcmp(argv[1], "conv_fwd") == 0)
36
    {
Chao Liu's avatar
Chao Liu committed
37
38
39
40
41
42
43
44
45
46
47
48
49
        return profile_conv_fwd(argc, argv);
    }
    else if(strcmp(argv[1], "conv_fwd_bias_relu") == 0)
    {
        return profile_conv_fwd_bias_relu(argc, argv);
    }
    else if(strcmp(argv[1], "conv_fwd_bias_relu_add") == 0)
    {
        return profile_conv_fwd_bias_relu_add(argc, argv);
    }
    else if(strcmp(argv[1], "conv_fwd_bias_relu_atomic_add") == 0)
    {
        return profile_conv_fwd_bias_relu_atomic_add(argc, argv);
50
51
52
    }
    else
    {
Chao Liu's avatar
Chao Liu committed
53
54
55
56
57
58
59
60
61
62
        // clang-format off
        printf("arg1: tensor operation (gemm: GEMM\n"
               "                        gemm_bias_relu: GEMM+Bias+ReLU\n"
               "                        gemm_bias_relu_add: GEMM+Bias+ReLU+Add\n"
               "                        conv_fwd: ForwardConvolution\n"
               "                        conv_fwd_bias_relu: ForwardConvolution+Bias+ReLU\n"
               "                        conv_fwd_bias_relu_add: ForwardConvolution+Bias+ReLU+Add\n"
               "                        conv_fwd_bias_relu_atomic_add: ForwardConvolution+Bias+ReLU+AtomicAdd\n");
        // clang-format on

63
64
65
        return 0;
    }
}