| 129 | }; |
| 130 | |
| 131 | std::vector<BenchArgs> get_square_matmul_args() { |
| 132 | std::vector<BenchArgs> args; |
| 133 | args.emplace_back(BenchArgs{128, 128, 128}); |
| 134 | args.emplace_back(BenchArgs{256, 256, 256}); |
| 135 | args.emplace_back(BenchArgs{512, 512, 512}); |
| 136 | args.emplace_back(BenchArgs{1024, 1024, 1024}); |
| 137 | args.emplace_back(BenchArgs{2048, 2048, 2048}); |
| 138 | args.emplace_back(BenchArgs{4096, 4096, 4096}); |
| 139 | |
| 140 | return args; |
| 141 | } |
| 142 | |
| 143 | std::vector<BenchArgs> get_feat_model_args() { |
| 144 | std::vector<BenchArgs> args; |