| 501 | } |
| 502 | )"; |
| 503 | void puzzle14(Context &ctx) { |
| 504 | printf("\n\nPuzzle 14\n\n"); |
| 505 | static constexpr size_t N = 2; |
| 506 | static constexpr size_t TPB = 3; |
| 507 | Tensor a = createTensor(ctx, {N, N}, kf32, makeData<N * N>().data()); |
| 508 | Tensor b = createTensor(ctx, {N, N}, kf32, makeData<N * N>().data()); |
| 509 | Tensor output = createTensor(ctx, {N, N}, kf32); |
| 510 | struct Params { |
| 511 | uint32_t TPB = TPB; |
| 512 | uint32_t size = N; |
| 513 | }; |
| 514 | |
| 515 | Kernel op = |
| 516 | createKernel(ctx, {kPuzzle14, {TPB, TPB, 1}}, |
| 517 | Bindings{a, b, output}, {1, 1, 1}, Params{TPB, N}); |
| 518 | showResult<N, N, N>(ctx, op, output); |
| 519 | } |
| 520 | |
| 521 | |
| 522 | int main(int argc, char **argv) { |
nothing calls this directly
no test coverage detected