Function TEST_F

dnn/test/cuda/local.cpp:12–25

namespace test {

TEST_F(CUDA, LOCAL_FORWARD) {
    auto args = local::get_args_for_cuda();
    for (auto&& arg : args) {
        Checker<LocalForward> checker(handle_cuda());
        cudaStream_t stream;
        ASSERT_EQ(
                megcoreSuccess,
                megcoreGetCUDAStream(
                        handle_cuda()->megcore_computing_handle(), &stream));
        pollute_shared_mem(stream);
        checker.set_param(arg.param).exec(
                TensorShapeArray{arg.sshape(), arg.fshape(), arg.dshape()});
    }
}

TEST_F(CUDA, LOCAL_BACKWARD_DATA) {
    using namespace local;