#include "test/naive/fixture.h" #include "megdnn/oprs.h" #include "test/common/checker.h" #include "test/common/extra_impl_helper.h" namespace megdnn { namespace test { TEST_F(NAIVE, ADD_UPDATE_BFLOAT16) { Checker checker(handle(), false); param::AddUpdate p{2, -1, 3}; auto extra_impl = extra_impl_helper(handle(), p); checker.set_param(p) .set_dtype(0, dtype::BFloat16()) .set_dtype(1, dtype::BFloat16()) .set_extra_opr_impl(extra_impl) .execs({{2, 2, 3}, {2, 2, 3}}); } } // namespace test } // namespace megdnn // vim: syntax=cpp.doxygen foldmethod=marker foldmarker=f{{{,f}}}