Function TEST_F

dnn/test/rocm/sleep.cpp:14–36

#if !(MEGDNN_AARCH64)

TEST_F(ROCM, SLEEP) {
    auto opr = this->handle_rocm()->create_operator<Sleep>();

    auto run = [&](float time) -> double {
        opr->param() = {time};
        hip_check(hipDeviceSynchronize());
        auto t0 = std::chrono::high_resolution_clock::now();
        opr->exec();
        hip_check(hipDeviceSynchronize());
        auto t1 = std::chrono::high_resolution_clock::now();
        std::chrono::duration<double> diff = t1 - t0;
        return diff.count();
    };

    // warmv7up
    run(0.01);

    for (auto i : {0.1, 0.3}) {
        auto get = run(i);
        ASSERT_GE(get, i);
        ASSERT_LE(get, i * 2);
    }
}

#endif