X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;ds=sidebyside;f=tmk_core%2Fcommon%2Fwait.h;h=a7cded942021482dbc7cec3b431108c55ebfa0f2;hb=66ef1e3d6770e388d631d25e4cd1c6b1640cea24;hp=82727be01254d93eb73f4eb9a05945df197583ff;hpb=685c8ba64c634d9ba419a714ca0e2cbae14b6dda;p=qmk_firmware.git

diff --git a/tmk_core/common/wait.h b/tmk_core/common/wait.h
index 82727be01..a7cded942 100644
--- a/tmk_core/common/wait.h
+++ b/tmk_core/common/wait.h
@@ -1,6 +1,8 @@
 #ifndef WAIT_H
 #define WAIT_H
 
+#include <inttypes.h>
+
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -9,13 +11,16 @@ extern "C" {
 #   include <util/delay.h>
 #   define wait_ms(ms)  _delay_ms(ms)
 #   define wait_us(us)  _delay_us(us)
-#elif defined(PROTOCOL_CHIBIOS) /* __AVR__ */
+#elif defined PROTOCOL_CHIBIOS
 #   include "ch.h"
 #   define wait_ms(ms) chThdSleepMilliseconds(ms)
 #   define wait_us(us) chThdSleepMicroseconds(us)
-#elif defined(__arm__) /* __AVR__ */
+#elif defined(__arm__)
 #   include "wait_api.h"
-#endif /* __AVR__ */
+#else  // Unit tests
+void wait_ms(uint32_t ms);
+#define wait_us(us) wait_ms(us / 1000)
+#endif
 
 #ifdef __cplusplus
 }