| 160 | }; |
| 161 | |
| 162 | void run_backward_test(DType dtype) { |
| 163 | IdxRng rng; |
| 164 | TaskRecordChecker<ArgsortBackward> checker(2); |
| 165 | checker.set_dtype(1, dtype::Int32()).set_rng(1, &rng); |
| 166 | checker.set_dtype(0, dtype); |
| 167 | checker.set_dtype(2, dtype); |
| 168 | for (size_t i = 16; i < 4096; i *= 2) { |
| 169 | checker.execs({{3, i}, {3, i}, {3, i}}); |
| 170 | checker.execs({{3, i + 3}, {3, i + 3}, {3, i + 3}}); |
| 171 | checker.execs({{3, i + 3}, {3, i + 3}, {3, i + 7}}); |
| 172 | } |
| 173 | } |
| 174 | |
| 175 | } // anonymous namespace |
| 176 |