| 141 | } |
| 142 | |
| 143 | std::vector<BenchArgs> get_feat_model_args() { |
| 144 | std::vector<BenchArgs> args; |
| 145 | |
| 146 | args.emplace_back(BenchArgs{2, 4096, 4096}); |
| 147 | args.emplace_back(BenchArgs{2, 1024, 6912}); |
| 148 | args.emplace_back(BenchArgs{2, 3456, 3456}); |
| 149 | args.emplace_back(BenchArgs{2, 2304, 2304}); |
| 150 | args.emplace_back(BenchArgs{1, 256, 8192}); |
| 151 | args.emplace_back(BenchArgs{2, 864, 864}); |
| 152 | args.emplace_back(BenchArgs{2, 9, 64}); |
| 153 | |
| 154 | args.emplace_back(BenchArgs{4, 4096, 4096}); |
| 155 | args.emplace_back(BenchArgs{4, 1024, 6912}); |
| 156 | args.emplace_back(BenchArgs{4, 3456, 3456}); |
| 157 | args.emplace_back(BenchArgs{4, 2304, 2304}); |
| 158 | args.emplace_back(BenchArgs{2, 256, 8192}); |
| 159 | args.emplace_back(BenchArgs{4, 864, 864}); |
| 160 | args.emplace_back(BenchArgs{4, 9, 64}); |
| 161 | |
| 162 | args.emplace_back(BenchArgs{8, 4096, 4096}); |
| 163 | args.emplace_back(BenchArgs{8, 1024, 6912}); |
| 164 | args.emplace_back(BenchArgs{8, 3456, 3456}); |
| 165 | args.emplace_back(BenchArgs{8, 2304, 2304}); |
| 166 | args.emplace_back(BenchArgs{4, 256, 8192}); |
| 167 | args.emplace_back(BenchArgs{8, 864, 864}); |
| 168 | args.emplace_back(BenchArgs{4, 9, 64}); |
| 169 | |
| 170 | args.emplace_back(BenchArgs{16, 4096, 4096}); |
| 171 | args.emplace_back(BenchArgs{16, 1024, 6912}); |
| 172 | args.emplace_back(BenchArgs{16, 3456, 3456}); |
| 173 | args.emplace_back(BenchArgs{16, 2304, 2304}); |
| 174 | args.emplace_back(BenchArgs{8, 256, 8192}); |
| 175 | args.emplace_back(BenchArgs{16, 864, 864}); |
| 176 | args.emplace_back(BenchArgs{8, 9, 64}); |
| 177 | |
| 178 | args.emplace_back(BenchArgs{32, 4096, 4096}); |
| 179 | args.emplace_back(BenchArgs{32, 1024, 6912}); |
| 180 | args.emplace_back(BenchArgs{32, 3456, 3456}); |
| 181 | args.emplace_back(BenchArgs{32, 2304, 2304}); |
| 182 | args.emplace_back(BenchArgs{16, 256, 8192}); |
| 183 | args.emplace_back(BenchArgs{32, 864, 864}); |
| 184 | args.emplace_back(BenchArgs{32, 9, 64}); |
| 185 | |
| 186 | args.emplace_back(BenchArgs{64, 4096, 4096}); |
| 187 | args.emplace_back(BenchArgs{64, 1024, 6912}); |
| 188 | args.emplace_back(BenchArgs{64, 3456, 3456}); |
| 189 | args.emplace_back(BenchArgs{64, 2304, 2304}); |
| 190 | args.emplace_back(BenchArgs{32, 256, 8192}); |
| 191 | args.emplace_back(BenchArgs{64, 864, 864}); |
| 192 | args.emplace_back(BenchArgs{64, 9, 64}); |
| 193 | |
| 194 | args.emplace_back(BenchArgs{128, 4096, 4096}); |
| 195 | args.emplace_back(BenchArgs{128, 1024, 6912}); |
| 196 | args.emplace_back(BenchArgs{128, 3456, 3456}); |
| 197 | args.emplace_back(BenchArgs{128, 2304, 2304}); |
| 198 | args.emplace_back(BenchArgs{64, 256, 8192}); |
| 199 | args.emplace_back(BenchArgs{128, 864, 864}); |
| 200 | args.emplace_back(BenchArgs{128, 9, 64}); |