| 1456 | } |
| 1457 | |
| 1458 | void checker_conv_bias_preprocess( |
| 1459 | std::vector<conv_bias::TestArg> args, Handle* handle, RNG* rng, float epsilon, |
| 1460 | DType type0, DType type1, DType type2, DType type3, const char* algo_name) { |
| 1461 | using namespace conv_bias; |
| 1462 | |
| 1463 | Checker<ConvBiasForward, OprWeightPreprocessProxy<ConvBiasForward>> checker(handle); |
| 1464 | checker.set_before_exec_callback( |
| 1465 | conv_bias::ConvBiasAlgoChecker<ConvBias>(algo_name)); |
| 1466 | checker.set_dtype(0, type0); |
| 1467 | checker.set_dtype(1, type1); |
| 1468 | checker.set_dtype(2, type2); |
| 1469 | checker.set_dtype(4, type3); |
| 1470 | checker.set_epsilon(epsilon); |
| 1471 | if (NULL != rng) { |
| 1472 | checker.set_rng(0, rng).set_rng(1, rng).set_rng(2, rng).set_rng(3, rng); |
| 1473 | } |
| 1474 | for (auto&& arg : args) { |
| 1475 | checker.set_param(arg.param).execs({arg.src, arg.filter, arg.bias, {}, {}}); |
| 1476 | } |
| 1477 | } |
| 1478 | |
| 1479 | } // namespace |
| 1480 | |