| 174 | }; |
| 175 | |
| 176 | std::vector<BenchArgs> get_facerec_bench_args(size_t batch = 64) { |
| 177 | std::vector<BenchArgs> args; |
| 178 | args.emplace_back(BenchArgs{1, 4096, 64, 64, 4096, 1, 1}); |
| 179 | args.emplace_back(BenchArgs{batch, 128, 24, 24, 128, 1, 1}); |
| 180 | args.emplace_back(BenchArgs{batch, 256, 12, 12, 256, 1, 1}); |
| 181 | args.emplace_back(BenchArgs{batch, 512, 6, 6, 512, 1, 1}); |
| 182 | args.emplace_back(BenchArgs{batch, 1024, 4, 2, 1024, 1, 1}); |
| 183 | args.emplace_back(BenchArgs{batch, 108, 32, 32, 192, 1, 1}); |
| 184 | args.emplace_back(BenchArgs{batch, 192, 16, 16, 384, 1, 1}); |
| 185 | args.emplace_back(BenchArgs{batch, 384, 8, 8, 640, 1, 1}); |
| 186 | args.emplace_back(BenchArgs{batch, 108, 32, 32, 192, 1, 2}); |
| 187 | args.emplace_back(BenchArgs{batch, 192, 16, 16, 192, 1, 1}); |
| 188 | args.emplace_back(BenchArgs{batch, 192, 16, 16, 384, 1, 2}); |
| 189 | args.emplace_back(BenchArgs{batch, 384, 8, 8, 384, 1, 1}); |
| 190 | args.emplace_back(BenchArgs{batch, 384, 8, 8, 640, 1, 2}); |
| 191 | args.emplace_back(BenchArgs{batch, 640, 4, 4, 640, 1, 1}); |
| 192 | |
| 193 | return args; |
| 194 | } |
| 195 | |
| 196 | void benchmark_target_algo( |
| 197 | Handle* handle, const std::vector<BenchArgs>& args, DType src_dtype, |