#include #include #include #include #include #include int gemm_profiler(int, char*[]); int conv_profiler(int, char*[]); int main(int argc, char* argv[]) { if(strcmp(argv[1], "gemm") == 0) { return gemm_profiler(argc, argv); } else if(strcmp(argv[1], "conv") == 0) { return conv_profiler(argc, argv); } else { printf("arg1: tensor operation (gemm=GEMM, conv=Convolution)\n"); return 0; } }