1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
|
/*
* Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
*
* This software is available to you under a choice of one of two
* licenses. You may choose to be licensed under the terms of the GNU
* General Public License (GPL) Version 2, available from the file
* COPYING in the main directory of this source tree, or the
* OpenIB.org BSD license below:
*
* Redistribution and use in source and binary forms, with or
* without modification, are permitted provided that the following
* conditions are met:
*
* - Redistributions of source code must retain the above
* copyright notice, this list of conditions and the following
* disclaimer.
*
* - Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials
* provided with the distribution.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#pragma once
#if __cplusplus
#include <chrono>
#include <thread>
#else
#include <time.h>
#include <stdint.h>
#endif
#if __cplusplus
namespace nbu
{
namespace mft
{
namespace common
{
/**
* @brief Pauses the execution of the current thread for at least the specified number of milliseconds.
* @param msecs The number of milliseconds to sleep.
*
* The execution of the current thread is stopped until at least msecs milliseconds has
# passed from now. Other threads continue their execution.
*/
inline void mft_msleep(uint32_t msecs)
{
::std::this_thread::sleep_for(std::chrono::milliseconds(msecs));
}
/**
* @brief Pauses the execution of the current thread for at least the specified number of microseconds.
* @param msecs The number of microseconds to sleep.
*
* The execution of the current thread is stopped until at least msecs microseconds has
# passed from now. Other threads continue their execution.
*/
inline void mft_usleep(uint64_t usecs)
{
::std::this_thread::sleep_for(std::chrono::microseconds(usecs));
}
} // namespace common
} // namespace mft
} // namespace nbu
#define msleep(x) nbu::mft::common::mft_msleep(x)
#else // C {
/**
* @brief Pauses the execution of the current thread for at least the specified number of milliseconds.
* @param msecs The number of milliseconds for which to pause the execution.
* @return 0 on success, -1 otherwise. If interrupted by a signal, `errno` is set to `EINTR`.
*
* The execution of the current thread is stopped until at least msecs milliseconds has passed from now.
* Other threads continue their execution. There no support for Windows platforms.
*/
static inline int mft_msleep(uint32_t msecs)
{
struct timespec req;
req.tv_sec = msecs / 1000;
req.tv_nsec = ((long int)msecs % 1000) * 1000000;
return nanosleep(&req, NULL);
}
/**
* @brief Pauses the execution of the current thread for at least the specified number of microseconds.
* @param msecs The number of microseconds for which to pause the execution.
* @return 0 on success, -1 otherwise. If interrupted by a signal, `errno` is set to `EINTR`.
*
* The execution of the current thread is stopped until at least msecs microseconds has passed from now.
* Other threads continue their execution. There no support for Windows platforms.
*/
static inline int mft_usleep(uint64_t usecs)
{
struct timespec req;
req.tv_sec = usecs / 1000000;
req.tv_nsec = ((long int)usecs % 1000000) * 1000;
return nanosleep(&req, NULL);
}
#define msleep(x) mft_msleep(x)
#endif // } C
|