ComputeShaderShowdown/main.cpp at master · dooglz/ComputeShaderShowdown · 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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#include <iostream>
#include <string>

#include "profiling/Profiling.h"


#ifdef ss_compile_VK
#include "vulkan/bk_vulkan.h"
#endif

#ifdef ss_compile_CL
#include "opencl/bk_opencl.h"
#endif

#ifdef ss_compile_CU
#include "cuda/bk_cuda.h"
#endif

#ifdef ss_compile_DX
#include "dx12/bk_dx12.h"
#endif

#include "deps/CLI11/include/CLI/CLI.hpp"


bool WAIT = false;
void wait() {
	if (WAIT) {
		do {
			std::cout << '\n' << "Press Retrun to continue...";
		} while (std::cin.get() != '\n');
	}
}

int main(int argc, char** argv) {
	CLI::App app{ "Shader Showdown" };

	bool doVk = false;
	bool doDX12 = false;
	bool doCL = false;
	bool doCU = false;
	bool useRD = false;
	bool useAMDTAL = false;
	bool all = false;

#ifdef ss_compile_RD
	app.add_flag("--rd,--renderdoc", useRD, "load & use RenderDoc API");
#endif

#ifdef ss_compile_AMDTAL
	app.add_flag("--codeXL,--amdtal,--AMDTActivityLogger", useAMDTAL, "load & use AMDTAL API");
#endif

	app.add_flag("-w,--wait", WAIT, "prompt for user input between tests");

#ifdef ss_compile_VK
	app.add_flag("--vk,--vulkan", doVk, "Vulkan");
#endif

#ifdef ss_compile_CL
	app.add_flag("--cl,--opencl", doCL, "CL");
#endif

#ifdef ss_compile_CU
	app.add_flag("--cu,--cuda", doCU, "CU");
#endif

#ifdef ss_compile_DX
	app.add_flag("--dx,--dx12", doDX12, "DX12");
#endif

	app.add_flag("-a,--all", all, "run all ");
	uint8_t dev = 0;
	app.add_option("-d,--device", dev, "Device to run on");
	std::string modulepath = "";
	app.add_option("-m,--module", modulepath, "Module to run on");
	CLI11_PARSE(app, argc, argv);

	if (all) {
		doVk = true;
		doDX12 = true;
		doCL = true;
		doCU = true;
	}

	profiling::init((useAMDTAL * profiling::AMDEVENT)| (useRD * profiling::RENDERDOC));


#ifdef ss_compile_VK
	if (doVk) {
		std::cout << '\n' << std::string(80, '-') << "\nVULKAN\n";
		profiling::startProfiling("Init");
		VK_init(dev);
		profiling::endProfiling();
		profiling::startProfiling("Go");
		VK_go(10);
		profiling::endProfiling();
		VK_deInit();
		wait();
	}
#endif
#ifdef ss_compile_DX
	if (doDX12) {
		std::cout << '\n' << std::string(80, '-') << "\nDX12\n";
		DX12_init(dev);
		DX12_go(10);
		DX12_deInit();
		wait();
	}
#endif
#ifdef ss_compile_CL
	if (doCL) {
		std::cout << '\n' << std::string(80, '-') << "\nOPENCL\n";
		OPENCL_init(dev);
		if (modulepath != "") {
			OPENCL_runModule(modulepath,10);
		}else{
			OPENCL_go(10);
		}
		OPENCL_deInit();
		wait();
	}
#endif
#ifdef ss_compile_CU
	if (doCU) {
		std::cout << '\n' << std::string(80, '-') << "\nCUDA\n";
		CUDA_init(dev);
		CUDA_go(10);
		CUDA_deInit();
		wait();
	}
#endif
	wait();
	return 0;
}